tcp_helper.c File Reference

#include "tcp_helper.h"
#include "lwip/tcp_impl.h"
#include "lwip/stats.h"
#include "lwip/pbuf.h"
#include "lwip/inet_chksum.h"

Include dependency graph for tcp_helper.c:

Go to the source code of this file.

Functions
static void	tcp_remove (struct tcp_pcb *pcb_list)
void	tcp_remove_all (void)
static struct pbuf *	tcp_create_segment_wnd (ip_addr_t *src_ip, ip_addr_t *dst_ip, u16_t src_port, u16_t dst_port, void *data, size_t data_len, u32_t seqno, u32_t ackno, u8_t headerflags, u16_t wnd)
struct pbuf *	tcp_create_segment (ip_addr_t *src_ip, ip_addr_t *dst_ip, u16_t src_port, u16_t dst_port, void *data, size_t data_len, u32_t seqno, u32_t ackno, u8_t headerflags)
struct pbuf *	tcp_create_rx_segment (struct tcp_pcb *pcb, void *data, size_t data_len, u32_t seqno_offset, u32_t ackno_offset, u8_t headerflags)
struct pbuf *	tcp_create_rx_segment_wnd (struct tcp_pcb *pcb, void *data, size_t data_len, u32_t seqno_offset, u32_t ackno_offset, u8_t headerflags, u16_t wnd)
void	tcp_set_state (struct tcp_pcb *pcb, enum tcp_state state, ip_addr_t *local_ip, ip_addr_t *remote_ip, u16_t local_port, u16_t remote_port)
void	test_tcp_counters_err (void *arg, err_t err)
static void	test_tcp_counters_check_rxdata (struct test_tcp_counters *counters, struct pbuf *p)
err_t	test_tcp_counters_recv (void *arg, struct tcp_pcb *pcb, struct pbuf *p, err_t err)
struct tcp_pcb *	test_tcp_new_counters_pcb (struct test_tcp_counters *counters)
void	test_tcp_input (struct pbuf *p, struct netif *inp)
static err_t	test_tcp_netif_output (struct netif *netif, struct pbuf *p, ip_addr_t *ipaddr)
void	test_tcp_init_netif (struct netif *netif, struct test_tcp_txcounters *txcounters, ip_addr_t *ip_addr, ip_addr_t *netmask)

Function Documentation

tcp_create_rx_segment()

struct pbuf* tcp_create_rx_segment	(	struct tcp_pcb *	pcb,
		void *	data,
		size_t	data_len,
		u32_t	seqno_offset,
		u32_t	ackno_offset,
		u8_t	headerflags
	)

Create a TCP segment usable for passing to tcp_input

• IP-addresses, ports, seqno and ackno are taken from pcb
• seqno and ackno can be altered with an offset

Definition at line 118 of file tcp_helper.c.

{
  return tcp_create_segment(&pcb->remote_ip, &pcb->local_ip, pcb->remote_port, pcb->local_port,
    data, data_len, pcb->rcv_nxt + seqno_offset, pcb->lastack + ackno_offset, headerflags);
}

Referenced by START_TEST(), test_tcp_recv_ooseq_double_FINs(), and test_tcp_tx_full_window_lost().

tcp_create_rx_segment_wnd()

struct pbuf* tcp_create_rx_segment_wnd	(	struct tcp_pcb *	pcb,
		void *	data,
		size_t	data_len,
		u32_t	seqno_offset,
		u32_t	ackno_offset,
		u8_t	headerflags,
		u16_t	wnd
	)

Create a TCP segment usable for passing to tcp_input

• IP-addresses, ports, seqno and ackno are taken from pcb
• seqno and ackno can be altered with an offset
• TCP window can be adjusted

Definition at line 130 of file tcp_helper.c.

{
  return tcp_create_segment_wnd(&pcb->remote_ip, &pcb->local_ip, pcb->remote_port, pcb->local_port,
    data, data_len, pcb->rcv_nxt + seqno_offset, pcb->lastack + ackno_offset, headerflags, wnd);
}

Referenced by test_tcp_tx_full_window_lost().

tcp_create_segment()

struct pbuf* tcp_create_segment	(	ip_addr_t *	src_ip,
		ip_addr_t *	dst_ip,
		u16_t	src_port,
		u16_t	dst_port,
		void *	data,
		size_t	data_len,
		u32_t	seqno,
		u32_t	ackno,
		u8_t	headerflags
	)

Create a TCP segment usable for passing to tcp_input

Definition at line 105 of file tcp_helper.c.

{
  return tcp_create_segment_wnd(src_ip, dst_ip, src_port, dst_port, data,
    data_len, seqno, ackno, headerflags, TCP_WND);
}

Referenced by tcp_create_rx_segment().

tcp_create_segment_wnd()

static struct pbuf* tcp_create_segment_wnd ( ip_addr_t *  src_ip, ip_addr_t *  dst_ip, u16_t  src_port, u16_t  dst_port, void *  data, size_t  data_len, u32_t  seqno, u32_t  ackno, u8_t  headerflags, u16_t  wnd  )

static

Create a TCP segment usable for passing to tcp_input

Definition at line 41 of file tcp_helper.c.

{
  struct pbuf *p, *q;
  struct ip_hdr* iphdr;
  struct tcp_hdr* tcphdr;
  u16_t pbuf_len = (u16_t)(sizeof(struct ip_hdr) + sizeof(struct tcp_hdr) + data_len);

  p = pbuf_alloc(PBUF_RAW, pbuf_len, PBUF_POOL);
  EXPECT_RETNULL(p != NULL);
  /* first pbuf must be big enough to hold the headers */
  EXPECT_RETNULL(p->len >= (sizeof(struct ip_hdr) + sizeof(struct tcp_hdr)));
  if (data_len > 0) {
    /* first pbuf must be big enough to hold at least 1 data byte, too */
    EXPECT_RETNULL(p->len > (sizeof(struct ip_hdr) + sizeof(struct tcp_hdr)));
  }

  for(q = p; q != NULL; q = q->next) {
    memset(q->payload, 0, q->len);
  }

  iphdr = p->payload;
  /* fill IP header */
  iphdr->dest.addr = dst_ip->addr;
  iphdr->src.addr = src_ip->addr;
  IPH_VHL_SET(iphdr, 4, IP_HLEN / 4);
  IPH_TOS_SET(iphdr, 0);
  IPH_LEN_SET(iphdr, htons(p->tot_len));
  IPH_CHKSUM_SET(iphdr, inet_chksum(iphdr, IP_HLEN));

  /* let p point to TCP header */
  pbuf_header(p, -(s16_t)sizeof(struct ip_hdr));

  tcphdr = p->payload;
  tcphdr->src   = htons(src_port);
  tcphdr->dest  = htons(dst_port);
  tcphdr->seqno = htonl(seqno);
  tcphdr->ackno = htonl(ackno);
  TCPH_HDRLEN_SET(tcphdr, sizeof(struct tcp_hdr)/4);
  TCPH_FLAGS_SET(tcphdr, headerflags);
  tcphdr->wnd   = htons(wnd);

  if (data_len > 0) {
    /* let p point to TCP data */
    pbuf_header(p, -(s16_t)sizeof(struct tcp_hdr));
    /* copy data */
    pbuf_take(p, data, data_len);
    /* let p point to TCP header again */
    pbuf_header(p, sizeof(struct tcp_hdr));
  }

  /* calculate checksum */

  tcphdr->chksum = inet_chksum_pseudo(p,
          IP_PROTO_TCP, p->tot_len, src_ip, dst_ip);

  pbuf_header(p, sizeof(struct ip_hdr));

  return p;
}

Referenced by tcp_create_rx_segment_wnd(), and tcp_create_segment().

tcp_remove()

static void tcp_remove ( struct tcp_pcb *  pcb_list )

static

Remove all pcbs on the given list.

Definition at line 14 of file tcp_helper.c.

{
  struct tcp_pcb *pcb = pcb_list;
  struct tcp_pcb *pcb2;

  while(pcb != NULL) {
    pcb2 = pcb;
    pcb = pcb->next;
    tcp_abort(pcb2);
  }
}

Referenced by tcp_remove_all().

tcp_remove_all()

void tcp_remove_all

(

void

)

Remove all pcbs on listen-, active- and time-wait-list (bound- isn't exported).

Definition at line 28 of file tcp_helper.c.

{
  tcp_remove(tcp_listen_pcbs.pcbs);
  tcp_remove(tcp_active_pcbs);
  tcp_remove(tcp_tw_pcbs);
  fail_unless(lwip_stats.memp[MEMP_TCP_PCB].used == 0);
  fail_unless(lwip_stats.memp[MEMP_TCP_PCB_LISTEN].used == 0);
  fail_unless(lwip_stats.memp[MEMP_TCP_SEG].used == 0);
  fail_unless(lwip_stats.memp[MEMP_PBUF_POOL].used == 0);
}

Referenced by tcp_oos_setup(), tcp_oos_teardown(), tcp_setup(), and tcp_teardown().

tcp_set_state()

void tcp_set_state	(	struct tcp_pcb *	pcb,
		enum tcp_state	state,
		ip_addr_t *	local_ip,
		ip_addr_t *	remote_ip,
		u16_t	local_port,
		u16_t	remote_port
	)

Safely bring a tcp_pcb into the requested state

Definition at line 139 of file tcp_helper.c.

{
  /* @todo: are these all states? */
  /* @todo: remove from previous list */
  pcb->state = state;
  if (state == ESTABLISHED) {
    TCP_REG(&tcp_active_pcbs, pcb);
    pcb->local_ip.addr = local_ip->addr;
    pcb->local_port = local_port;
    pcb->remote_ip.addr = remote_ip->addr;
    pcb->remote_port = remote_port;
  } else if(state == LISTEN) {
    TCP_REG(&tcp_listen_pcbs.pcbs, pcb);
    pcb->local_ip.addr = local_ip->addr;
    pcb->local_port = local_port;
  } else if(state == TIME_WAIT) {
    TCP_REG(&tcp_tw_pcbs, pcb);
    pcb->local_ip.addr = local_ip->addr;
    pcb->local_port = local_port;
    pcb->remote_ip.addr = remote_ip->addr;
    pcb->remote_port = remote_port;
  } else {
    fail();
  }
}

test_tcp_counters_check_rxdata()

static void test_tcp_counters_check_rxdata ( struct test_tcp_counters *  counters, struct pbuf *  p  )

static

Definition at line 176 of file tcp_helper.c.

{
  struct pbuf* q;
  u32_t i, received;
  if(counters->expected_data == NULL) {
    /* no data to compare */
    return;
  }
  EXPECT_RET(counters->recved_bytes + p->tot_len <= counters->expected_data_len);
  received = counters->recved_bytes;
  for(q = p; q != NULL; q = q->next) {
    char *data = q->payload;
    for(i = 0; i < q->len; i++) {
      EXPECT_RET(data[i] == counters->expected_data[received]);
      received++;
    }
  }
  EXPECT(received == counters->recved_bytes + p->tot_len);
}

Referenced by test_tcp_counters_recv().

test_tcp_counters_err()

void test_tcp_counters_err	(	void *	arg,
		err_t	err
	)

Definition at line 167 of file tcp_helper.c.

{
  struct test_tcp_counters* counters = arg;
  EXPECT_RET(arg != NULL);
  counters->err_calls++;
  counters->last_err = err;
}

Referenced by test_tcp_new_counters_pcb().

test_tcp_counters_recv()

err_t test_tcp_counters_recv	(	void *	arg,
		struct tcp_pcb *	pcb,
		struct pbuf *	p,
		err_t	err
	)

Definition at line 197 of file tcp_helper.c.

{
  struct test_tcp_counters* counters = arg;
  EXPECT_RETX(arg != NULL, ERR_OK);
  EXPECT_RETX(pcb != NULL, ERR_OK);
  EXPECT_RETX(err == ERR_OK, ERR_OK);

  if (p != NULL) {
    if (counters->close_calls == 0) {
      counters->recv_calls++;
      test_tcp_counters_check_rxdata(counters, p);
      counters->recved_bytes += p->tot_len;
    } else {
      counters->recv_calls_after_close++;
      counters->recved_bytes_after_close += p->tot_len;
    }
    pbuf_free(p);
  } else {
    counters->close_calls++;
  }
  EXPECT(counters->recv_calls_after_close == 0 && counters->recved_bytes_after_close == 0);
  return ERR_OK;
}

Referenced by test_tcp_new_counters_pcb().

test_tcp_init_netif()

void test_tcp_init_netif	(	struct netif *	netif,
		struct test_tcp_txcounters *	txcounters,
		ip_addr_t *	ip_addr,
		ip_addr_t *	netmask
	)

Definition at line 283 of file tcp_helper.c.

{
  struct netif *n;
  memset(netif, 0, sizeof(struct netif));
  if (txcounters != NULL) {
    memset(txcounters, 0, sizeof(struct test_tcp_txcounters));
    netif->state = txcounters;
  }
  netif->output = test_tcp_netif_output;
  netif->flags |= NETIF_FLAG_UP;
  ip_addr_copy(netif->netmask, *netmask);
  ip_addr_copy(netif->ip_addr, *ip_addr);
  for (n = netif_list; n != NULL; n = n->next) {
    if (n == netif) {
      return;
    }
  }
  netif->next = NULL;
  netif_list = netif;
}

Referenced by START_TEST(), test_tcp_recv_ooseq_double_FINs(), and test_tcp_tx_full_window_lost().

test_tcp_input()

void test_tcp_input	(	struct pbuf *	p,
		struct netif *	inp
	)

Calls tcp_input() after adjusting current_iphdr_dest

Definition at line 238 of file tcp_helper.c.

{
  struct ip_hdr *iphdr = (struct ip_hdr*)p->payload;
  /* these lines are a hack, don't use them as an example :-) */
  ip_addr_copy(*ipX_current_dest_addr(), iphdr->dest);
  ip_addr_copy(*ipX_current_src_addr(), iphdr->src);
  ip_current_netif() = inp;
  ip_current_header() = iphdr;

  /* since adding IPv6, p->payload must point to tcp header, not ip header */
  pbuf_header(p, -(s16_t)sizeof(struct ip_hdr));

  tcp_input(p, inp);

  ipX_current_dest_addr()->addr = 0;
  ipX_current_src_addr()->addr = 0;
  ip_current_netif() = NULL;
  ip_current_header() = NULL;
}

Referenced by START_TEST(), test_tcp_recv_ooseq_double_FINs(), and test_tcp_tx_full_window_lost().

test_tcp_netif_output()

static err_t test_tcp_netif_output ( struct netif *  netif, struct pbuf *  p, ip_addr_t *  ipaddr  )

static

Definition at line 258 of file tcp_helper.c.

{
  struct test_tcp_txcounters *txcounters = (struct test_tcp_txcounters*)netif->state;
  LWIP_UNUSED_ARG(ipaddr);
  if (txcounters != NULL)
  {
    txcounters->num_tx_calls++;
    txcounters->num_tx_bytes += p->tot_len;
    if (txcounters->copy_tx_packets) {
      struct pbuf *p_copy = pbuf_alloc(PBUF_LINK, p->tot_len, PBUF_RAM);
      err_t err;
      EXPECT(p_copy != NULL);
      err = pbuf_copy(p_copy, p);
      EXPECT(err == ERR_OK);
      if (txcounters->tx_packets == NULL) {
        txcounters->tx_packets = p_copy;
      } else {
        pbuf_cat(txcounters->tx_packets, p_copy);
      }
    }
  }
  return ERR_OK;
}

Referenced by test_tcp_init_netif().

test_tcp_new_counters_pcb()

struct tcp_pcb* test_tcp_new_counters_pcb

(

struct test_tcp_counters *

counters

)

Allocate a pcb and set up the test_tcp_counters_* callbacks

Definition at line 223 of file tcp_helper.c.

{
  struct tcp_pcb* pcb = tcp_new();
  if (pcb != NULL) {
    /* set up args and callbacks */
    tcp_arg(pcb, counters);
    tcp_recv(pcb, test_tcp_counters_recv);
    tcp_err(pcb, test_tcp_counters_err);
    pcb->snd_wnd = TCP_WND;
    pcb->snd_wnd_max = TCP_WND;
  }
  return pcb;
}

Referenced by START_TEST(), test_tcp_recv_ooseq_double_FINs(), and test_tcp_tx_full_window_lost().