emo/busybox
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

telnet: someone tried to do data optimization before me. Complete it.

Browse Source 
text    data     bss     dec     hex filename
   2558       0     404    2962     b92 busybox.t2/networking/telnet.o
   2542       0       0    2542     9ee busybox.t3/networking/telnet.o
This commit is contained in:
Denis Vlasenko 2007-03-19 14:47:09 +00:00
parent dfce08f281
commit f24cdf1554
1 changed files with 85 additions and 98 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

183
networking/telnet.c
View File

@ -32,10 +32,10 @@
#define TRACE(x, y) #define TRACE(x, y)
#endif #endif
#define DATABUFSIZE 128
#define IACBUFSIZE 128
enum { enum {
DATABUFSIZE = 128,
IACBUFSIZE = 128,
CHM_TRY = 0, CHM_TRY = 0,
CHM_ON = 1, CHM_ON = 1,
CHM_OFF = 2, CHM_OFF = 2,
@ -50,40 +50,36 @@ enum {
TS_SUB2 = 5, TS_SUB2 = 5,
}; };
#define WriteCS(fd, str) write(fd, str, sizeof str -1)
typedef unsigned char byte; typedef unsigned char byte;
/* use globals to reduce size ??? */ /* test this hypothesis later */
static struct Globalvars { struct globals {
int netfd; /* console fd:s are 0 and 1 (and 2) */ int netfd; /* console fd:s are 0 and 1 (and 2) */
/* same buffer used both for network and console read/write */ short iaclen; /* could even use byte */
char buf[DATABUFSIZE]; /* allocating so static size is smaller */
byte telstate; /* telnet negotiation state from network input */ byte telstate; /* telnet negotiation state from network input */
byte telwish; /* DO, DONT, WILL, WONT */ byte telwish; /* DO, DONT, WILL, WONT */
byte charmode; byte charmode;
byte telflags; byte telflags;
byte gotsig; byte gotsig;
byte do_termios; byte do_termios;
/* buffer to handle telnet negotiations */ #if ENABLE_FEATURE_TELNET_TTYPE
char *ttype;
#endif
#if ENABLE_FEATURE_TELNET_AUTOLOGIN
const char *autologin;
#endif
#if ENABLE_FEATURE_AUTOWIDTH
int win_width, win_height;
#endif
/* same buffer used both for network and console read/write */
char buf[DATABUFSIZE]; /* allocating so static size is smaller */
char iacbuf[IACBUFSIZE]; char iacbuf[IACBUFSIZE];
short iaclen; /* could even use byte */
struct termios termios_def; struct termios termios_def;
struct termios termios_raw; struct termios termios_raw;
} G; };
#define xUSE_GLOBALVAR_PTR /* xUSE... -> don't use :D (makes smaller code) */ #define G (*(struct globals*)bb_common_bufsiz1)
#ifdef USE_GLOBALVAR_PTR
struct Globalvars * Gptr;
#define G (*Gptr)
#endif
static void iacflush(void)
{
write(G.netfd, G.iacbuf, G.iaclen);
G.iaclen = 0;
}
/* Function prototypes */ /* Function prototypes */
static void rawmode(void); static void rawmode(void);
@ -93,20 +89,16 @@ static void will_charmode(void);
static void telopt(byte c); static void telopt(byte c);
static int subneg(byte c); static int subneg(byte c);
/* Some globals */ static void iacflush(void)
#ifdef CONFIG_FEATURE_TELNET_TTYPE {
static char *ttype; write(G.netfd, G.iacbuf, G.iaclen);
#endif G.iaclen = 0;
}
#ifdef CONFIG_FEATURE_TELNET_AUTOLOGIN #define write_str(fd, str) write(fd, str, sizeof(str) - 1)
static const char *autologin;
#endif
#ifdef CONFIG_FEATURE_AUTOWIDTH static void /* buffer to handle telnet negotiations */
static int win_width, win_height; doexit(int ev)
#endif
static void doexit(int ev)
{ {
cookmode(); cookmode();
exit(ev); exit(ev);
@ -119,7 +111,7 @@ static void conescape(void)
if (G.gotsig) /* came from line mode... go raw */ if (G.gotsig) /* came from line mode... go raw */
rawmode(); rawmode();
WriteCS(1, "\r\nConsole escape. Commands are:\r\n\n" write_str(1, "\r\nConsole escape. Commands are:\r\n\n"
" l go to line mode\r\n" " l go to line mode\r\n"
" c go to character mode\r\n" " c go to character mode\r\n"
" z suspend telnet\r\n" " z suspend telnet\r\n"
@ -128,18 +120,15 @@ static void conescape(void)
if (read(0, &b, 1) <= 0) if (read(0, &b, 1) <= 0)
doexit(1); doexit(1);
switch (b) switch (b) {
{
case 'l': case 'l':
if (!G.gotsig) if (!G.gotsig) {
{
do_linemode(); do_linemode();
goto rrturn; goto rrturn;
} }
break; break;
case 'c': case 'c':
if (G.gotsig) if (G.gotsig) {
{
will_charmode(); will_charmode();
goto rrturn; goto rrturn;
} }
@ -153,7 +142,7 @@ static void conescape(void)
doexit(0); doexit(0);
} }
WriteCS(1, "continuing...\r\n"); write_str(1, "continuing...\r\n");
if (G.gotsig) if (G.gotsig)
cookmode(); cookmode();
@ -296,7 +285,7 @@ static void putiac2(byte wwdd, byte c)
putiac(c); putiac(c);
} }
#ifdef CONFIG_FEATURE_TELNET_TTYPE #if ENABLE_FEATURE_TELNET_TTYPE
static void putiac_subopt(byte c, char *str) static void putiac_subopt(byte c, char *str)
{ {
int len = strlen(str) + 6; // ( 2 + 1 + 1 + strlen + 2 ) int len = strlen(str) + 6; // ( 2 + 1 + 1 + strlen + 2 )
@ -317,10 +306,10 @@ static void putiac_subopt(byte c, char *str)
} }
#endif #endif
#ifdef CONFIG_FEATURE_TELNET_AUTOLOGIN #if ENABLE_FEATURE_TELNET_AUTOLOGIN
static void putiac_subopt_autologin(void) static void putiac_subopt_autologin(void)
{ {
int len = strlen(autologin) + 6; // (2 + 1 + 1 + strlen + 2) int len = strlen(G.autologin) + 6; // (2 + 1 + 1 + strlen + 2)
const char *user = "USER"; const char *user = "USER";
if (G.iaclen + len > IACBUFSIZE) if (G.iaclen + len > IACBUFSIZE)
@ -337,15 +326,15 @@ static void putiac_subopt_autologin(void)
putiac(NEW_ENV_VALUE); putiac(NEW_ENV_VALUE);
while (*autologin) while (*G.autologin)
putiac(*autologin++); putiac(*G.autologin++);
putiac(IAC); putiac(IAC);
putiac(SE); putiac(SE);
} }
#endif #endif
#ifdef CONFIG_FEATURE_AUTOWIDTH #if ENABLE_FEATURE_AUTOWIDTH
static void putiac_naws(byte c, int x, int y) static void putiac_naws(byte c, int x, int y)
{ {
if (G.iaclen + 9 > IACBUFSIZE) if (G.iaclen + 9 > IACBUFSIZE)
@ -419,24 +408,27 @@ static void do_linemode(void)
static void to_notsup(char c) static void to_notsup(char c)
{ {
if (G.telwish == WILL) putiac2(DONT, c); if (G.telwish == WILL)
else if (G.telwish == DO) putiac2(WONT, c); putiac2(DONT, c);
else if (G.telwish == DO)
putiac2(WONT, c);
} }
static void to_echo(void) static void to_echo(void)
{ {
/* if server requests ECHO, don't agree */ /* if server requests ECHO, don't agree */
if (G.telwish == DO) { putiac2(WONT, TELOPT_ECHO); return; } if (G.telwish == DO) {
else if (G.telwish == DONT) return; putiac2(WONT, TELOPT_ECHO);
return;
}
if (G.telwish == DONT)
return;
if (G.telflags & UF_ECHO) if (G.telflags & UF_ECHO) {
{
if (G.telwish == WILL) if (G.telwish == WILL)
return; return;
} } else if (G.telwish == WONT)
else return;
if (G.telwish == WONT)
return;
if (G.charmode != CHM_OFF) if (G.charmode != CHM_OFF)
G.telflags ^= UF_ECHO; G.telflags ^= UF_ECHO;
@ -447,21 +439,18 @@ static void to_echo(void)
putiac2(DONT, TELOPT_ECHO); putiac2(DONT, TELOPT_ECHO);
setConMode(); setConMode();
WriteCS(1, "\r\n"); /* sudden modec */ write_str(1, "\r\n"); /* sudden modec */
} }
static void to_sga(void) static void to_sga(void)
{ {
/* daemon always sends will/wont, client do/dont */ /* daemon always sends will/wont, client do/dont */
if (G.telflags & UF_SGA) if (G.telflags & UF_SGA) {
{
if (G.telwish == WILL) if (G.telwish == WILL)
return; return;
} } else if (G.telwish == WONT)
else return;
if (G.telwish == WONT)
return;
if ((G.telflags ^= UF_SGA) & UF_SGA) /* toggle */ if ((G.telflags ^= UF_SGA) & UF_SGA) /* toggle */
putiac2(DO, TELOPT_SGA); putiac2(DO, TELOPT_SGA);
@ -469,31 +458,31 @@ static void to_sga(void)
putiac2(DONT, TELOPT_SGA); putiac2(DONT, TELOPT_SGA);
} }
#ifdef CONFIG_FEATURE_TELNET_TTYPE #if ENABLE_FEATURE_TELNET_TTYPE
static void to_ttype(void) static void to_ttype(void)
{ {
/* Tell server we will (or won't) do TTYPE */ /* Tell server we will (or won't) do TTYPE */
if(ttype) if (G.ttype)
putiac2(WILL, TELOPT_TTYPE); putiac2(WILL, TELOPT_TTYPE);
else else
putiac2(WONT, TELOPT_TTYPE); putiac2(WONT, TELOPT_TTYPE);
} }
#endif #endif
#ifdef CONFIG_FEATURE_TELNET_AUTOLOGIN #if ENABLE_FEATURE_TELNET_AUTOLOGIN
static void to_new_environ(void) static void to_new_environ(void)
{ {
/* Tell server we will (or will not) do AUTOLOGIN */ /* Tell server we will (or will not) do AUTOLOGIN */
if (autologin) if (G.autologin)
putiac2(WILL, TELOPT_NEW_ENVIRON); putiac2(WILL, TELOPT_NEW_ENVIRON);
else else
putiac2(WONT, TELOPT_NEW_ENVIRON); putiac2(WONT, TELOPT_NEW_ENVIRON);
} }
#endif #endif
#ifdef CONFIG_FEATURE_AUTOWIDTH #if ENABLE_FEATURE_AUTOWIDTH
static void to_naws(void) static void to_naws(void)
{ {
/* Tell server we will do NAWS */ /* Tell server we will do NAWS */
@ -508,18 +497,18 @@ static void telopt(byte c)
to_echo(); break; to_echo(); break;
case TELOPT_SGA: case TELOPT_SGA:
to_sga(); break; to_sga(); break;
#ifdef CONFIG_FEATURE_TELNET_TTYPE #if ENABLE_FEATURE_TELNET_TTYPE
case TELOPT_TTYPE: case TELOPT_TTYPE:
to_ttype(); break; to_ttype(); break;
#endif #endif
#ifdef CONFIG_FEATURE_TELNET_AUTOLOGIN #if ENABLE_FEATURE_TELNET_AUTOLOGIN
case TELOPT_NEW_ENVIRON: case TELOPT_NEW_ENVIRON:
to_new_environ(); break; to_new_environ(); break;
#endif #endif
#ifdef CONFIG_FEATURE_AUTOWIDTH #if ENABLE_FEATURE_AUTOWIDTH
case TELOPT_NAWS: case TELOPT_NAWS:
to_naws(); to_naws();
putiac_naws(c, win_width, win_height); putiac_naws(c, G.win_width, G.win_height);
break; break;
#endif #endif
default: default:
@ -535,17 +524,16 @@ static void telopt(byte c)
static int subneg(byte c) static int subneg(byte c)
{ {
switch (G.telstate) switch (G.telstate) {
{
case TS_SUB1: case TS_SUB1:
if (c == IAC) if (c == IAC)
G.telstate = TS_SUB2; G.telstate = TS_SUB2;
#ifdef CONFIG_FEATURE_TELNET_TTYPE #if ENABLE_FEATURE_TELNET_TTYPE
else else
if (c == TELOPT_TTYPE) if (c == TELOPT_TTYPE)
putiac_subopt(TELOPT_TTYPE,ttype); putiac_subopt(TELOPT_TTYPE, G.ttype);
#endif #endif
#ifdef CONFIG_FEATURE_TELNET_AUTOLOGIN #if ENABLE_FEATURE_TELNET_AUTOLOGIN
else else
if (c == TELOPT_NEW_ENVIRON) if (c == TELOPT_NEW_ENVIRON)
putiac_subopt_autologin(); putiac_subopt_autologin();
@ -578,6 +566,8 @@ static void cookmode(void)
if (G.do_termios) tcsetattr(0, TCSADRAIN, &G.termios_def); if (G.do_termios) tcsetattr(0, TCSADRAIN, &G.termios_def);
} }
void BUG_telnet_globals_too_big(void);
int telnet_main(int argc, char** argv); int telnet_main(int argc, char** argv);
int telnet_main(int argc, char** argv) int telnet_main(int argc, char** argv)
{ {
@ -591,16 +581,18 @@ int telnet_main(int argc, char** argv)
int maxfd; int maxfd;
#endif #endif
#ifdef CONFIG_FEATURE_AUTOWIDTH if (sizeof(G) > sizeof(bb_common_bufsiz1))
get_terminal_width_height(0, &win_width, &win_height); BUG_telnet_globals_too_big();
#endif
#ifdef CONFIG_FEATURE_TELNET_TTYPE
ttype = getenv("TERM");
#endif
/* memset(&G, 0, sizeof G); - already is */ /* memset(&G, 0, sizeof G); - already is */
#if ENABLE_FEATURE_AUTOWIDTH
get_terminal_width_height(0, &G.win_width, &G.win_height);
#endif
#if ENABLE_FEATURE_TELNET_TTYPE
G.ttype = getenv("TERM");
#endif
if (tcgetattr(0, &G.termios_def) >= 0) { if (tcgetattr(0, &G.termios_def) >= 0) {
G.do_termios = 1; G.do_termios = 1;
G.termios_raw = G.termios_def; G.termios_raw = G.termios_def;
@ -610,9 +602,9 @@ int telnet_main(int argc, char** argv)
if (argc < 2) if (argc < 2)
bb_show_usage(); bb_show_usage();
#ifdef CONFIG_FEATURE_TELNET_AUTOLOGIN #if ENABLE_FEATURE_TELNET_AUTOLOGIN
if (1 & getopt32(argc, argv, "al:", &autologin)) if (1 & getopt32(argc, argv, "al:", &G.autologin))
autologin = getenv("USER"); G.autologin = getenv("USER");
argv += optind; argv += optind;
#else #else
argv++; argv++;
@ -667,12 +659,9 @@ int telnet_main(int argc, char** argv)
#endif #endif
{ {
len = read(0, G.buf, DATABUFSIZE); len = read(0, G.buf, DATABUFSIZE);
if (len <= 0) if (len <= 0)
doexit(0); doexit(0);
TRACE(0, ("Read con: %d\n", len)); TRACE(0, ("Read con: %d\n", len));
handlenetoutput(len); handlenetoutput(len);
} }
@ -683,13 +672,11 @@ int telnet_main(int argc, char** argv)
#endif #endif
{ {
len = read(G.netfd, G.buf, DATABUFSIZE); len = read(G.netfd, G.buf, DATABUFSIZE);
if (len <= 0) { if (len <= 0) {
WriteCS(1, "Connection closed by foreign host.\r\n"); write_str(1, "Connection closed by foreign host\r\n");
doexit(1); doexit(1);
} }
TRACE(0, ("Read netfd (%d): %d\n", G.netfd, len)); TRACE(0, ("Read netfd (%d): %d\n", G.netfd, len));
handlenetinput(len); handlenetinput(len);
} }
} }