Blame view

RIOT/sys/net/gnrc/conn/gnrc_conn.c 3.03 KB
fb11e647   vrobic   reseau statique a...
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
  /*
   * Copyright (C) 2015 Martine Lenders <mlenders@inf.fu-berlin.de>
   *
   * This file is subject to the terms and conditions of the GNU Lesser
   * General Public License v2.1. See the file LICENSE in the top level
   * directory for more details.
   */
  
  /**
   * @{
   *
   * @file
   * @author  Martine Lenders <mlenders@inf.fu-berlin.de>
   * @author  Oliver Hahm <oliver.hahm@inria.fr>
   */
  
  #include "net/conn.h"
  #include "net/ipv6/hdr.h"
  #include "net/gnrc/conn.h"
  #include "net/gnrc/ipv6/hdr.h"
  #include "net/gnrc/ipv6/netif.h"
  #include "net/udp.h"
  
  int gnrc_conn_recvfrom(conn_t *conn, void *data, size_t max_len, void *addr, size_t *addr_len,
                         uint16_t *port)
  {
      msg_t msg;
      int timeout = 3;
      while ((timeout--) > 0) {
          gnrc_pktsnip_t *pkt, *l3hdr;
          size_t size = 0;
          msg_receive(&msg);
          switch (msg.type) {
              case GNRC_NETAPI_MSG_TYPE_RCV:
                  pkt = msg.content.ptr;
                  if (pkt->size > max_len) {
                      return -ENOMEM;
                  }
                  l3hdr = gnrc_pktsnip_search_type(pkt, conn->l3_type);
                  if (l3hdr == NULL) {
                      msg_send_to_self(&msg); /* requeue invalid messages */
                      continue;
                  }
  #if defined(MODULE_CONN_UDP) || defined(MODULE_CONN_TCP)
                  if ((conn->l4_type != GNRC_NETTYPE_UNDEF) && (port != NULL)) {
                      gnrc_pktsnip_t *l4hdr;
                      l4hdr = gnrc_pktsnip_search_type(pkt, conn->l4_type);
                      if (l4hdr == NULL) {
                          msg_send_to_self(&msg); /* requeue invalid messages */
                          continue;
                      }
                      *port = byteorder_ntohs(((udp_hdr_t *)l4hdr->data)->src_port);
                  }
  #endif  /* defined(MODULE_CONN_UDP) */
                  if (addr != NULL) {
                      memcpy(addr, &((ipv6_hdr_t *)l3hdr->data)->src, sizeof(ipv6_addr_t));
                      *addr_len = sizeof(ipv6_addr_t);
                  }
                  memcpy(data, pkt->data, pkt->size);
                  size = pkt->size;
                  gnrc_pktbuf_release(pkt);
                  return (int)size;
              default:
                  (void)port;
                  msg_send_to_self(&msg); /* requeue invalid messages */
                  break;
          }
      }
      return -ETIMEDOUT;
  }
  
  #ifdef MODULE_GNRC_IPV6
  bool gnrc_conn6_set_local_addr(uint8_t *conn_addr, const ipv6_addr_t *addr)
  {
      ipv6_addr_t *tmp;
      if (!ipv6_addr_is_unspecified(addr) &&
          !ipv6_addr_is_loopback(addr) &&
          gnrc_ipv6_netif_find_by_addr(&tmp, addr) == KERNEL_PID_UNDEF) {
          return false;
      }
      else if (ipv6_addr_is_loopback(addr) || ipv6_addr_is_unspecified(addr)) {
          ipv6_addr_set_unspecified((ipv6_addr_t *)conn_addr);
      }
      else {
          memcpy(conn_addr, addr, sizeof(ipv6_addr_t));
      }
      return true;
  }
  
  ipv6_addr_t *conn_find_best_source(const ipv6_addr_t *dst)
  {
      ipv6_addr_t *local = NULL;
      gnrc_ipv6_netif_find_by_prefix(&local, dst);
      return local;
  }
  #endif
  
  /** @} */