Ignore:
File:
1 edited

Legend:

Unmodified
Added
Removed
  • uspace/srv/net/tcp/pdu.c

    r4a0bc99 r12df1f1  
    11/*
    2  * Copyright (c) 2015 Jiri Svoboda
     2 * Copyright (c) 2011 Jiri Svoboda
    33 * All rights reserved.
    44 *
     
    3838#include <byteorder.h>
    3939#include <errno.h>
    40 #include <inet/endpoint.h>
    4140#include <mem.h>
    4241#include <stdlib.h>
     42#include <net/socket_codes.h>
    4343#include "pdu.h"
    4444#include "segment.h"
     
    126126}
    127127
    128 static void tcp_header_setup(inet_ep2_t *epp, tcp_segment_t *seg, tcp_header_t *hdr)
     128static void tcp_header_setup(tcp_sockpair_t *sp, tcp_segment_t *seg, tcp_header_t *hdr)
    129129{
    130130        uint16_t doff_flags;
    131131        uint16_t doff;
    132132
    133         hdr->src_port = host2uint16_t_be(epp->local.port);
    134         hdr->dest_port = host2uint16_t_be(epp->remote.port);
     133        hdr->src_port = host2uint16_t_be(sp->local.port);
     134        hdr->dest_port = host2uint16_t_be(sp->foreign.port);
    135135        hdr->seq = host2uint32_t_be(seg->seq);
    136136        hdr->ack = host2uint32_t_be(seg->ack);
     
    145145}
    146146
    147 static ip_ver_t tcp_phdr_setup(tcp_pdu_t *pdu, tcp_phdr_t *phdr,
     147static uint16_t tcp_phdr_setup(tcp_pdu_t *pdu, tcp_phdr_t *phdr,
    148148    tcp_phdr6_t *phdr6)
    149149{
    150150        addr32_t src_v4;
    151151        addr128_t src_v6;
    152         uint16_t src_ver = inet_addr_get(&pdu->src, &src_v4, &src_v6);
    153 
     152        uint16_t src_af = inet_addr_get(&pdu->src, &src_v4, &src_v6);
     153       
    154154        addr32_t dest_v4;
    155155        addr128_t dest_v6;
    156         uint16_t dest_ver = inet_addr_get(&pdu->dest, &dest_v4, &dest_v6);
    157 
    158         assert(src_ver == dest_ver);
    159 
    160         switch (src_ver) {
    161         case ip_v4:
     156        uint16_t dest_af = inet_addr_get(&pdu->dest, &dest_v4, &dest_v6);
     157       
     158        assert(src_af == dest_af);
     159       
     160        switch (src_af) {
     161        case AF_INET:
    162162                phdr->src = host2uint32_t_be(src_v4);
    163163                phdr->dest = host2uint32_t_be(dest_v4);
     
    167167                    host2uint16_t_be(pdu->header_size + pdu->text_size);
    168168                break;
    169         case ip_v6:
     169        case AF_INET6:
    170170                host2addr128_t_be(src_v6, phdr6->src);
    171171                host2addr128_t_be(dest_v6, phdr6->dest);
     
    178178                assert(false);
    179179        }
    180 
    181         return src_ver;
     180       
     181        return src_af;
    182182}
    183183
     
    191191}
    192192
    193 static int tcp_header_encode(inet_ep2_t *epp, tcp_segment_t *seg,
     193static int tcp_header_encode(tcp_sockpair_t *sp, tcp_segment_t *seg,
    194194    void **header, size_t *size)
    195195{
     
    200200                return ENOMEM;
    201201
    202         tcp_header_setup(epp, seg, hdr);
     202        tcp_header_setup(sp, seg, hdr);
    203203        *header = hdr;
    204204        *size = sizeof(tcp_header_t);
     
    249249        if (pdu->text != NULL)
    250250                free(pdu->text);
    251         free(pdu);
    252251
    253252        return NULL;
     
    267266        tcp_phdr_t phdr;
    268267        tcp_phdr6_t phdr6;
    269 
    270         ip_ver_t ver = tcp_phdr_setup(pdu, &phdr, &phdr6);
    271         switch (ver) {
    272         case ip_v4:
     268       
     269        uint16_t af = tcp_phdr_setup(pdu, &phdr, &phdr6);
     270        switch (af) {
     271        case AF_INET:
    273272                cs_phdr = tcp_checksum_calc(TCP_CHECKSUM_INIT, (void *) &phdr,
    274273                    sizeof(tcp_phdr_t));
    275274                break;
    276         case ip_v6:
     275        case AF_INET6:
    277276                cs_phdr = tcp_checksum_calc(TCP_CHECKSUM_INIT, (void *) &phdr6,
    278277                    sizeof(tcp_phdr6_t));
     
    281280                assert(false);
    282281        }
    283 
     282       
    284283        cs_headers = tcp_checksum_calc(cs_phdr, pdu->header, pdu->header_size);
    285284        return tcp_checksum_calc(cs_headers, pdu->text, pdu->text_size);
     
    295294
    296295/** Decode incoming PDU */
    297 int tcp_pdu_decode(tcp_pdu_t *pdu, inet_ep2_t *epp, tcp_segment_t **seg)
     296int tcp_pdu_decode(tcp_pdu_t *pdu, tcp_sockpair_t *sp, tcp_segment_t **seg)
    298297{
    299298        tcp_segment_t *nseg;
     
    309308        hdr = (tcp_header_t *)pdu->header;
    310309
    311         epp->local.port = uint16_t_be2host(hdr->dest_port);
    312         epp->local.addr = pdu->dest;
    313         epp->remote.port = uint16_t_be2host(hdr->src_port);
    314         epp->remote.addr = pdu->src;
     310        sp->local.port = uint16_t_be2host(hdr->dest_port);
     311        sp->local.addr = pdu->dest;
     312        sp->foreign.port = uint16_t_be2host(hdr->src_port);
     313        sp->foreign.addr = pdu->src;
    315314
    316315        *seg = nseg;
     
    319318
    320319/** Encode outgoing PDU */
    321 int tcp_pdu_encode(inet_ep2_t *epp, tcp_segment_t *seg, tcp_pdu_t **pdu)
     320int tcp_pdu_encode(tcp_sockpair_t *sp, tcp_segment_t *seg, tcp_pdu_t **pdu)
    322321{
    323322        tcp_pdu_t *npdu;
    324323        size_t text_size;
    325324        uint16_t checksum;
    326         int rc;
    327325
    328326        npdu = tcp_pdu_new();
     
    330328                return ENOMEM;
    331329
    332         npdu->src = epp->local.addr;
    333         npdu->dest = epp->remote.addr;
    334         rc = tcp_header_encode(epp, seg, &npdu->header, &npdu->header_size);
    335         if (rc != EOK) {
    336                 free(npdu);
    337                 return rc;
    338         }
     330        npdu->src = sp->local.addr;
     331        npdu->dest = sp->foreign.addr;
     332        tcp_header_encode(sp, seg, &npdu->header, &npdu->header_size);
    339333
    340334        text_size = tcp_segment_text_size(seg);
    341335        npdu->text = calloc(1, text_size);
    342         if (npdu->text == NULL) {
    343                 free(npdu->header);
    344                 free(npdu);
     336        if (npdu->text == NULL)
    345337                return ENOMEM;
    346         }
    347338
    348339        npdu->text_size = text_size;
Note: See TracChangeset for help on using the changeset viewer.