Daniele Lacamera / PicoTCP-Experimental_CDC_ECM_Branch

Free (GPLv2) TCP/IP stack developed by TASS Belgium

Fork of PicoTCP by Daniele Lacamera

modules/pico_icmp4.c@51:18637a3d071f, 2013-08-03 (annotated)

Committer:
daniele
Date:
Sat Aug 03 08:50:27 2013 +0000
Revision:
51:18637a3d071f
Branch for CDC-ECM: Work in progress

Who changed what in which revision?

UserRevisionLine numberNew contents of line
daniele 51:18637a3d071f 1 /*********************************************************************
daniele 51:18637a3d071f 2 PicoTCP. Copyright (c) 2012 TASS Belgium NV. Some rights reserved.
daniele 51:18637a3d071f 3 See LICENSE and COPYING for usage.
daniele 51:18637a3d071f 4
daniele 51:18637a3d071f 5 .
daniele 51:18637a3d071f 6
daniele 51:18637a3d071f 7 Authors: Daniele Lacamera
daniele 51:18637a3d071f 8 *********************************************************************/
daniele 51:18637a3d071f 9
daniele 51:18637a3d071f 10
daniele 51:18637a3d071f 11 #include "pico_icmp4.h"
daniele 51:18637a3d071f 12 #include "pico_config.h"
daniele 51:18637a3d071f 13 #include "pico_ipv4.h"
daniele 51:18637a3d071f 14 #include "pico_eth.h"
daniele 51:18637a3d071f 15 #include "pico_device.h"
daniele 51:18637a3d071f 16 #include "pico_stack.h"
daniele 51:18637a3d071f 17 #include "pico_tree.h"
daniele 51:18637a3d071f 18
daniele 51:18637a3d071f 19 /* Queues */
daniele 51:18637a3d071f 20 static struct pico_queue icmp_in = {};
daniele 51:18637a3d071f 21 static struct pico_queue icmp_out = {};
daniele 51:18637a3d071f 22
daniele 51:18637a3d071f 23
daniele 51:18637a3d071f 24 /* Functions */
daniele 51:18637a3d071f 25
daniele 51:18637a3d071f 26 static int pico_icmp4_checksum(struct pico_frame *f)
daniele 51:18637a3d071f 27 {
daniele 51:18637a3d071f 28 struct pico_icmp4_hdr *hdr = (struct pico_icmp4_hdr *) f->transport_hdr;
daniele 51:18637a3d071f 29 if (!hdr) {
daniele 51:18637a3d071f 30 pico_err = PICO_ERR_EINVAL;
daniele 51:18637a3d071f 31 return -1;
daniele 51:18637a3d071f 32 }
daniele 51:18637a3d071f 33 hdr->crc = 0;
daniele 51:18637a3d071f 34 hdr->crc = short_be(pico_checksum(hdr, f->transport_len));
daniele 51:18637a3d071f 35 return 0;
daniele 51:18637a3d071f 36 }
daniele 51:18637a3d071f 37
daniele 51:18637a3d071f 38 #ifdef PICO_SUPPORT_PING
daniele 51:18637a3d071f 39 static void ping_recv_reply(struct pico_frame *f);
daniele 51:18637a3d071f 40 #endif
daniele 51:18637a3d071f 41
daniele 51:18637a3d071f 42 static int pico_icmp4_process_in(struct pico_protocol *self, struct pico_frame *f)
daniele 51:18637a3d071f 43 {
daniele 51:18637a3d071f 44 struct pico_icmp4_hdr *hdr = (struct pico_icmp4_hdr *) f->transport_hdr;
daniele 51:18637a3d071f 45 if (hdr->type == PICO_ICMP_ECHO) {
daniele 51:18637a3d071f 46 hdr->type = PICO_ICMP_ECHOREPLY;
daniele 51:18637a3d071f 47 /* outgoing frames require a f->len without the ethernet header len */
daniele 51:18637a3d071f 48 if (f->dev->eth)
daniele 51:18637a3d071f 49 f->len -= PICO_SIZE_ETHHDR;
daniele 51:18637a3d071f 50 pico_icmp4_checksum(f);
daniele 51:18637a3d071f 51 pico_ipv4_rebound(f);
daniele 51:18637a3d071f 52 } else if (hdr->type == PICO_ICMP_UNREACH) {
daniele 51:18637a3d071f 53 f->net_hdr = f->transport_hdr + PICO_ICMPHDR_UN_SIZE;
daniele 51:18637a3d071f 54 pico_ipv4_unreachable(f, hdr->code);
daniele 51:18637a3d071f 55 } else if (hdr->type == PICO_ICMP_ECHOREPLY) {
daniele 51:18637a3d071f 56 #ifdef PICO_SUPPORT_PING
daniele 51:18637a3d071f 57 ping_recv_reply(f);
daniele 51:18637a3d071f 58 #endif
daniele 51:18637a3d071f 59 pico_frame_discard(f);
daniele 51:18637a3d071f 60 } else {
daniele 51:18637a3d071f 61 pico_frame_discard(f);
daniele 51:18637a3d071f 62 }
daniele 51:18637a3d071f 63 return 0;
daniele 51:18637a3d071f 64 }
daniele 51:18637a3d071f 65
daniele 51:18637a3d071f 66 static int pico_icmp4_process_out(struct pico_protocol *self, struct pico_frame *f)
daniele 51:18637a3d071f 67 {
daniele 51:18637a3d071f 68 dbg("Called %s\n", __FUNCTION__);
daniele 51:18637a3d071f 69 return 0;
daniele 51:18637a3d071f 70 }
daniele 51:18637a3d071f 71
daniele 51:18637a3d071f 72 /* Interface: protocol definition */
daniele 51:18637a3d071f 73 struct pico_protocol pico_proto_icmp4 = {
daniele 51:18637a3d071f 74 .name = "icmp4",
daniele 51:18637a3d071f 75 .proto_number = PICO_PROTO_ICMP4,
daniele 51:18637a3d071f 76 .layer = PICO_LAYER_TRANSPORT,
daniele 51:18637a3d071f 77 .process_in = pico_icmp4_process_in,
daniele 51:18637a3d071f 78 .process_out = pico_icmp4_process_out,
daniele 51:18637a3d071f 79 .q_in = &icmp_in,
daniele 51:18637a3d071f 80 .q_out = &icmp_out,
daniele 51:18637a3d071f 81 };
daniele 51:18637a3d071f 82
daniele 51:18637a3d071f 83 static int pico_icmp4_notify(struct pico_frame *f, uint8_t type, uint8_t code)
daniele 51:18637a3d071f 84 {
daniele 51:18637a3d071f 85 struct pico_frame *reply;
daniele 51:18637a3d071f 86 struct pico_icmp4_hdr *hdr;
daniele 51:18637a3d071f 87 struct pico_ipv4_hdr *info;
daniele 51:18637a3d071f 88 if (f == NULL) {
daniele 51:18637a3d071f 89 pico_err = PICO_ERR_EINVAL;
daniele 51:18637a3d071f 90 return -1;
daniele 51:18637a3d071f 91 }
daniele 51:18637a3d071f 92 reply = pico_proto_ipv4.alloc(&pico_proto_ipv4, 8 + sizeof(struct pico_ipv4_hdr) + PICO_ICMPHDR_UN_SIZE);
daniele 51:18637a3d071f 93 info = (struct pico_ipv4_hdr*)(f->net_hdr);
daniele 51:18637a3d071f 94 hdr = (struct pico_icmp4_hdr *) reply->transport_hdr;
daniele 51:18637a3d071f 95 hdr->type = type;
daniele 51:18637a3d071f 96 hdr->code = code;
daniele 51:18637a3d071f 97 hdr->hun.ih_pmtu.ipm_nmtu = short_be(1500);
daniele 51:18637a3d071f 98 hdr->hun.ih_pmtu.ipm_void = 0;
daniele 51:18637a3d071f 99 reply->transport_len = 8 + sizeof(struct pico_ipv4_hdr) + PICO_ICMPHDR_UN_SIZE;
daniele 51:18637a3d071f 100 reply->payload = reply->transport_hdr + PICO_ICMPHDR_UN_SIZE;
daniele 51:18637a3d071f 101 memcpy(reply->payload, f->net_hdr, 8 + sizeof(struct pico_ipv4_hdr));
daniele 51:18637a3d071f 102 pico_icmp4_checksum(reply);
daniele 51:18637a3d071f 103 pico_ipv4_frame_push(reply, &info->src, PICO_PROTO_ICMP4);
daniele 51:18637a3d071f 104 return 0;
daniele 51:18637a3d071f 105 }
daniele 51:18637a3d071f 106
daniele 51:18637a3d071f 107 int pico_icmp4_port_unreachable(struct pico_frame *f)
daniele 51:18637a3d071f 108 {
daniele 51:18637a3d071f 109 /*Parameter check executed in pico_icmp4_notify*/
daniele 51:18637a3d071f 110 return pico_icmp4_notify(f, PICO_ICMP_UNREACH, PICO_ICMP_UNREACH_PORT);
daniele 51:18637a3d071f 111 }
daniele 51:18637a3d071f 112
daniele 51:18637a3d071f 113 int pico_icmp4_proto_unreachable(struct pico_frame *f)
daniele 51:18637a3d071f 114 {
daniele 51:18637a3d071f 115 /*Parameter check executed in pico_icmp4_notify*/
daniele 51:18637a3d071f 116 return pico_icmp4_notify(f, PICO_ICMP_UNREACH, PICO_ICMP_UNREACH_PROTOCOL);
daniele 51:18637a3d071f 117 }
daniele 51:18637a3d071f 118
daniele 51:18637a3d071f 119 int pico_icmp4_dest_unreachable(struct pico_frame *f)
daniele 51:18637a3d071f 120 {
daniele 51:18637a3d071f 121 /*Parameter check executed in pico_icmp4_notify*/
daniele 51:18637a3d071f 122 return pico_icmp4_notify(f, PICO_ICMP_UNREACH, PICO_ICMP_UNREACH_HOST);
daniele 51:18637a3d071f 123 }
daniele 51:18637a3d071f 124
daniele 51:18637a3d071f 125 int pico_icmp4_ttl_expired(struct pico_frame *f)
daniele 51:18637a3d071f 126 {
daniele 51:18637a3d071f 127 /*Parameter check executed in pico_icmp4_notify*/
daniele 51:18637a3d071f 128 return pico_icmp4_notify(f, PICO_ICMP_TIME_EXCEEDED, PICO_ICMP_TIMXCEED_INTRANS);
daniele 51:18637a3d071f 129 }
daniele 51:18637a3d071f 130
daniele 51:18637a3d071f 131 int pico_icmp4_packet_filtered(struct pico_frame *f)
daniele 51:18637a3d071f 132 {
daniele 51:18637a3d071f 133 /*Parameter check executed in pico_icmp4_notify*/
daniele 51:18637a3d071f 134 /*Packet Filtered: type 3, code 13 (Communication Administratively Prohibited)*/
daniele 51:18637a3d071f 135 return pico_icmp4_notify(f, PICO_ICMP_UNREACH, PICO_ICMP_UNREACH_FILTER_PROHIB);
daniele 51:18637a3d071f 136 }
daniele 51:18637a3d071f 137
daniele 51:18637a3d071f 138 /***********************/
daniele 51:18637a3d071f 139 /* Ping implementation */
daniele 51:18637a3d071f 140 /***********************/
daniele 51:18637a3d071f 141 /***********************/
daniele 51:18637a3d071f 142 /***********************/
daniele 51:18637a3d071f 143 /***********************/
daniele 51:18637a3d071f 144
daniele 51:18637a3d071f 145
daniele 51:18637a3d071f 146 #ifdef PICO_SUPPORT_PING
daniele 51:18637a3d071f 147
daniele 51:18637a3d071f 148
daniele 51:18637a3d071f 149 struct pico_icmp4_ping_cookie
daniele 51:18637a3d071f 150 {
daniele 51:18637a3d071f 151 struct pico_ip4 dst;
daniele 51:18637a3d071f 152 uint16_t err;
daniele 51:18637a3d071f 153 uint16_t id;
daniele 51:18637a3d071f 154 uint16_t seq;
daniele 51:18637a3d071f 155 uint16_t size;
daniele 51:18637a3d071f 156 int count;
daniele 51:18637a3d071f 157 unsigned long timestamp;
daniele 51:18637a3d071f 158 int interval;
daniele 51:18637a3d071f 159 int timeout;
daniele 51:18637a3d071f 160 void (*cb)(struct pico_icmp4_stats*);
daniele 51:18637a3d071f 161
daniele 51:18637a3d071f 162 };
daniele 51:18637a3d071f 163
daniele 51:18637a3d071f 164 static int cookie_compare(void *ka, void *kb)
daniele 51:18637a3d071f 165 {
daniele 51:18637a3d071f 166 struct pico_icmp4_ping_cookie *a = ka, *b = kb;
daniele 51:18637a3d071f 167 if (a->id < b->id)
daniele 51:18637a3d071f 168 return -1;
daniele 51:18637a3d071f 169 if (a->id > b->id)
daniele 51:18637a3d071f 170 return 1;
daniele 51:18637a3d071f 171 return (a->seq - b->seq);
daniele 51:18637a3d071f 172 }
daniele 51:18637a3d071f 173
daniele 51:18637a3d071f 174 PICO_TREE_DECLARE(Pings,cookie_compare);
daniele 51:18637a3d071f 175
daniele 51:18637a3d071f 176 static int pico_icmp4_send_echo(struct pico_icmp4_ping_cookie *cookie)
daniele 51:18637a3d071f 177 {
daniele 51:18637a3d071f 178 struct pico_frame *echo = pico_proto_ipv4.alloc(&pico_proto_ipv4, PICO_ICMPHDR_UN_SIZE + cookie->size);
daniele 51:18637a3d071f 179 struct pico_icmp4_hdr *hdr;
daniele 51:18637a3d071f 180
daniele 51:18637a3d071f 181 hdr = (struct pico_icmp4_hdr *) echo->transport_hdr;
daniele 51:18637a3d071f 182
daniele 51:18637a3d071f 183 hdr->type = PICO_ICMP_ECHO;
daniele 51:18637a3d071f 184 hdr->code = 0;
daniele 51:18637a3d071f 185 hdr->hun.ih_idseq.idseq_id = short_be(cookie->id);
daniele 51:18637a3d071f 186 hdr->hun.ih_idseq.idseq_seq = short_be(cookie->seq);
daniele 51:18637a3d071f 187 echo->transport_len = PICO_ICMPHDR_UN_SIZE + cookie->size;
daniele 51:18637a3d071f 188 echo->payload = echo->transport_hdr + PICO_ICMPHDR_UN_SIZE;
daniele 51:18637a3d071f 189 echo->payload_len = cookie->size;
daniele 51:18637a3d071f 190 /* XXX: Fill payload */
daniele 51:18637a3d071f 191 pico_icmp4_checksum(echo);
daniele 51:18637a3d071f 192 pico_ipv4_frame_push(echo, &cookie->dst, PICO_PROTO_ICMP4);
daniele 51:18637a3d071f 193 return 0;
daniele 51:18637a3d071f 194 }
daniele 51:18637a3d071f 195
daniele 51:18637a3d071f 196
daniele 51:18637a3d071f 197 static void ping_timeout(unsigned long now, void *arg)
daniele 51:18637a3d071f 198 {
daniele 51:18637a3d071f 199 struct pico_icmp4_ping_cookie *cookie = (struct pico_icmp4_ping_cookie *)arg;
daniele 51:18637a3d071f 200 if(pico_tree_findKey(&Pings,cookie)){
daniele 51:18637a3d071f 201 if (cookie->err == PICO_PING_ERR_PENDING) {
daniele 51:18637a3d071f 202 struct pico_icmp4_stats stats;
daniele 51:18637a3d071f 203 stats.dst = cookie->dst;
daniele 51:18637a3d071f 204 stats.seq = cookie->seq;
daniele 51:18637a3d071f 205 stats.time = 0;
daniele 51:18637a3d071f 206 stats.size = cookie->size;
daniele 51:18637a3d071f 207 stats.err = PICO_PING_ERR_TIMEOUT;
daniele 51:18637a3d071f 208 dbg(" ---- Ping timeout!!!\n");
daniele 51:18637a3d071f 209 cookie->cb(&stats);
daniele 51:18637a3d071f 210 }
daniele 51:18637a3d071f 211
daniele 51:18637a3d071f 212 pico_tree_delete(&Pings,cookie);
daniele 51:18637a3d071f 213 pico_free(cookie);
daniele 51:18637a3d071f 214 }
daniele 51:18637a3d071f 215 }
daniele 51:18637a3d071f 216
daniele 51:18637a3d071f 217 static void next_ping(unsigned long now, void *arg);
daniele 51:18637a3d071f 218 static inline void send_ping(struct pico_icmp4_ping_cookie *cookie)
daniele 51:18637a3d071f 219 {
daniele 51:18637a3d071f 220 pico_icmp4_send_echo(cookie);
daniele 51:18637a3d071f 221 cookie->timestamp = pico_tick;
daniele 51:18637a3d071f 222 pico_timer_add(cookie->timeout, ping_timeout, cookie);
daniele 51:18637a3d071f 223 if (cookie->seq < cookie->count)
daniele 51:18637a3d071f 224 pico_timer_add(cookie->interval, next_ping, cookie);
daniele 51:18637a3d071f 225 }
daniele 51:18637a3d071f 226
daniele 51:18637a3d071f 227 static void next_ping(unsigned long now, void *arg)
daniele 51:18637a3d071f 228 {
daniele 51:18637a3d071f 229 struct pico_icmp4_ping_cookie *newcookie, *cookie = (struct pico_icmp4_ping_cookie *)arg;
daniele 51:18637a3d071f 230
daniele 51:18637a3d071f 231 if(pico_tree_findKey(&Pings,cookie)){
daniele 51:18637a3d071f 232 if (cookie->seq < cookie->count) {
daniele 51:18637a3d071f 233 newcookie = pico_zalloc(sizeof(struct pico_icmp4_ping_cookie));
daniele 51:18637a3d071f 234 if (!newcookie)
daniele 51:18637a3d071f 235 return;
daniele 51:18637a3d071f 236 memcpy(newcookie, cookie, sizeof(struct pico_icmp4_ping_cookie));
daniele 51:18637a3d071f 237 newcookie->seq++;
daniele 51:18637a3d071f 238
daniele 51:18637a3d071f 239 pico_tree_insert(&Pings,newcookie);
daniele 51:18637a3d071f 240 send_ping(newcookie);
daniele 51:18637a3d071f 241 }
daniele 51:18637a3d071f 242 }
daniele 51:18637a3d071f 243 }
daniele 51:18637a3d071f 244
daniele 51:18637a3d071f 245
daniele 51:18637a3d071f 246 static void ping_recv_reply(struct pico_frame *f)
daniele 51:18637a3d071f 247 {
daniele 51:18637a3d071f 248 struct pico_icmp4_ping_cookie test, *cookie;
daniele 51:18637a3d071f 249 struct pico_icmp4_hdr *hdr = (struct pico_icmp4_hdr *) f->transport_hdr;
daniele 51:18637a3d071f 250 test.id = short_be(hdr->hun.ih_idseq.idseq_id );
daniele 51:18637a3d071f 251 test.seq = short_be(hdr->hun.ih_idseq.idseq_seq);
daniele 51:18637a3d071f 252
daniele 51:18637a3d071f 253 cookie = pico_tree_findKey(&Pings, &test);
daniele 51:18637a3d071f 254 if (cookie) {
daniele 51:18637a3d071f 255 struct pico_icmp4_stats stats;
daniele 51:18637a3d071f 256 cookie->err = PICO_PING_ERR_REPLIED;
daniele 51:18637a3d071f 257 stats.dst = cookie->dst;
daniele 51:18637a3d071f 258 stats.seq = cookie->seq;
daniele 51:18637a3d071f 259 stats.size = cookie->size;
daniele 51:18637a3d071f 260 stats.time = pico_tick - cookie->timestamp;
daniele 51:18637a3d071f 261 stats.err = cookie->err;
daniele 51:18637a3d071f 262 stats.ttl = ((struct pico_ipv4_hdr *)f->net_hdr)->ttl;
daniele 51:18637a3d071f 263 if(cookie->cb != NULL)
daniele 51:18637a3d071f 264 cookie->cb(&stats);
daniele 51:18637a3d071f 265 } else {
daniele 51:18637a3d071f 266 dbg("Reply for seq=%d, not found.\n", test.seq);
daniele 51:18637a3d071f 267 }
daniele 51:18637a3d071f 268 }
daniele 51:18637a3d071f 269
daniele 51:18637a3d071f 270 int pico_icmp4_ping(char *dst, int count, int interval, int timeout, int size, void (*cb)(struct pico_icmp4_stats *))
daniele 51:18637a3d071f 271 {
daniele 51:18637a3d071f 272 static uint16_t next_id = 0x91c0;
daniele 51:18637a3d071f 273 struct pico_icmp4_ping_cookie *cookie;
daniele 51:18637a3d071f 274
daniele 51:18637a3d071f 275 if((dst == NULL) || (interval == 0) || (timeout == 0) || (count == 0)){
daniele 51:18637a3d071f 276 pico_err = PICO_ERR_EINVAL;
daniele 51:18637a3d071f 277 return -1;
daniele 51:18637a3d071f 278 }
daniele 51:18637a3d071f 279
daniele 51:18637a3d071f 280 cookie = pico_zalloc(sizeof(struct pico_icmp4_ping_cookie));
daniele 51:18637a3d071f 281 if (!cookie) {
daniele 51:18637a3d071f 282 pico_err = PICO_ERR_ENOMEM;
daniele 51:18637a3d071f 283 return -1;
daniele 51:18637a3d071f 284 }
daniele 51:18637a3d071f 285
daniele 51:18637a3d071f 286 if (pico_string_to_ipv4(dst, &cookie->dst.addr) < 0) {
daniele 51:18637a3d071f 287 pico_err = PICO_ERR_EINVAL;
daniele 51:18637a3d071f 288 pico_free(cookie);
daniele 51:18637a3d071f 289 return -1;
daniele 51:18637a3d071f 290 }
daniele 51:18637a3d071f 291 cookie->seq = 1;
daniele 51:18637a3d071f 292 cookie->id = next_id++;
daniele 51:18637a3d071f 293 cookie->err = PICO_PING_ERR_PENDING;
daniele 51:18637a3d071f 294 cookie->size = size;
daniele 51:18637a3d071f 295 cookie->interval = interval;
daniele 51:18637a3d071f 296 cookie->timeout = timeout;
daniele 51:18637a3d071f 297 cookie->cb = cb;
daniele 51:18637a3d071f 298 cookie->count = count;
daniele 51:18637a3d071f 299
daniele 51:18637a3d071f 300 pico_tree_insert(&Pings,cookie);
daniele 51:18637a3d071f 301 send_ping(cookie);
daniele 51:18637a3d071f 302
daniele 51:18637a3d071f 303 return 0;
daniele 51:18637a3d071f 304
daniele 51:18637a3d071f 305 }
daniele 51:18637a3d071f 306
daniele 51:18637a3d071f 307 #endif