123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826 |
- #include <string.h>
- #include <stdlib.h>
- #include <assert.h>
- #include <byteswap.h>
- #include <latch.h>
- #include <errno.h>
- #include <gpxe/process.h>
- #include <gpxe/init.h>
- #include <gpxe/netdevice.h>
- #include <gpxe/pkbuff.h>
- #include <gpxe/ip.h>
- #include <gpxe/tcp.h>
- #include <gpxe/tcpip.h>
- #include "uip/uip.h"
-
-
-
- #if USE_UIP
-
-
- static void *tcp_buffer = uip_buf + ( 40 + UIP_LLH_LEN );
-
-
- static size_t tcp_buflen = UIP_BUFSIZE - ( 40 + UIP_LLH_LEN );
-
-
- void tcp_connect ( struct tcp_connection *conn ) {
- struct uip_conn *uip_conn;
- u16_t ipaddr[2];
-
- assert ( conn->sin.sin_addr.s_addr != 0 );
- assert ( conn->sin.sin_port != 0 );
- assert ( conn->tcp_op != NULL );
- assert ( sizeof ( uip_conn->appstate ) == sizeof ( conn ) );
-
- * ( ( uint32_t * ) ipaddr ) = conn->sin.sin_addr.s_addr;
- uip_conn = uip_connect ( ipaddr, conn->sin.sin_port );
- #warning "Use linked lists so that uip_connect() cannot fail"
- assert ( uip_conn != NULL );
- *( ( void ** ) uip_conn->appstate ) = conn;
- }
-
-
- void tcp_send ( struct tcp_connection *conn __unused,
- const void *data, size_t len ) {
-
- assert ( conn = *( ( void ** ) uip_conn->appstate ) );
-
- if ( len > tcp_buflen )
- len = tcp_buflen;
- memmove ( tcp_buffer, data, len );
-
- uip_send ( tcp_buffer, len );
- }
-
-
- void tcp_close ( struct tcp_connection *conn __unused ) {
- assert ( conn = *( ( void ** ) uip_conn->appstate ) );
- uip_close();
- }
-
-
- void uip_tcp_appcall ( void ) {
- struct tcp_connection *conn = *( ( void ** ) uip_conn->appstate );
- struct tcp_operations *op = conn->tcp_op;
-
- if ( op->closed ) {
- if ( uip_aborted() )
- op->closed ( conn, -ECONNABORTED );
- if ( uip_timedout() )
- op->closed ( conn, -ETIMEDOUT );
- if ( uip_closed() )
- op->closed ( conn, 0 );
- }
- if ( uip_connected() && op->connected )
- op->connected ( conn );
- if ( uip_acked() && op->acked )
- op->acked ( conn, uip_conn->len );
- if ( uip_newdata() && op->newdata )
- op->newdata ( conn, ( void * ) uip_appdata, uip_len );
- if ( ( uip_rexmit() || uip_newdata() || uip_acked() ||
- uip_connected() || uip_poll() ) && op->senddata )
- op->senddata ( conn, tcp_buffer, tcp_buflen );
- }
-
-
- void uip_udp_appcall ( void ) {
- }
-
-
- static void tcp_periodic ( void ) {
- struct pk_buff *pkb;
- int i;
-
- for ( i = 0 ; i < UIP_CONNS ; i++ ) {
- uip_periodic ( i );
- if ( uip_len > 0 ) {
- pkb = alloc_pkb ( uip_len + MAX_LL_HEADER_LEN);
- if ( ! pkb )
- continue;
-
- pkb_reserve ( pkb, MAX_LL_HEADER_LEN );
- pkb_put ( pkb, uip_len );
- memcpy ( pkb->data, uip_buf, uip_len );
-
- ipv4_uip_tx ( pkb );
- }
- }
- }
-
-
- void tcp_kick ( struct tcp_connection *conn __unused ) {
-
- tcp_periodic();
- }
-
-
- static void tcp_step ( struct process *process ) {
- static unsigned long timeout = 0;
-
- if ( currticks() > timeout ) {
- timeout = currticks() + ( TICKS_PER_SEC / 10 );
- tcp_periodic ();
- }
-
- schedule ( process );
- }
-
-
- static struct process tcp_process = {
- .step = tcp_step,
- };
-
-
- static void init_tcp ( void ) {
- schedule ( &tcp_process );
- }
-
- INIT_FN ( INIT_PROCESS, init_tcp, NULL, NULL );
-
- #else
-
-
- static LIST_HEAD ( tcp_conns );
-
-
- static const char *tcp_states[] = {
- "CLOSED",
- "LISTEN",
- "SYN_SENT",
- "SYN_RCVD",
- "ESTABLISHED",
- "FIN_WAIT_1",
- "FIN_WAIT_2",
- "CLOSING",
- "TIME_WAIT",
- "CLOSE_WAIT",
- "LAST_ACK",
- "INVALID" };
-
-
- void tcp_trans ( struct tcp_connection *conn, int nxt_state ) {
-
- conn->tcp_lstate = conn->tcp_state;
- conn->tcp_state = nxt_state;
-
-
- if ( conn->tcp_lstate == conn->tcp_state ||
- conn->tcp_state == TCP_INVALID ) {
- conn->tcp_flags = 0;
- return;
- }
-
-
- switch ( conn->tcp_state ) {
- case TCP_CLOSED:
- if ( conn->tcp_lstate == TCP_SYN_RCVD ) {
- conn->tcp_flags |= TCP_RST;
- }
- break;
- case TCP_LISTEN:
- break;
- case TCP_SYN_SENT:
- if ( conn->tcp_lstate == TCP_LISTEN ||
- conn->tcp_lstate == TCP_CLOSED ) {
- conn->tcp_flags |= TCP_SYN;
- }
- break;
- case TCP_SYN_RCVD:
- if ( conn->tcp_lstate == TCP_LISTEN ||
- conn->tcp_lstate == TCP_SYN_SENT ) {
- conn->tcp_flags |= ( TCP_SYN | TCP_ACK );
- }
- break;
- case TCP_ESTABLISHED:
- if ( conn->tcp_lstate == TCP_SYN_SENT ) {
- conn->tcp_flags |= TCP_ACK;
- }
- break;
- case TCP_FIN_WAIT_1:
- if ( conn->tcp_lstate == TCP_SYN_RCVD ||
- conn->tcp_lstate == TCP_ESTABLISHED ) {
- conn->tcp_flags |= TCP_FIN;
- }
- break;
- case TCP_FIN_WAIT_2:
- break;
- case TCP_CLOSING:
- if ( conn->tcp_lstate == TCP_FIN_WAIT_1 ) {
- conn->tcp_flags |= TCP_ACK;
- }
- break;
- case TCP_TIME_WAIT:
- if ( conn->tcp_lstate == TCP_FIN_WAIT_1 ||
- conn->tcp_lstate == TCP_FIN_WAIT_2 ) {
- conn->tcp_flags |= TCP_ACK;
- }
- break;
- case TCP_CLOSE_WAIT:
- if ( conn->tcp_lstate == TCP_ESTABLISHED ) {
- conn->tcp_flags |= TCP_ACK;
- }
- break;
- case TCP_LAST_ACK:
- if ( conn->tcp_lstate == TCP_CLOSE_WAIT ) {
- conn->tcp_flags |= TCP_FIN;
- }
- break;
- default:
- DBG ( "TCP_INVALID state %d\n", conn->tcp_state );
- return;
- }
- }
-
-
- void tcp_dump ( struct tcp_header *tcphdr ) {
- DBG ( "TCP header at %p+%d\n", tcphdr, sizeof ( *tcphdr ) );
- DBG ( "\tSource port = %d, Destination port = %d\n",
- ntohs ( tcphdr->src ), ntohs ( tcphdr->dest ) );
- DBG ( "\tSequence Number = %ld, Acknowledgement Number = %ld\n",
- ntohl ( tcphdr->seq ), ntohl ( tcphdr->ack ) );
- DBG ( "\tHeader length (/4) = %hd, Flags [..RAPUSF]= %#x\n",
- ( ( tcphdr->hlen & TCP_MASK_HLEN ) / 16 ),
- ( tcphdr->flags & TCP_MASK_FLAGS ) );
- DBG ( "\tAdvertised window = %ld, Checksum = %x, Urgent Pointer = %d\n",
- ntohs ( tcphdr->win ), tcphdr->csum, ntohs ( tcphdr->urg ) );
- }
-
-
- void tcp_init_conn ( struct tcp_connection *conn ) {
- conn->local_port = 0;
- conn->tcp_state = TCP_CLOSED;
- conn->tcp_lstate = TCP_INVALID;
- conn->tx_pkb = NULL;
- conn->tcp_op = NULL;
- }
-
-
- int tcp_connectto ( struct tcp_connection *conn,
- struct sockaddr_tcpip *peer ) {
- int rc;
-
-
- if ( conn->tcp_state != TCP_CLOSED ) {
- DBG ( "Error opening connection: Invalid state %s\n",
- tcp_states[conn->tcp_state] );
- return -EISCONN;
- }
-
-
- if ( ( rc = tcp_listen ( conn, conn->local_port ) ) != 0 ) {
- return rc;
- }
- memcpy ( &conn->peer, peer, sizeof ( conn->peer ) );
-
-
- conn->snd_una = ( ( ( uint32_t ) random() ) << 16 ) & random();
- tcp_trans ( conn, TCP_SYN_SENT );
-
- free_pkb ( conn->tx_pkb );
- conn->tx_pkb = alloc_pkb ( MIN_PKB_LEN );
- pkb_reserve ( conn->tx_pkb, MAX_HDR_LEN );
- conn->rcv_win = MAX_PKB_LEN - MAX_HDR_LEN;
- return tcp_send ( conn, TCP_NOMSG, TCP_NOMSG_LEN );
- }
-
- int tcp_connect ( struct tcp_connection *conn ) {
- return tcp_connectto ( conn, &conn->peer );
- }
-
-
- int tcp_close ( struct tcp_connection *conn ) {
-
- switch ( conn->tcp_state ) {
- case TCP_SYN_RCVD:
- case TCP_ESTABLISHED:
- tcp_trans ( conn, TCP_FIN_WAIT_1 );
- conn->tcp_op->closed ( conn, CONN_SNDCLOSE );
-
-
- goto send_tcp_nomsg;
- case TCP_SYN_SENT:
- case TCP_LISTEN:
-
-
- list_del ( &conn->list );
- tcp_trans ( conn, TCP_CLOSED );
- conn->tcp_op->closed ( conn, CONN_SNDCLOSE );
- return 0;
- case TCP_CLOSE_WAIT:
- tcp_trans ( conn, TCP_LAST_ACK );
- conn->tcp_op->closed ( conn, CONN_SNDCLOSE );
-
-
- goto send_tcp_nomsg;
- default:
- DBG ( "tcp_close(): Invalid state %s\n",
- tcp_states[conn->tcp_state] );
- return -EPROTO;
- }
-
- send_tcp_nomsg:
- free_pkb ( conn->tx_pkb );
- conn->tx_pkb = alloc_pkb ( MIN_PKB_LEN );
- conn->tcp_flags = TCP_FIN;
- pkb_reserve ( conn->tx_pkb, MAX_HDR_LEN );
- return tcp_send ( conn, TCP_NOMSG, TCP_NOMSG_LEN );
- }
-
-
-
- int tcp_listen ( struct tcp_connection *conn, uint16_t port ) {
- struct tcp_connection *cconn;
- if ( port != 0 ) {
- list_for_each_entry ( cconn, &tcp_conns, list ) {
- if ( cconn->local_port == port ) {
- DBG ( "Error listening to %d\n",
- ntohs ( port ) );
- return -EISCONN;
- }
- }
-
- conn->local_port = port;
- list_add ( &conn->list, &tcp_conns );
- return 0;
- }
-
- DBG ( "Assigning lowest port not implemented\n");
- return -ENOSYS;
- }
-
-
- int tcp_senddata ( struct tcp_connection *conn ) {
-
- switch ( conn->tcp_state ) {
- case TCP_LISTEN:
- tcp_trans ( conn, TCP_SYN_SENT );
- conn->snd_una = ( ( ( uint32_t ) random() ) << 16 ) & random();
- break;
- case TCP_ESTABLISHED:
- case TCP_CLOSE_WAIT:
- break;
- default:
- DBG ( "tcp_senddata: Invalid state %s\n",
- tcp_states[conn->tcp_state] );
- return -EPROTO;
- }
-
-
- free_pkb ( conn->tx_pkb );
- conn->tx_pkb = alloc_pkb ( MAX_PKB_LEN );
- if ( !conn->tx_pkb ) {
- DBG ( "Insufficient memory\n" );
- return -ENOMEM;
- }
- pkb_reserve ( conn->tx_pkb, MAX_HDR_LEN );
-
- conn->rcv_win = pkb_available ( conn->tx_pkb );
-
- conn->tcp_op->senddata ( conn, conn->tx_pkb->data,
- pkb_available ( conn->tx_pkb ) );
- return 0;
- }
-
-
- int tcp_send ( struct tcp_connection *conn, const void *data, size_t len ) {
- struct sockaddr_tcpip *peer = &conn->peer;
- struct pk_buff *pkb = conn->tx_pkb;
- int slen;
-
-
- slen = len < conn->snd_win ? len : conn->snd_win;
-
- memmove ( pkb_put ( pkb, slen ), data, slen );
-
-
- struct tcp_header *tcphdr = pkb_push ( pkb, sizeof ( *tcphdr ) );
-
-
- tcphdr->src = conn->local_port;
-
- tcphdr->dest = peer->st_port;
- tcphdr->seq = htonl ( conn->snd_una );
- tcphdr->ack = htonl ( conn->rcv_nxt );
-
- tcphdr->hlen = ( uint8_t ) ( ( sizeof ( *tcphdr ) / 4 ) << 4 );
-
- tcphdr->flags = conn->tcp_flags;
- conn->tcp_flags = 0;
-
- tcphdr->win = htons ( conn->rcv_win );
-
- tcphdr->urg = 0;
-
- tcphdr->csum = 0;
- tcphdr->csum = tcpip_chksum ( pkb->data, pkb_len ( pkb ) );
-
-
- tcp_dump ( tcphdr );
-
-
- return tcpip_tx ( pkb, &tcp_protocol, peer );
- }
-
-
- static int tcp_rx ( struct pk_buff *pkb,
- struct sockaddr_tcpip *st_src __unused,
- struct sockaddr_tcpip *st_dest __unused ) {
- struct tcp_connection *conn;
- struct tcp_header *tcphdr;
- uint32_t acked, toack;
- int hlen;
-
-
- if ( pkb_len ( pkb ) < sizeof ( *tcphdr ) ) {
- DBG ( "Packet too short (%d bytes)\n", pkb_len ( pkb ) );
- return -EINVAL;
- }
-
-
- tcphdr = pkb->data;
-
-
- hlen = ( ( tcphdr->hlen & TCP_MASK_HLEN ) / 16 ) * 4;
- if ( hlen != sizeof ( *tcphdr ) ) {
- DBG ( "Bad header length (%d bytes)\n", hlen );
- return -EINVAL;
- }
-
-
-
-
- list_for_each_entry ( conn, &tcp_conns, list ) {
- if ( tcphdr->dest == conn->local_port ) {
- goto found_conn;
- }
- }
-
- DBG ( "No connection found on port %d\n", ntohs ( tcphdr->dest ) );
- return 0;
-
- found_conn:
-
- conn->snd_win = tcphdr->win;
-
-
- uint8_t out_flags = 0;
- conn->tcp_lstate = conn->tcp_state;
- switch ( conn->tcp_state ) {
- case TCP_CLOSED:
- DBG ( "tcp_rx(): Invalid state %s\n",
- tcp_states[conn->tcp_state] );
- return -EINVAL;
- case TCP_LISTEN:
- if ( tcphdr->flags & TCP_SYN ) {
- tcp_trans ( conn, TCP_SYN_RCVD );
-
- conn->rcv_nxt = ntohl ( tcphdr->seq ) + 1;
- out_flags |= TCP_ACK;
-
-
- conn->snd_una = ( ( ( uint32_t ) random() ) << 16 );
- conn->snd_una &= random();
- out_flags |= TCP_SYN;
-
-
- goto send_tcp_nomsg;
- }
-
- goto unexpected;
- case TCP_SYN_SENT:
- if ( tcphdr->flags & TCP_SYN ) {
-
- conn->rcv_nxt = ntohl ( tcphdr->seq ) + 1;
- out_flags |= TCP_ACK;
-
- if ( tcphdr->flags & TCP_ACK ) {
- tcp_trans ( conn, TCP_ESTABLISHED );
-
-
- conn->snd_una = ntohl ( tcphdr->ack );
- conn->tcp_op->connected ( conn );
- } else {
- tcp_trans ( conn, TCP_SYN_RCVD );
- out_flags |= TCP_SYN;
- }
-
- goto send_tcp_nomsg;
- }
-
- goto unexpected;
- case TCP_SYN_RCVD:
- if ( tcphdr->flags & TCP_RST ) {
- tcp_trans ( conn, TCP_LISTEN );
- conn->tcp_op->closed ( conn, CONN_RESTART );
- return 0;
- }
- if ( tcphdr->flags & TCP_ACK ) {
- tcp_trans ( conn, TCP_ESTABLISHED );
-
-
- conn->snd_una = tcphdr->ack - 1;
- conn->tcp_op->connected ( conn );
- return 0;
- }
-
- goto unexpected;
- case TCP_ESTABLISHED:
- if ( tcphdr->flags & TCP_FIN ) {
- tcp_trans ( conn, TCP_CLOSE_WAIT );
-
- conn->rcv_nxt++;
- out_flags |= TCP_ACK;
-
- goto send_tcp_nomsg;
- }
-
- break;
- case TCP_FIN_WAIT_1:
- if ( tcphdr->flags & TCP_FIN ) {
- conn->rcv_nxt++;
- out_flags |= TCP_ACK;
- conn->tcp_op->closed ( conn, CONN_SNDCLOSE );
-
- if ( tcphdr->flags & TCP_ACK ) {
- tcp_trans ( conn, TCP_TIME_WAIT );
- } else {
- tcp_trans ( conn, TCP_CLOSING );
- }
-
- goto send_tcp_nomsg;
- }
- if ( tcphdr->flags & TCP_ACK ) {
- tcp_trans ( conn, TCP_FIN_WAIT_2 );
- }
-
- break;
- case TCP_FIN_WAIT_2:
- if ( tcphdr->flags & TCP_FIN ) {
- tcp_trans ( conn, TCP_TIME_WAIT );
-
- conn->rcv_nxt++;
- out_flags |= TCP_ACK;
- goto send_tcp_nomsg;
- }
-
- break;
- case TCP_CLOSING:
- if ( tcphdr->flags & TCP_ACK ) {
- tcp_trans ( conn, TCP_TIME_WAIT );
- return 0;
- }
-
- goto unexpected;
- case TCP_TIME_WAIT:
-
- goto unexpected;
- case TCP_CLOSE_WAIT:
-
- break;
- case TCP_LAST_ACK:
- if ( tcphdr->flags & TCP_ACK ) {
- tcp_trans ( conn, TCP_CLOSED );
- return 0;
- }
-
- goto unexpected;
- }
-
-
-
- assert ( ( tcphdr->flags & TCP_ACK ) ||
- pkb_len ( pkb ) > sizeof ( *tcphdr ) );
-
-
- toack = pkb_len ( pkb ) - hlen;
- if ( toack > 0 ) {
-
- if ( conn->rcv_nxt == ntohl ( tcphdr->seq ) ) {
- conn->rcv_nxt += toack;
- conn->tcp_op->newdata ( conn, pkb->data + sizeof ( *tcphdr ), toack );
- }
-
-
- out_flags |= TCP_ACK;
- if ( !( tcphdr->flags & TCP_ACK ) ) {
- goto send_tcp_nomsg;
- }
- }
-
-
- if ( tcphdr->flags & TCP_ACK ) {
- acked = ntohl ( tcphdr->ack ) - conn->snd_una;
- if ( acked < 0 ) {
- DBG ( "Previously ACKed (%d)\n", tcphdr->ack );
- return 0;
- }
-
- conn->snd_una += acked;
-
- conn->tcp_flags |= TCP_ACK;
-
- conn->tcp_op->acked ( conn, acked );
-
- tcp_senddata ( conn );
- }
- return 0;
-
- send_tcp_nomsg:
- free_pkb ( conn->tx_pkb );
- conn->tx_pkb = alloc_pkb ( MIN_PKB_LEN );
- pkb_reserve ( conn->tx_pkb, MAX_HDR_LEN );
- int rc;
- if ( ( rc = tcp_send ( conn, TCP_NOMSG, TCP_NOMSG_LEN ) ) != 0 ) {
- DBG ( "Error sending TCP message (rc = %d)\n", rc );
- }
- return 0;
-
- unexpected:
- DBG ( "Unexpected packet received in %d state with flags = %hd\n",
- conn->tcp_state, tcphdr->flags & TCP_MASK_FLAGS );
- free_pkb ( conn->tx_pkb );
- return -EINVAL;
- }
-
-
- struct tcpip_protocol tcp_protocol = {
- .name = "TCP",
- .rx = tcp_rx,
- .tcpip_proto = IP_TCP,
- .csum_offset = 16,
- };
-
- TCPIP_PROTOCOL ( tcp_protocol );
-
- #endif
|