char *
read_filename (void)
{
char fname[_POSIX_PATH_MAX];
int r;
fd_printf(STO, "\r\n*** file: ");
r = fd_readline(STI, STO, fname, sizeof(fname));
fd_printf(STO, "\r\n");
if ( r < 0 )
return NULL;
else
return strdup(fname);
}
int
read_baud (void)
{
char baudstr[9], *ep;
int baud = -1, r;
do {
fd_printf(STO, "\r\n*** baud: ");
r = fd_readline(STI, STO, baudstr, sizeof(baudstr));
fd_printf(STO, "\r\n");
if ( r < 0 )
break;
baud = strtol(baudstr, &ep, 0);
if ( ! ep || *ep != '\0' || ! term_baud_ok(baud) || baud == 0 ) {
fd_printf(STO, "*** Invalid baudrate!");
baud = -1;
}
} while (baud < 0);
return baud;
}
void
loop(void)
{
enum {
ST_COMMAND,
ST_TRANSPARENT
} state;
int dtr_up;
fd_set rdset, wrset;
int newbaud, newflow, newparity, newbits;
char *newflow_str, *newparity_str;
char fname[128];
int r, n;
unsigned char c;
tty_q.len = 0;
state = ST_TRANSPARENT;
dtr_up = 0;
for (;;) {
FD_ZERO(&rdset);
FD_ZERO(&wrset);
FD_SET(STI, &rdset);
FD_SET(tty_fd, &rdset);
if ( tty_q.len ) FD_SET(tty_fd, &wrset);
if (select(FD_SETSIZE, &rdset, &wrset, NULL, NULL) < 0)
fatal("select failed: %d : %s", errno, strerror(errno));
if ( FD_ISSET(STI, &rdset) ) {
/* read from terminal */
do {
n = read(STI, &c, 1);
} while (n < 0 && errno == EINTR);
if (n == 0)
fatal("stdin closed");
else if (n < 0)
fatal("read from stdin failed: %s", strerror(errno));
switch (state) {
case ST_COMMAND:
if ( c == opts.escape ) {
state = ST_TRANSPARENT;
/* pass the escape character down */
if (tty_q.len <= TTY_Q_SZ)
tty_q.buff[tty_q.len++] = c;
else
fd_printf(STO, "\x07");
break;
}
state = ST_TRANSPARENT;
switch (c) {
case KEY_EXIT:
return;
case KEY_QUIT:
term_set_hupcl(tty_fd, 0);
term_flush(tty_fd);
term_apply(tty_fd);
term_erase(tty_fd);
return;
case KEY_STATUS:
fd_printf(STO, "\r\n");
fd_printf(STO, "*** baud: %d\r\n", opts.baud);
fd_printf(STO, "*** flow: %s\r\n", opts.flow_str);
fd_printf(STO, "*** parity: %s\r\n", opts.parity_str);
fd_printf(STO, "*** databits: %d\r\n", opts.databits);
fd_printf(STO, "*** dtr: %s\r\n", dtr_up ? "up" : "down");
break;
case KEY_PULSE:
fd_printf(STO, "\r\n*** pulse DTR ***\r\n");
if ( term_pulse_dtr(tty_fd) < 0 )
fd_printf(STO, "*** FAILED\r\n");
break;
case KEY_TOGGLE:
if ( dtr_up )
r = term_lower_dtr(tty_fd);
else
r = term_raise_dtr(tty_fd);
if ( r >= 0 ) dtr_up = ! dtr_up;
fd_printf(STO, "\r\n*** DTR: %s ***\r\n",
dtr_up ? "up" : "down");
break;
case KEY_BAUD_UP:
newbaud = baud_up(opts.baud);
term_set_baudrate(tty_fd, newbaud);
tty_q.len = 0; term_flush(tty_fd);
if ( term_apply(tty_fd) >= 0 ) opts.baud = newbaud;
fd_printf(STO, "\r\n*** baud: %d ***\r\n", opts.baud);
break;
case KEY_BAUD_DN:
newbaud = baud_down(opts.baud);
term_set_baudrate(tty_fd, newbaud);
tty_q.len = 0; term_flush(tty_fd);
if ( term_apply(tty_fd) >= 0 ) opts.baud = newbaud;
fd_printf(STO, "\r\n*** baud: %d ***\r\n", opts.baud);
break;
case KEY_FLOW:
newflow = flow_next(opts.flow, &newflow_str);
term_set_flowcntrl(tty_fd, newflow);
tty_q.len = 0; term_flush(tty_fd);
if ( term_apply(tty_fd) >= 0 ) {
opts.flow = newflow;
opts.flow_str = newflow_str;
}
fd_printf(STO, "\r\n*** flow: %s ***\r\n", opts.flow_str);
break;
case KEY_PARITY:
newparity = parity_next(opts.parity, &newparity_str);
term_set_parity(tty_fd, newparity);
tty_q.len = 0; term_flush(tty_fd);
if ( term_apply(tty_fd) >= 0 ) {
opts.parity = newparity;
opts.parity_str = newparity_str;
}
fd_printf(STO, "\r\n*** parity: %s ***\r\n",
opts.parity_str);
break;
case KEY_BITS:
newbits = bits_next(opts.databits);
term_set_databits(tty_fd, newbits);
tty_q.len = 0; term_flush(tty_fd);
if ( term_apply(tty_fd) >= 0 ) opts.databits = newbits;
fd_printf(STO, "\r\n*** databits: %d ***\r\n",
opts.databits);
break;
case KEY_SEND:
fd_printf(STO, "\r\n*** file: ");
r = fd_readline(STI, STO, fname, sizeof(fname));
fd_printf(STO, "\r\n");
if ( r < -1 && errno == EINTR ) break;
if ( r <= -1 )
fatal("cannot read filename: %s", strerror(errno));
run_cmd(tty_fd, opts.send_cmd, fname, NULL);
break;
case KEY_RECEIVE:
fd_printf(STO, "*** file: ");
r = fd_readline(STI, STO, fname, sizeof(fname));
fd_printf(STO, "\r\n");
if ( r < -1 && errno == EINTR ) break;
if ( r <= -1 )
fatal("cannot read filename: %s", strerror(errno));
if ( fname[0] )
run_cmd(tty_fd, opts.send_cmd, fname, NULL);
else
run_cmd(tty_fd, opts.receive_cmd, NULL);
break;
case KEY_BREAK:
term_break(tty_fd);
fd_printf(STO, "\r\n*** break sent ***\r\n");
break;
default:
break;
}
break;
case ST_TRANSPARENT:
if ( c == opts.escape ) {
state = ST_COMMAND;
} else {
if (tty_q.len <= TTY_Q_SZ)
tty_q.buff[tty_q.len++] = c;
else
fd_printf(STO, "\x07");
}
break;
default:
assert(0);
break;
}
}
if ( FD_ISSET(tty_fd, &rdset) ) {
/* read from port */
do {
n = read(tty_fd, &c, 1);
} while (n < 0 && errno == EINTR);
if (n == 0)
fatal("term closed");
else if ( n < 0 )
fatal("read from term failed: %s", strerror(errno));
do {
n = write(STO, &c, 1);
} while ( errno == EAGAIN
|| errno == EWOULDBLOCK
|| errno == EINTR );
if ( n <= 0 )
fatal("write to stdout failed: %s", strerror(errno));
}
if ( FD_ISSET(tty_fd, &wrset) ) {
/* write to port */
do {
n = write(tty_fd, tty_q.buff, tty_q.len);
} while ( n < 0 && errno == EINTR );
if ( n <= 0 )
fatal("write to term failed: %s", strerror(errno));
memcpy(tty_q.buff, tty_q.buff + n, tty_q.len - n);
tty_q.len -= n;
}
}
}
int main(/*int argc, char *argv[]*/) {
INFO("Started\n");
int server_socket, client_socket, port, res, client_len, n;
struct sockaddr_in srv_addr, client_addr;
int iSetOption = 1;
port = SRVPORT;
server_socket = socket(AF_INET, SOCK_STREAM, 0);
setsockopt(server_socket, SOL_SOCKET, SO_REUSEADDR, (char*)&iSetOption,
sizeof(iSetOption));
if (server_socket < 0) {
ERROR("Error opening socket\n");
IFERROR(perror(""));
}
INFO("Made server socket\n");
memset((char *) &srv_addr, 0, sizeof(srv_addr));
srv_addr.sin_family = AF_INET;
srv_addr.sin_port = htons(port);
srv_addr.sin_addr.s_addr = INADDR_ANY;
while ((res = bind(server_socket,
(struct sockaddr *) &srv_addr,
sizeof(srv_addr))) < 0) {
ERROR("Error binding socket\n");
IFERROR(perror(""));
}
listen(server_socket, 5);
INFO("Listening on port %d\n", port);
client_len = sizeof(client_addr);
if (chdir("content")) {
ERROR("Could not chdir to the content directory\n");
IFERROR(perror(""));
exit(1);
}
do {
client_socket = accept(server_socket,
(struct sockaddr *) &client_addr,
(socklen_t *) &client_len);
INFO("Accepted client socket\n");
if (client_socket < 0) {
ERROR("Error on client accept\n");
IFERROR(perror(""));
}
request *req = request_new();
char line[MAXLEN];
while ((n = fd_readline(client_socket, line, MAXLEN)) > 0) {
if (n == MAXLEN-1 && line[MAXLEN-1] != '\n') {
WARN("Line length exceeded MAXLEN %d\n", MAXLEN);
char c;
int dropped = 0;
while ((n = read(client_socket, &c, 1)) > 0) {
dropped += 1;
if (c == '\n')
break;
}
WARN("Skipped to the next line, dropped %d characters\n", dropped);
}
if (is_crlf(line)) { // We've reached the end of the headers
int left_to_read = request_get_content_length(req);
int total_read = 0, len;
if (left_to_read > 0) {
DEBUG("There's %d chars of body to read\n", left_to_read);
// There's a Content-Length so there must be a body
req->body = malloc(sizeof(char) * (left_to_read+1));
len = MAXLEN < left_to_read ? MAXLEN : left_to_read;
len += 1;
while (left_to_read && (n = fd_readline(client_socket, line, len)) > 0) {
DEBUG("Read %d characters of body: %s\n", n, line);
strcpy(&req->body[total_read], line);
total_read += n;
left_to_read -= n;
DEBUG("total_read: %d, left_to_read: %d\n", total_read, left_to_read);
len = MAXLEN < left_to_read ? MAXLEN : left_to_read;
}
req->body[total_read+1] = '\0';
DEBUG("Done reading body in %d chars:\n", total_read);
DEBUG("%s\n", req->body);
respond(&req, client_socket);
continue;
} else {
DEBUG("No body to read, generating response\n");
respond(&req, client_socket);
continue;
}
}
if (request_parse(req, line)) {
WARN("An error occurred while parsing the previous line\n");
}
}
DEBUG("Done with this request! n = %d\n", n);
} while (1);
close(server_socket);
close(client_socket);
return 0;
}
