123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277 |
- #include <stdint.h>
- #include <string.h>
- #include <assert.h>
- #include <byteswap.h>
- #include <latch.h>
- #include <errno.h>
- #include <gpxe/in.h>
- #include <gpxe/ip.h>
- #include <gpxe/udp.h>
- #include <gpxe/init.h>
- #include <gpxe/pkbuff.h>
- #include <gpxe/netdevice.h>
- #include <gpxe/interface.h>
-
- /** @file
- *
- * UDP protocol
- */
-
- static inline void copy_sockaddr ( struct sockaddr *source, struct sockaddr *dest ) {
- memcpy ( dest, source, sizeof ( *dest ) );
- }
-
- static inline uint16_t dest_port ( struct sockaddr *sock, uint16_t *dest ) {
- switch ( sock->sa_family ) {
- case AF_INET:
- dest = &sock->sin.sin_port;
- break;
- case AF_INET6:
- dest = &sock->sin6.sin6_port;
- break;
- default:
- DBG ( "Network family %d not supported\n", sock->sa_family );
- return -EAFNOSUPPORT;
- }
- return 0;
- }
-
- /**
- * Dump the UDP header
- *
- * @v udphdr UDP header
- */
- void udp_dump ( struct udp_header *udphdr ) {
-
- /* Print UDP header for debugging */
- DBG ( "UDP header at %#x + %d\n", udphdr, sizeof ( *udphdr ) );
- DBG ( "\tSource Port = %d\n", ntohs ( udphdr->source_port ) );
- DBG ( "\tDestination Port = %d\n", ntohs ( udphdr->dest_port ) );
- DBG ( "\tLength = %d\n", ntohs ( udphdr->len ) );
- DBG ( "\tChecksum = %d\n", ntohs ( udphdr->chksum ) );
- DBG ( "\tChecksum located at %#x\n", &udphdr->chksum );
- }
-
- /**
- * Open a UDP connection
- *
- * @v conn UDP connection
- * @v peer Destination socket address
- *
- * This function stores the socket address within the connection
- */
- void udp_connect ( struct udp_connection *conn, struct sockaddr *peer ) {
- copy_sockaddr ( peer, &conn->sin );
-
- /* Not sure if this should add the connection to udp_conns; If it does,
- * uncomment the following code
- */
- // list_add ( &conn->list, &udp_conns );
- }
-
- /**
- * Initialize a UDP connection
- *
- * @v conn UDP connection
- * @v udp_op UDP operations
- */
- void udp_init ( struct udp_connection *conn, struct udp_operations *udp_op ) {
- conn->local_port = 0;
- conn->tx_pkb = NULL;
- if ( udp_op != NULL ) {
- conn->udp_op = udp_op;
- }
- }
-
- /**
- * Allocate space to the UDP buffer
- *
- * @v conn UDP connection
- * @v len Length to allocate
- * @ret rc Status
- *
- * Allocate "len" amount of space in the transmit buffer
- */
- int udp_buf_alloc ( struct udp_connection *conn, size_t len ) {
- if ( conn->tx_pkb != NULL ) {
- free_pkb ( conn->tx_pkb );
- conn->tx_pkb = NULL;
- }
- conn->tx_pkb = alloc_pkb ( len < UDP_MIN_TXPKB ? UDP_MIN_TXPKB : len );
- return !conn ? -ENOMEM : 0;
- }
-
- /**
- * Send data via a UDP connection
- *
- * @v conn UDP connection
- * @v data Data to send
- * @v len Length of data
- *
- * This function fills up the UDP headers and sends the data. Discover the
- * network protocol through the sa_family field in the destination socket
- * address.
- */
- int udp_send ( struct udp_connection *conn, const void *data, size_t len ) {
- struct udp_header *udphdr; /* UDP header */
- struct sockaddr *sock = &conn->sin; /* Destination sockaddr */
- uint16_t *dest;
- int rc;
-
- /* Copy data into packet buffer, if necessary */
- if ( data != conn->tx_pkb->data ) {
- /* Allocate space for data and lower layer headers */
- if ( ( rc = udp_buf_alloc ( conn, len + UDP_MAX_HLEN ) ) != 0 ) {
- DBG ( "Error allocating buffer" );
- return rc;
- }
-
- /* Reserve space for the headers and copy contents */
- pkb_reserve ( conn->tx_pkb, UDP_MAX_HLEN );
- memcpy ( pkb_put ( conn->tx_pkb, len ), data, len );
- }
-
- /*
- * Add the UDP header
- *
- * Covert all 16- and 32- bit integers into network btye order before
- * sending it over the network
- */
- udphdr = pkb_push ( conn->tx_pkb, sizeof ( *udphdr ) );
- if ( (rc = dest_port ( sock, dest ) ) != 0 ) {
- return rc;
- }
- udphdr->dest_port = htons ( *dest );
- udphdr->source_port = htons ( conn->local_port );
- udphdr->len = htons ( pkb_len ( conn->tx_pkb ) );
- udphdr->chksum = htons ( calc_chksum ( udphdr, sizeof ( *udphdr ) ) );
-
- udp_dump ( udphdr );
-
- /* Send it to the next layer for processing */
- return trans_tx ( conn->tx_pkb, IP_UDP, sock );
- }
-
- /**
- * Send data to a specified address
- *
- * @v conn UDP connection
- * @v peer Destination address
- * @v data Data to send
- * @v len Length of data
- */
- int udp_sendto ( struct udp_connection *conn, struct sockaddr *peer,
- const void *data, size_t len ) {
- struct sockaddr tempsock;
- copy_sockaddr ( &conn->sin, &tempsock );
- copy_sockaddr ( peer, &conn->sin );
- int rc = udp_send ( conn, data, len );
- copy_sockaddr ( &tempsock, &conn->sin );
- return rc;
- }
-
- /**
- * Close a UDP connection
- *
- * @v conn UDP connection
- */
- void udp_close ( struct udp_connection *conn ) {
- list_del ( &conn->list );
- }
-
- /**
- * Open a local port
- *
- * @v conn UDP connection
- * @v local_port Local port on which to open connection
- *
- * This does not support the 0 port option correctly yet
- */
- int udp_open ( struct udp_connection *conn, uint16_t local_port ) {
- struct udp_connection *connr;
- uint16_t min_port = 0xffff;
-
- /* Iterate through udp_conns to see if local_port is available */
- list_for_each_entry ( connr, &udp_conns, list ) {
- if ( connr->local_port == local_port ) {
- return -EISCONN;
- }
- if ( min_port > connr->local_port ) {
- min_port = connr->local_port;
- }
- }
- /* This code is buggy. I will update it soon :) */
- conn->local_port = local_port == 0 ? min_port > 1024 ? 1024 :
- min_port + 1 : local_port;
-
- /* Add the connection to the list of listening connections */
- list_add ( &conn->list, &udp_conns );
- return 0;
- }
-
- /**
- * Process a received packet
- *
- * @v pkb Packet buffer
- * @v src_net_addr Source network address
- * @v dest_net_addr Destination network address
- */
- void udp_rx ( struct pk_buff *pkb, struct in_addr *src_net_addr __unused,
- struct in_addr *dest_net_addr __unused ) {
- struct udp_header *udphdr = pkb->data;
- struct udp_connection *conn;
- uint16_t ulen;
- uint16_t chksum;
-
- udp_dump ( udphdr );
-
- /* Validate the packet and the UDP length */
- if ( pkb_len ( pkb ) < sizeof ( *udphdr ) ) {
- DBG ( "UDP packet too short (%d bytes)\n",
- pkb_len ( pkb ) );
- return;
- }
-
- ulen = ntohs ( udphdr->len );
- if ( ulen != pkb_len ( pkb ) ) {
- DBG ( "Inconsistent UDP packet length (%d bytes)\n",
- pkb_len ( pkb ) );
- return;
- }
-
- /* Verify the checksum */
- chksum = calc_chksum ( pkb->data, pkb_len ( pkb ) );
- if ( chksum != 0xffff ) {
- DBG ( "Bad checksum %d\n", chksum );
- return;
- }
-
- /* Todo: Check if it is a broadcast or multicast address */
-
- /* Demux the connection */
- list_for_each_entry ( conn, &udp_conns, list ) {
- if ( conn->local_port == ntohs ( udphdr->dest_port ) ) {
- goto conn;
- }
- }
- return;
-
- conn:
- /** Strip off the UDP header */
- pkb_pull ( pkb, sizeof ( *udphdr ) );
-
- /** Allocate max possible buffer space to the tx buffer */
- conn->tx_pkb = alloc_pkb ( UDP_MAX_TXPKB );
- pkb_reserve ( conn->tx_pkb, UDP_MAX_HLEN );
-
- /** Call the application's callback */
- conn->udp_op->newdata ( conn, pkb->data, ulen - sizeof ( *udphdr ) );
- }
-
- struct trans_protocol udp_protocol = {
- .name = "UDP",
- .rx = udp_rx,
- .trans_proto = IP_UDP,
- };
-
- TRANS_PROTOCOL ( udp_protocol );
|