Free (GPLv2) TCP/IP stack developed by TASS Belgium
Fork of PicoTCP by
modules/pico_dhcp_client.c@51:18637a3d071f, 2013-08-03 (annotated)
- Committer:
- daniele
- Date:
- Sat Aug 03 08:50:27 2013 +0000
- Revision:
- 51:18637a3d071f
- Parent:
- 29:1a47b7151851
Branch for CDC-ECM: Work in progress
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
daniele | 29:1a47b7151851 | 1 | /********************************************************************* |
daniele | 29:1a47b7151851 | 2 | PicoTCP. Copyright (c) 2012 TASS Belgium NV. Some rights reserved. |
daniele | 29:1a47b7151851 | 3 | See LICENSE and COPYING for usage. |
daniele | 29:1a47b7151851 | 4 | |
daniele | 29:1a47b7151851 | 5 | Authors: Frederik Van Slycken, Kristof Roelants |
daniele | 29:1a47b7151851 | 6 | *********************************************************************/ |
daniele | 29:1a47b7151851 | 7 | |
daniele | 29:1a47b7151851 | 8 | #include "pico_dhcp_client.h" |
daniele | 29:1a47b7151851 | 9 | #include "pico_stack.h" |
daniele | 29:1a47b7151851 | 10 | #include "pico_config.h" |
daniele | 29:1a47b7151851 | 11 | #include "pico_device.h" |
daniele | 29:1a47b7151851 | 12 | #include "pico_ipv4.h" |
daniele | 29:1a47b7151851 | 13 | #include "pico_socket.h" |
daniele | 29:1a47b7151851 | 14 | |
daniele | 29:1a47b7151851 | 15 | #ifdef PICO_SUPPORT_DHCPC |
daniele | 29:1a47b7151851 | 16 | |
daniele | 29:1a47b7151851 | 17 | /*********** |
daniele | 29:1a47b7151851 | 18 | * structs * |
daniele | 29:1a47b7151851 | 19 | ***********/ |
daniele | 29:1a47b7151851 | 20 | |
daniele | 29:1a47b7151851 | 21 | static uint8_t dhcp_client_mutex = 1; /* to serialize client negotations if multiple devices */ |
daniele | 29:1a47b7151851 | 22 | |
daniele | 29:1a47b7151851 | 23 | struct dhcp_timer_param{ |
daniele | 29:1a47b7151851 | 24 | uint16_t type; |
daniele | 29:1a47b7151851 | 25 | struct pico_dhcp_client_cookie* cli; |
daniele | 29:1a47b7151851 | 26 | int valid; |
daniele | 29:1a47b7151851 | 27 | }; |
daniele | 29:1a47b7151851 | 28 | |
daniele | 29:1a47b7151851 | 29 | struct pico_dhcp_client_cookie |
daniele | 29:1a47b7151851 | 30 | { |
daniele | 29:1a47b7151851 | 31 | uint32_t xid; |
daniele | 29:1a47b7151851 | 32 | uint32_t *xid_user; |
daniele | 29:1a47b7151851 | 33 | struct pico_ip4 address; |
daniele | 29:1a47b7151851 | 34 | struct pico_ip4 netmask; |
daniele | 29:1a47b7151851 | 35 | struct pico_ip4 gateway; |
daniele | 29:1a47b7151851 | 36 | struct pico_ip4 nameserver; |
daniele | 29:1a47b7151851 | 37 | struct pico_ip4 server_id; |
daniele | 29:1a47b7151851 | 38 | uint32_t lease_time; |
daniele | 29:1a47b7151851 | 39 | uint32_t T1; |
daniele | 29:1a47b7151851 | 40 | uint32_t T2; |
daniele | 29:1a47b7151851 | 41 | struct pico_socket* socket; |
daniele | 29:1a47b7151851 | 42 | int connected; |
daniele | 29:1a47b7151851 | 43 | struct pico_device* device; |
daniele | 29:1a47b7151851 | 44 | unsigned long start_time; |
daniele | 29:1a47b7151851 | 45 | int attempt; |
daniele | 29:1a47b7151851 | 46 | enum dhcp_negotiation_state state; |
daniele | 29:1a47b7151851 | 47 | void (*cb)(void* cli, int code); |
daniele | 29:1a47b7151851 | 48 | struct dhcp_timer_param* timer_param_1; |
daniele | 29:1a47b7151851 | 49 | struct dhcp_timer_param* timer_param_2; |
daniele | 29:1a47b7151851 | 50 | struct dhcp_timer_param* timer_param_lease; |
daniele | 29:1a47b7151851 | 51 | struct dhcp_timer_param* timer_param_retransmit; |
daniele | 29:1a47b7151851 | 52 | int link_added; |
daniele | 29:1a47b7151851 | 53 | }; |
daniele | 29:1a47b7151851 | 54 | |
daniele | 29:1a47b7151851 | 55 | static int dhcp_cookies_cmp(void *ka, void *kb) |
daniele | 29:1a47b7151851 | 56 | { |
daniele | 29:1a47b7151851 | 57 | struct pico_dhcp_client_cookie *a = ka, *b = kb; |
daniele | 29:1a47b7151851 | 58 | if (a->xid < b->xid) |
daniele | 29:1a47b7151851 | 59 | return -1; |
daniele | 29:1a47b7151851 | 60 | else if (a->xid > b->xid) |
daniele | 29:1a47b7151851 | 61 | return 1; |
daniele | 29:1a47b7151851 | 62 | else |
daniele | 29:1a47b7151851 | 63 | return 0; |
daniele | 29:1a47b7151851 | 64 | } |
daniele | 29:1a47b7151851 | 65 | PICO_TREE_DECLARE(DHCPCookies, dhcp_cookies_cmp); |
daniele | 29:1a47b7151851 | 66 | |
daniele | 29:1a47b7151851 | 67 | /************************* |
daniele | 29:1a47b7151851 | 68 | * function declarations * |
daniele | 29:1a47b7151851 | 69 | *************************/ |
daniele | 29:1a47b7151851 | 70 | static void pico_dhcp_state_machine(int type, struct pico_dhcp_client_cookie* cli, uint8_t* data, int len); |
daniele | 29:1a47b7151851 | 71 | static void pico_dhcp_reinitiate_negotiation(unsigned long now, void *arg); |
daniele | 29:1a47b7151851 | 72 | |
daniele | 29:1a47b7151851 | 73 | //cb |
daniele | 29:1a47b7151851 | 74 | static void pico_dhcp_wakeup(uint16_t ev, struct pico_socket *s); |
daniele | 29:1a47b7151851 | 75 | static void dhcp_timer_cb(unsigned long tick, void* param); |
daniele | 29:1a47b7151851 | 76 | |
daniele | 29:1a47b7151851 | 77 | //util |
daniele | 29:1a47b7151851 | 78 | static void pico_dhcp_retry(struct pico_dhcp_client_cookie *client); |
daniele | 29:1a47b7151851 | 79 | static int dhclient_send(struct pico_dhcp_client_cookie *cli, uint8_t msg_type); |
daniele | 29:1a47b7151851 | 80 | static int pico_dhcp_verify_and_identify_type(uint8_t* data, int len, struct pico_dhcp_client_cookie *cli); |
daniele | 29:1a47b7151851 | 81 | static int init_cookie(struct pico_dhcp_client_cookie* cli); |
daniele | 29:1a47b7151851 | 82 | static struct pico_dhcp_client_cookie* get_cookie_by_xid(uint32_t xid); |
daniele | 29:1a47b7151851 | 83 | static uint32_t get_xid(uint8_t* data); |
daniele | 29:1a47b7151851 | 84 | |
daniele | 29:1a47b7151851 | 85 | //fsm functions |
daniele | 29:1a47b7151851 | 86 | static int recv_offer(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len); |
daniele | 29:1a47b7151851 | 87 | static int recv_ack(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len); |
daniele | 29:1a47b7151851 | 88 | static int renew(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len); |
daniele | 29:1a47b7151851 | 89 | static int reset(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len); |
daniele | 29:1a47b7151851 | 90 | static int retransmit(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len); |
daniele | 29:1a47b7151851 | 91 | |
daniele | 29:1a47b7151851 | 92 | //fsm implementation |
daniele | 29:1a47b7151851 | 93 | static void pico_dhcp_state_machine(int type, struct pico_dhcp_client_cookie* cli, uint8_t* data, int len); |
daniele | 29:1a47b7151851 | 94 | |
daniele | 29:1a47b7151851 | 95 | /*************** |
daniele | 29:1a47b7151851 | 96 | * entry point * |
daniele | 29:1a47b7151851 | 97 | ***************/ |
daniele | 29:1a47b7151851 | 98 | |
daniele | 29:1a47b7151851 | 99 | static uint32_t pico_dhcp_execute_init(struct pico_dhcp_client_cookie *cli) |
daniele | 29:1a47b7151851 | 100 | { |
daniele | 29:1a47b7151851 | 101 | if (!dhcp_client_mutex) { |
daniele | 29:1a47b7151851 | 102 | pico_timer_add(3000, pico_dhcp_reinitiate_negotiation, cli); |
daniele | 29:1a47b7151851 | 103 | return 0; |
daniele | 29:1a47b7151851 | 104 | } |
daniele | 29:1a47b7151851 | 105 | dhcp_client_mutex--; |
daniele | 29:1a47b7151851 | 106 | |
daniele | 29:1a47b7151851 | 107 | if (init_cookie(cli) < 0) |
daniele | 29:1a47b7151851 | 108 | return -1; |
daniele | 29:1a47b7151851 | 109 | |
daniele | 29:1a47b7151851 | 110 | dbg("DHCPC: cookie with xid %u\n", cli->xid); |
daniele | 29:1a47b7151851 | 111 | |
daniele | 29:1a47b7151851 | 112 | if (pico_tree_insert(&DHCPCookies, cli)) { |
daniele | 29:1a47b7151851 | 113 | pico_err = PICO_ERR_EAGAIN; |
daniele | 29:1a47b7151851 | 114 | if(cli->cb != NULL) { |
daniele | 29:1a47b7151851 | 115 | cli->cb(cli, PICO_DHCP_ERROR); |
daniele | 29:1a47b7151851 | 116 | } |
daniele | 29:1a47b7151851 | 117 | pico_free(cli); |
daniele | 29:1a47b7151851 | 118 | return -1; /* Element key already exists */ |
daniele | 29:1a47b7151851 | 119 | } |
daniele | 29:1a47b7151851 | 120 | |
daniele | 29:1a47b7151851 | 121 | if (dhclient_send(cli, PICO_DHCP_MSG_DISCOVER) < 0) |
daniele | 29:1a47b7151851 | 122 | return -1; |
daniele | 29:1a47b7151851 | 123 | |
daniele | 29:1a47b7151851 | 124 | return 0; |
daniele | 29:1a47b7151851 | 125 | } |
daniele | 29:1a47b7151851 | 126 | |
daniele | 29:1a47b7151851 | 127 | /* returns a pointer to the client cookie. The user should pass this pointer every time he calls a dhcp-function. This is so that we can (one day) support dhcp on multiple interfaces */ |
daniele | 29:1a47b7151851 | 128 | int pico_dhcp_initiate_negotiation(struct pico_device *device, void (*callback)(void *cli, int code), uint32_t *xid) |
daniele | 29:1a47b7151851 | 129 | { |
daniele | 29:1a47b7151851 | 130 | struct pico_dhcp_client_cookie *cli; |
daniele | 29:1a47b7151851 | 131 | |
daniele | 29:1a47b7151851 | 132 | if(!device || !callback || !xid){ |
daniele | 29:1a47b7151851 | 133 | pico_err = PICO_ERR_EINVAL; |
daniele | 29:1a47b7151851 | 134 | return -1; |
daniele | 29:1a47b7151851 | 135 | } |
daniele | 29:1a47b7151851 | 136 | cli = pico_zalloc(sizeof(struct pico_dhcp_client_cookie)); |
daniele | 29:1a47b7151851 | 137 | if(!cli){ |
daniele | 29:1a47b7151851 | 138 | pico_err = PICO_ERR_ENOMEM; |
daniele | 29:1a47b7151851 | 139 | return -1; |
daniele | 29:1a47b7151851 | 140 | } |
daniele | 29:1a47b7151851 | 141 | |
daniele | 29:1a47b7151851 | 142 | cli->device = device; |
daniele | 29:1a47b7151851 | 143 | cli->cb = callback; |
daniele | 29:1a47b7151851 | 144 | cli->xid_user = xid; |
daniele | 29:1a47b7151851 | 145 | *(cli->xid_user) = 0; |
daniele | 29:1a47b7151851 | 146 | |
daniele | 29:1a47b7151851 | 147 | return pico_dhcp_execute_init(cli); |
daniele | 29:1a47b7151851 | 148 | } |
daniele | 29:1a47b7151851 | 149 | |
daniele | 29:1a47b7151851 | 150 | static void pico_dhcp_reinitiate_negotiation(unsigned long now, void *arg) |
daniele | 29:1a47b7151851 | 151 | { |
daniele | 29:1a47b7151851 | 152 | struct pico_dhcp_client_cookie *cli = (struct pico_dhcp_client_cookie *) arg; |
daniele | 29:1a47b7151851 | 153 | |
daniele | 29:1a47b7151851 | 154 | pico_dhcp_execute_init(cli); |
daniele | 29:1a47b7151851 | 155 | |
daniele | 29:1a47b7151851 | 156 | return; |
daniele | 29:1a47b7151851 | 157 | } |
daniele | 29:1a47b7151851 | 158 | |
daniele | 29:1a47b7151851 | 159 | /******************** |
daniele | 29:1a47b7151851 | 160 | * access functions * |
daniele | 29:1a47b7151851 | 161 | ********************/ |
daniele | 29:1a47b7151851 | 162 | |
daniele | 29:1a47b7151851 | 163 | struct pico_ip4 pico_dhcp_get_address(void* cli) |
daniele | 29:1a47b7151851 | 164 | { |
daniele | 29:1a47b7151851 | 165 | return ((struct pico_dhcp_client_cookie*)cli)->address; |
daniele | 29:1a47b7151851 | 166 | } |
daniele | 29:1a47b7151851 | 167 | |
daniele | 29:1a47b7151851 | 168 | struct pico_ip4 pico_dhcp_get_gateway(void* cli) |
daniele | 29:1a47b7151851 | 169 | { |
daniele | 29:1a47b7151851 | 170 | return ((struct pico_dhcp_client_cookie*)cli)->gateway; |
daniele | 29:1a47b7151851 | 171 | } |
daniele | 29:1a47b7151851 | 172 | |
daniele | 29:1a47b7151851 | 173 | struct pico_ip4 pico_dhcp_get_nameserver(void* cli) |
daniele | 29:1a47b7151851 | 174 | { |
daniele | 29:1a47b7151851 | 175 | return ((struct pico_dhcp_client_cookie*)cli)->nameserver; |
daniele | 29:1a47b7151851 | 176 | } |
daniele | 29:1a47b7151851 | 177 | |
daniele | 29:1a47b7151851 | 178 | /************* |
daniele | 29:1a47b7151851 | 179 | * callbacks * |
daniele | 29:1a47b7151851 | 180 | *************/ |
daniele | 29:1a47b7151851 | 181 | |
daniele | 29:1a47b7151851 | 182 | static void pico_dhcp_wakeup(uint16_t ev, struct pico_socket *s) |
daniele | 29:1a47b7151851 | 183 | { |
daniele | 29:1a47b7151851 | 184 | uint8_t buf[DHCPC_DATAGRAM_SIZE]; |
daniele | 29:1a47b7151851 | 185 | int r=0; |
daniele | 29:1a47b7151851 | 186 | uint32_t peer; |
daniele | 29:1a47b7151851 | 187 | uint16_t port; |
daniele | 29:1a47b7151851 | 188 | int type; |
daniele | 29:1a47b7151851 | 189 | |
daniele | 29:1a47b7151851 | 190 | struct pico_dhcp_client_cookie *cli; |
daniele | 29:1a47b7151851 | 191 | dbg("DHCPC: called dhcp_wakeup\n"); |
daniele | 29:1a47b7151851 | 192 | if (ev == PICO_SOCK_EV_RD) { |
daniele | 29:1a47b7151851 | 193 | do { |
daniele | 29:1a47b7151851 | 194 | r = pico_socket_recvfrom(s, buf, DHCPC_DATAGRAM_SIZE, &peer, &port); |
daniele | 29:1a47b7151851 | 195 | cli = get_cookie_by_xid(get_xid(buf)); |
daniele | 29:1a47b7151851 | 196 | if(cli == NULL) |
daniele | 29:1a47b7151851 | 197 | return; |
daniele | 29:1a47b7151851 | 198 | if (r > 0 && port == PICO_DHCPD_PORT) { |
daniele | 29:1a47b7151851 | 199 | type = pico_dhcp_verify_and_identify_type(buf, r, cli); |
daniele | 29:1a47b7151851 | 200 | pico_dhcp_state_machine(type, cli, buf, r); |
daniele | 29:1a47b7151851 | 201 | } |
daniele | 29:1a47b7151851 | 202 | } while(r>0); |
daniele | 29:1a47b7151851 | 203 | } |
daniele | 29:1a47b7151851 | 204 | } |
daniele | 29:1a47b7151851 | 205 | |
daniele | 29:1a47b7151851 | 206 | static void dhcp_timer_cb(unsigned long tick, void* param) |
daniele | 29:1a47b7151851 | 207 | { |
daniele | 29:1a47b7151851 | 208 | struct dhcp_timer_param* param2 = (struct dhcp_timer_param*) param; |
daniele | 29:1a47b7151851 | 209 | if(param2->valid == 1){ |
daniele | 29:1a47b7151851 | 210 | //dbg("called timer cb on active timer type %d\n",param2->type); |
daniele | 29:1a47b7151851 | 211 | pico_dhcp_state_machine(param2->type, param2->cli, NULL, 0); |
daniele | 29:1a47b7151851 | 212 | } |
daniele | 29:1a47b7151851 | 213 | if(param2->cli->timer_param_1 == param){ |
daniele | 29:1a47b7151851 | 214 | param2->cli->timer_param_1 = NULL; |
daniele | 29:1a47b7151851 | 215 | } |
daniele | 29:1a47b7151851 | 216 | if(param2->cli->timer_param_2 == param){ |
daniele | 29:1a47b7151851 | 217 | param2->cli->timer_param_2 = NULL; |
daniele | 29:1a47b7151851 | 218 | } |
daniele | 29:1a47b7151851 | 219 | if(param2->cli->timer_param_lease == param){ |
daniele | 29:1a47b7151851 | 220 | param2->cli->timer_param_lease = NULL; |
daniele | 29:1a47b7151851 | 221 | } |
daniele | 29:1a47b7151851 | 222 | if(param2->cli->timer_param_retransmit == param){ |
daniele | 29:1a47b7151851 | 223 | param2->cli->timer_param_retransmit = NULL; |
daniele | 29:1a47b7151851 | 224 | } |
daniele | 29:1a47b7151851 | 225 | |
daniele | 29:1a47b7151851 | 226 | pico_free(param); |
daniele | 29:1a47b7151851 | 227 | |
daniele | 29:1a47b7151851 | 228 | } |
daniele | 29:1a47b7151851 | 229 | /***************** |
daniele | 29:1a47b7151851 | 230 | * fsm functions * |
daniele | 29:1a47b7151851 | 231 | *****************/ |
daniele | 29:1a47b7151851 | 232 | |
daniele | 29:1a47b7151851 | 233 | static int recv_offer(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len) |
daniele | 29:1a47b7151851 | 234 | { |
daniele | 29:1a47b7151851 | 235 | struct pico_dhcphdr *dhdr = (struct pico_dhcphdr *) data; |
daniele | 29:1a47b7151851 | 236 | uint8_t *nextopt, opt_data[20], opt_type; |
daniele | 29:1a47b7151851 | 237 | int opt_len = 20; |
daniele | 29:1a47b7151851 | 238 | uint8_t msg_type = 0xFF; |
daniele | 29:1a47b7151851 | 239 | int T1_set = 0; |
daniele | 29:1a47b7151851 | 240 | int T2_set = 0; |
daniele | 29:1a47b7151851 | 241 | |
daniele | 29:1a47b7151851 | 242 | cli->address.addr = dhdr->yiaddr; |
daniele | 29:1a47b7151851 | 243 | |
daniele | 29:1a47b7151851 | 244 | opt_type = dhcp_get_next_option(dhdr->options, opt_data, &opt_len, &nextopt); |
daniele | 29:1a47b7151851 | 245 | while (opt_type != PICO_DHCPOPT_END) { |
daniele | 29:1a47b7151851 | 246 | if (opt_type == PICO_DHCPOPT_MSGTYPE) |
daniele | 29:1a47b7151851 | 247 | msg_type = opt_data[0]; |
daniele | 29:1a47b7151851 | 248 | if ((opt_type == PICO_DHCPOPT_LEASETIME) && (opt_len == 4)){ |
daniele | 29:1a47b7151851 | 249 | memcpy(&cli->lease_time, opt_data, 4); |
daniele | 29:1a47b7151851 | 250 | cli->lease_time = long_be(cli->lease_time); |
daniele | 29:1a47b7151851 | 251 | } |
daniele | 29:1a47b7151851 | 252 | if ((opt_type == PICO_DHCPOPT_RENEWALTIME) && (opt_len == 4)){ |
daniele | 29:1a47b7151851 | 253 | memcpy(&cli->T1, opt_data, 4); |
daniele | 29:1a47b7151851 | 254 | cli->T1 = long_be(cli->T1); |
daniele | 29:1a47b7151851 | 255 | T1_set =1; |
daniele | 29:1a47b7151851 | 256 | } |
daniele | 29:1a47b7151851 | 257 | if ((opt_type == PICO_DHCPOPT_REBINDINGTIME) && (opt_len == 4)){ |
daniele | 29:1a47b7151851 | 258 | memcpy(&cli->T2, opt_data, 4); |
daniele | 29:1a47b7151851 | 259 | cli->T2 = long_be(cli->T2); |
daniele | 29:1a47b7151851 | 260 | T2_set =1; |
daniele | 29:1a47b7151851 | 261 | } |
daniele | 29:1a47b7151851 | 262 | if ((opt_type == PICO_DHCPOPT_ROUTER) && (opt_len == 4)) //XXX assuming only one router will be advertised... |
daniele | 29:1a47b7151851 | 263 | memcpy(&cli->gateway.addr, opt_data, 4); |
daniele | 29:1a47b7151851 | 264 | if ((opt_type == PICO_DHCPOPT_DNS) && (opt_len == 4)) |
daniele | 29:1a47b7151851 | 265 | memcpy(&cli->nameserver.addr, opt_data, 4); |
daniele | 29:1a47b7151851 | 266 | if ((opt_type == PICO_DHCPOPT_NETMASK) && (opt_len == 4)) |
daniele | 29:1a47b7151851 | 267 | memcpy(&cli->netmask.addr, opt_data, 4); |
daniele | 29:1a47b7151851 | 268 | if ((opt_type == PICO_DHCPOPT_SERVERID) && (opt_len == 4)) |
daniele | 29:1a47b7151851 | 269 | memcpy(&cli->server_id.addr, opt_data, 4); |
daniele | 29:1a47b7151851 | 270 | if (opt_type == PICO_DHCPOPT_OPTIONOVERLOAD) |
daniele | 29:1a47b7151851 | 271 | dbg("DHCPC: WARNING: option overload present (not processed)"); |
daniele | 29:1a47b7151851 | 272 | |
daniele | 29:1a47b7151851 | 273 | opt_len = 20; |
daniele | 29:1a47b7151851 | 274 | opt_type = dhcp_get_next_option(NULL, opt_data, &opt_len, &nextopt); |
daniele | 29:1a47b7151851 | 275 | } |
daniele | 29:1a47b7151851 | 276 | |
daniele | 29:1a47b7151851 | 277 | /* default values for T1 and T2 if necessary */ |
daniele | 29:1a47b7151851 | 278 | if(T1_set != 1) |
daniele | 29:1a47b7151851 | 279 | cli->T1 = cli->lease_time >> 1; |
daniele | 29:1a47b7151851 | 280 | if(T2_set != 1) |
daniele | 29:1a47b7151851 | 281 | cli->T2 = (cli->lease_time * 875) / 1000; |
daniele | 29:1a47b7151851 | 282 | |
daniele | 29:1a47b7151851 | 283 | |
daniele | 29:1a47b7151851 | 284 | |
daniele | 29:1a47b7151851 | 285 | if ((msg_type != PICO_DHCP_MSG_OFFER) || !cli->lease_time || !cli->netmask.addr || !cli->server_id.addr ) |
daniele | 29:1a47b7151851 | 286 | return 0; |
daniele | 29:1a47b7151851 | 287 | |
daniele | 29:1a47b7151851 | 288 | |
daniele | 29:1a47b7151851 | 289 | dhclient_send(cli, PICO_DHCP_MSG_REQUEST); |
daniele | 29:1a47b7151851 | 290 | cli->state = DHCPSTATE_REQUEST; |
daniele | 29:1a47b7151851 | 291 | return 1; |
daniele | 29:1a47b7151851 | 292 | } |
daniele | 29:1a47b7151851 | 293 | |
daniele | 29:1a47b7151851 | 294 | static int recv_ack(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len) |
daniele | 29:1a47b7151851 | 295 | { |
daniele | 29:1a47b7151851 | 296 | struct pico_ip4 address = {0}; |
daniele | 29:1a47b7151851 | 297 | |
daniele | 29:1a47b7151851 | 298 | if(cli->link_added == 0){ |
daniele | 29:1a47b7151851 | 299 | /* close the socket bound on address 0.0.0.0 */ |
daniele | 29:1a47b7151851 | 300 | pico_socket_close(cli->socket); |
daniele | 29:1a47b7151851 | 301 | cli->socket = NULL; |
daniele | 29:1a47b7151851 | 302 | pico_ipv4_link_del(cli->device, address); |
daniele | 29:1a47b7151851 | 303 | pico_ipv4_link_add(cli->device, cli->address, cli->netmask); |
daniele | 29:1a47b7151851 | 304 | cli->link_added = 1; |
daniele | 29:1a47b7151851 | 305 | } |
daniele | 29:1a47b7151851 | 306 | cli->state = DHCPSTATE_BOUND; |
daniele | 29:1a47b7151851 | 307 | |
daniele | 29:1a47b7151851 | 308 | dbg("DHCPC: T1: %d\n",cli->T1); |
daniele | 29:1a47b7151851 | 309 | dbg("DHCPC: T2: %d\n",cli->T2); |
daniele | 29:1a47b7151851 | 310 | dbg("DHCPC: lease time: %d\n",cli->lease_time); |
daniele | 29:1a47b7151851 | 311 | |
daniele | 29:1a47b7151851 | 312 | if(cli->timer_param_1) |
daniele | 29:1a47b7151851 | 313 | cli->timer_param_1->valid = 0; |
daniele | 29:1a47b7151851 | 314 | if(cli->timer_param_2) |
daniele | 29:1a47b7151851 | 315 | cli->timer_param_2->valid = 0; |
daniele | 29:1a47b7151851 | 316 | if(cli->timer_param_lease) |
daniele | 29:1a47b7151851 | 317 | cli->timer_param_lease->valid = 0; |
daniele | 29:1a47b7151851 | 318 | if(cli->timer_param_retransmit) |
daniele | 29:1a47b7151851 | 319 | cli->timer_param_retransmit->valid = 0; |
daniele | 29:1a47b7151851 | 320 | |
daniele | 29:1a47b7151851 | 321 | |
daniele | 29:1a47b7151851 | 322 | cli->timer_param_1 = pico_zalloc(sizeof(struct dhcp_timer_param)); |
daniele | 29:1a47b7151851 | 323 | if(!cli->timer_param_1){ |
daniele | 29:1a47b7151851 | 324 | if(cli->cb != NULL){ |
daniele | 29:1a47b7151851 | 325 | pico_err = PICO_ERR_ENOMEM; |
daniele | 29:1a47b7151851 | 326 | cli->cb(cli, PICO_DHCP_ERROR); |
daniele | 29:1a47b7151851 | 327 | } |
daniele | 29:1a47b7151851 | 328 | return 0; |
daniele | 29:1a47b7151851 | 329 | } |
daniele | 29:1a47b7151851 | 330 | cli->timer_param_2 = pico_zalloc(sizeof(struct dhcp_timer_param)); |
daniele | 29:1a47b7151851 | 331 | if(!cli->timer_param_2){ |
daniele | 29:1a47b7151851 | 332 | if(cli->cb != NULL){ |
daniele | 29:1a47b7151851 | 333 | pico_err = PICO_ERR_ENOMEM; |
daniele | 29:1a47b7151851 | 334 | cli->cb(cli, PICO_DHCP_ERROR); |
daniele | 29:1a47b7151851 | 335 | } |
daniele | 29:1a47b7151851 | 336 | return 0; |
daniele | 29:1a47b7151851 | 337 | } |
daniele | 29:1a47b7151851 | 338 | cli->timer_param_lease = pico_zalloc(sizeof(struct dhcp_timer_param)); |
daniele | 29:1a47b7151851 | 339 | if(!cli->timer_param_lease){ |
daniele | 29:1a47b7151851 | 340 | if(cli->cb != NULL){ |
daniele | 29:1a47b7151851 | 341 | pico_err = PICO_ERR_ENOMEM; |
daniele | 29:1a47b7151851 | 342 | cli->cb(cli, PICO_DHCP_ERROR); |
daniele | 29:1a47b7151851 | 343 | } |
daniele | 29:1a47b7151851 | 344 | return 0; |
daniele | 29:1a47b7151851 | 345 | } |
daniele | 29:1a47b7151851 | 346 | cli->timer_param_1->valid = 1; |
daniele | 29:1a47b7151851 | 347 | cli->timer_param_2->valid = 1; |
daniele | 29:1a47b7151851 | 348 | cli->timer_param_lease->valid = 1; |
daniele | 29:1a47b7151851 | 349 | |
daniele | 29:1a47b7151851 | 350 | cli->timer_param_1->cli = cli; |
daniele | 29:1a47b7151851 | 351 | cli->timer_param_2->cli = cli; |
daniele | 29:1a47b7151851 | 352 | cli->timer_param_lease->cli = cli; |
daniele | 29:1a47b7151851 | 353 | |
daniele | 29:1a47b7151851 | 354 | cli->timer_param_1->type = PICO_DHCP_EVENT_T1; |
daniele | 29:1a47b7151851 | 355 | cli->timer_param_2->type = PICO_DHCP_EVENT_T2; |
daniele | 29:1a47b7151851 | 356 | cli->timer_param_lease->type = PICO_DHCP_EVENT_LEASE; |
daniele | 29:1a47b7151851 | 357 | //add timer |
daniele | 29:1a47b7151851 | 358 | pico_timer_add(cli->T1*1000, dhcp_timer_cb, cli->timer_param_1); |
daniele | 29:1a47b7151851 | 359 | pico_timer_add(cli->T2*1000, dhcp_timer_cb, cli->timer_param_2); |
daniele | 29:1a47b7151851 | 360 | pico_timer_add(cli->lease_time*1000, dhcp_timer_cb, cli->timer_param_lease); |
daniele | 29:1a47b7151851 | 361 | |
daniele | 29:1a47b7151851 | 362 | *(cli->xid_user) = cli->xid; |
daniele | 29:1a47b7151851 | 363 | if(cli->cb != NULL) |
daniele | 29:1a47b7151851 | 364 | cli->cb(cli, PICO_DHCP_SUCCESS); |
daniele | 29:1a47b7151851 | 365 | else |
daniele | 29:1a47b7151851 | 366 | dbg("DHCPC: no callback\n"); |
daniele | 29:1a47b7151851 | 367 | |
daniele | 29:1a47b7151851 | 368 | dhcp_client_mutex++; |
daniele | 29:1a47b7151851 | 369 | cli->state = DHCPSTATE_BOUND; |
daniele | 29:1a47b7151851 | 370 | return 0; |
daniele | 29:1a47b7151851 | 371 | } |
daniele | 29:1a47b7151851 | 372 | |
daniele | 29:1a47b7151851 | 373 | static int renew(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len) |
daniele | 29:1a47b7151851 | 374 | { |
daniele | 29:1a47b7151851 | 375 | uint16_t port = PICO_DHCP_CLIENT_PORT; |
daniele | 29:1a47b7151851 | 376 | |
daniele | 29:1a47b7151851 | 377 | /* open and bind to currently acquired address */ |
daniele | 29:1a47b7151851 | 378 | if (!cli->socket){ |
daniele | 29:1a47b7151851 | 379 | cli->socket = pico_socket_open(PICO_PROTO_IPV4, PICO_PROTO_UDP, &pico_dhcp_wakeup); |
daniele | 29:1a47b7151851 | 380 | if (!cli->socket) { |
daniele | 29:1a47b7151851 | 381 | dbg("DHCPC: error opening socket on renew: %s\n", strerror(pico_err)); |
daniele | 29:1a47b7151851 | 382 | if(cli->cb != NULL) |
daniele | 29:1a47b7151851 | 383 | cli->cb(cli, PICO_DHCP_ERROR); |
daniele | 29:1a47b7151851 | 384 | return -1; |
daniele | 29:1a47b7151851 | 385 | } |
daniele | 29:1a47b7151851 | 386 | if (pico_socket_bind(cli->socket, &cli->address, &port) != 0){ |
daniele | 29:1a47b7151851 | 387 | dbg("DHCPC: error binding socket on renew: %s\n", strerror(pico_err)); |
daniele | 29:1a47b7151851 | 388 | pico_socket_close(cli->socket); |
daniele | 29:1a47b7151851 | 389 | if(cli->cb != NULL) |
daniele | 29:1a47b7151851 | 390 | cli->cb(cli, PICO_DHCP_ERROR); |
daniele | 29:1a47b7151851 | 391 | return -1; |
daniele | 29:1a47b7151851 | 392 | } |
daniele | 29:1a47b7151851 | 393 | } |
daniele | 29:1a47b7151851 | 394 | cli->state = DHCPSTATE_RENEWING; |
daniele | 29:1a47b7151851 | 395 | dhclient_send(cli, PICO_DHCP_MSG_REQUEST); |
daniele | 29:1a47b7151851 | 396 | |
daniele | 29:1a47b7151851 | 397 | return 0; |
daniele | 29:1a47b7151851 | 398 | } |
daniele | 29:1a47b7151851 | 399 | |
daniele | 29:1a47b7151851 | 400 | static int reset(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len) |
daniele | 29:1a47b7151851 | 401 | { |
daniele | 29:1a47b7151851 | 402 | if(cli->cb != NULL) |
daniele | 29:1a47b7151851 | 403 | cli->cb(cli, PICO_DHCP_RESET); |
daniele | 29:1a47b7151851 | 404 | //reset pretty much everything |
daniele | 29:1a47b7151851 | 405 | |
daniele | 29:1a47b7151851 | 406 | if(cli->timer_param_1) |
daniele | 29:1a47b7151851 | 407 | cli->timer_param_1->valid = 0; |
daniele | 29:1a47b7151851 | 408 | if(cli->timer_param_2) |
daniele | 29:1a47b7151851 | 409 | cli->timer_param_2->valid = 0; |
daniele | 29:1a47b7151851 | 410 | if(cli->timer_param_lease) |
daniele | 29:1a47b7151851 | 411 | cli->timer_param_lease->valid = 0; |
daniele | 29:1a47b7151851 | 412 | if(cli->timer_param_retransmit) |
daniele | 29:1a47b7151851 | 413 | cli->timer_param_retransmit->valid = 0; |
daniele | 29:1a47b7151851 | 414 | |
daniele | 29:1a47b7151851 | 415 | pico_socket_close(cli->socket); |
daniele | 29:1a47b7151851 | 416 | pico_ipv4_link_del(cli->device, cli->address); |
daniele | 29:1a47b7151851 | 417 | |
daniele | 29:1a47b7151851 | 418 | //initiate negotiations again |
daniele | 29:1a47b7151851 | 419 | init_cookie(cli); |
daniele | 29:1a47b7151851 | 420 | pico_dhcp_retry(cli); |
daniele | 29:1a47b7151851 | 421 | dhclient_send(cli, PICO_DHCP_MSG_DISCOVER); |
daniele | 29:1a47b7151851 | 422 | |
daniele | 29:1a47b7151851 | 423 | return 0; |
daniele | 29:1a47b7151851 | 424 | |
daniele | 29:1a47b7151851 | 425 | } |
daniele | 29:1a47b7151851 | 426 | |
daniele | 29:1a47b7151851 | 427 | static int retransmit(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len) |
daniele | 29:1a47b7151851 | 428 | { |
daniele | 29:1a47b7151851 | 429 | pico_dhcp_retry(cli); |
daniele | 29:1a47b7151851 | 430 | |
daniele | 29:1a47b7151851 | 431 | if(cli->state == DHCPSTATE_DISCOVER) |
daniele | 29:1a47b7151851 | 432 | dhclient_send(cli, PICO_DHCP_MSG_DISCOVER); |
daniele | 29:1a47b7151851 | 433 | else if(cli->state == DHCPSTATE_RENEWING) |
daniele | 29:1a47b7151851 | 434 | dhclient_send(cli, PICO_DHCP_MSG_REQUEST); |
daniele | 29:1a47b7151851 | 435 | else |
daniele | 29:1a47b7151851 | 436 | dbg("DHCPC: WARNING: should not get here in state %d!\n", cli->state); |
daniele | 29:1a47b7151851 | 437 | |
daniele | 29:1a47b7151851 | 438 | return 0; |
daniele | 29:1a47b7151851 | 439 | |
daniele | 29:1a47b7151851 | 440 | } |
daniele | 29:1a47b7151851 | 441 | |
daniele | 29:1a47b7151851 | 442 | /********************** |
daniele | 29:1a47b7151851 | 443 | * fsm implementation * |
daniele | 29:1a47b7151851 | 444 | **********************/ |
daniele | 29:1a47b7151851 | 445 | |
daniele | 29:1a47b7151851 | 446 | struct dhcp_action_entry { |
daniele | 29:1a47b7151851 | 447 | uint16_t tcpstate; |
daniele | 29:1a47b7151851 | 448 | int (*offer)(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len); |
daniele | 29:1a47b7151851 | 449 | int (*ack)(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len); |
daniele | 29:1a47b7151851 | 450 | int (*nak)(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len); |
daniele | 29:1a47b7151851 | 451 | int (*timer1)(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len); |
daniele | 29:1a47b7151851 | 452 | int (*timer_lease)(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len); |
daniele | 29:1a47b7151851 | 453 | int (*timer_retransmit)(struct pico_dhcp_client_cookie *cli, uint8_t *data, int len); |
daniele | 29:1a47b7151851 | 454 | }; |
daniele | 29:1a47b7151851 | 455 | |
daniele | 29:1a47b7151851 | 456 | static struct dhcp_action_entry dhcp_fsm[] = { |
daniele | 29:1a47b7151851 | 457 | /* State offer ack nak timer1 timer_lease timer_retransmit*/ |
daniele | 29:1a47b7151851 | 458 | { DHCPSTATE_DISCOVER, recv_offer, NULL, NULL, NULL, reset, retransmit}, |
daniele | 29:1a47b7151851 | 459 | { DHCPSTATE_OFFER, NULL, NULL, NULL, NULL, reset, NULL}, |
daniele | 29:1a47b7151851 | 460 | { DHCPSTATE_REQUEST, NULL, recv_ack, reset, NULL, reset, retransmit}, |
daniele | 29:1a47b7151851 | 461 | { DHCPSTATE_BOUND, NULL, NULL, reset, renew, reset, NULL}, |
daniele | 29:1a47b7151851 | 462 | { DHCPSTATE_RENEWING, NULL, recv_ack, reset, NULL, reset, retransmit}, |
daniele | 29:1a47b7151851 | 463 | }; |
daniele | 29:1a47b7151851 | 464 | |
daniele | 29:1a47b7151851 | 465 | |
daniele | 29:1a47b7151851 | 466 | static void pico_dhcp_state_machine(int type, struct pico_dhcp_client_cookie* cli, uint8_t* data, int len) |
daniele | 29:1a47b7151851 | 467 | { |
daniele | 29:1a47b7151851 | 468 | dbg("DHCPC: received incoming event of type %d\n", type); |
daniele | 29:1a47b7151851 | 469 | switch(type){ |
daniele | 29:1a47b7151851 | 470 | case PICO_DHCP_MSG_OFFER: |
daniele | 29:1a47b7151851 | 471 | if(dhcp_fsm[cli->state].offer != NULL) |
daniele | 29:1a47b7151851 | 472 | dhcp_fsm[cli->state].offer(cli, data, len); |
daniele | 29:1a47b7151851 | 473 | break; |
daniele | 29:1a47b7151851 | 474 | case PICO_DHCP_MSG_ACK: |
daniele | 29:1a47b7151851 | 475 | if(dhcp_fsm[cli->state].ack != NULL){ |
daniele | 29:1a47b7151851 | 476 | dhcp_fsm[cli->state].ack(cli, data, len); |
daniele | 29:1a47b7151851 | 477 | } |
daniele | 29:1a47b7151851 | 478 | break; |
daniele | 29:1a47b7151851 | 479 | case PICO_DHCP_MSG_NAK: |
daniele | 29:1a47b7151851 | 480 | if(dhcp_fsm[cli->state].nak!= NULL){ |
daniele | 29:1a47b7151851 | 481 | dhcp_fsm[cli->state].nak(cli, data, len); |
daniele | 29:1a47b7151851 | 482 | } |
daniele | 29:1a47b7151851 | 483 | break; |
daniele | 29:1a47b7151851 | 484 | case PICO_DHCP_EVENT_T1: |
daniele | 29:1a47b7151851 | 485 | if(dhcp_fsm[cli->state].timer1!= NULL){ |
daniele | 29:1a47b7151851 | 486 | dhcp_fsm[cli->state].timer1(cli, NULL, 0); |
daniele | 29:1a47b7151851 | 487 | } |
daniele | 29:1a47b7151851 | 488 | break; |
daniele | 29:1a47b7151851 | 489 | case PICO_DHCP_EVENT_LEASE: |
daniele | 29:1a47b7151851 | 490 | if(dhcp_fsm[cli->state].timer_lease!= NULL){ |
daniele | 29:1a47b7151851 | 491 | dhcp_fsm[cli->state].timer_lease(cli, NULL, 0); |
daniele | 29:1a47b7151851 | 492 | } |
daniele | 29:1a47b7151851 | 493 | break; |
daniele | 29:1a47b7151851 | 494 | case PICO_DHCP_EVENT_RETRANSMIT: |
daniele | 29:1a47b7151851 | 495 | if(dhcp_fsm[cli->state].timer_retransmit!= NULL){ |
daniele | 29:1a47b7151851 | 496 | dhcp_fsm[cli->state].timer_retransmit(cli, NULL, 0); |
daniele | 29:1a47b7151851 | 497 | } |
daniele | 29:1a47b7151851 | 498 | break; |
daniele | 29:1a47b7151851 | 499 | default: |
daniele | 29:1a47b7151851 | 500 | dbg("DHCPC: event not supported yet!!\n"); |
daniele | 29:1a47b7151851 | 501 | break; |
daniele | 29:1a47b7151851 | 502 | } |
daniele | 29:1a47b7151851 | 503 | } |
daniele | 29:1a47b7151851 | 504 | |
daniele | 29:1a47b7151851 | 505 | |
daniele | 29:1a47b7151851 | 506 | /********************* |
daniele | 29:1a47b7151851 | 507 | * utility functions * |
daniele | 29:1a47b7151851 | 508 | *********************/ |
daniele | 29:1a47b7151851 | 509 | |
daniele | 29:1a47b7151851 | 510 | static void pico_dhcp_retry(struct pico_dhcp_client_cookie *cli) |
daniele | 29:1a47b7151851 | 511 | { |
daniele | 29:1a47b7151851 | 512 | const int MAX_RETRY = 3; |
daniele | 29:1a47b7151851 | 513 | uint32_t new_xid; |
daniele | 29:1a47b7151851 | 514 | if (++cli->attempt > MAX_RETRY) { |
daniele | 29:1a47b7151851 | 515 | cli->start_time = pico_tick; |
daniele | 29:1a47b7151851 | 516 | cli->attempt = 0; |
daniele | 29:1a47b7151851 | 517 | new_xid = pico_rand(); |
daniele | 29:1a47b7151851 | 518 | while(get_cookie_by_xid(new_xid) != NULL){ |
daniele | 29:1a47b7151851 | 519 | new_xid = pico_rand(); |
daniele | 29:1a47b7151851 | 520 | } |
daniele | 29:1a47b7151851 | 521 | cli->xid = new_xid; |
daniele | 29:1a47b7151851 | 522 | cli->state = DHCPSTATE_DISCOVER; |
daniele | 29:1a47b7151851 | 523 | } |
daniele | 29:1a47b7151851 | 524 | } |
daniele | 29:1a47b7151851 | 525 | |
daniele | 29:1a47b7151851 | 526 | static int dhclient_send(struct pico_dhcp_client_cookie *cli, uint8_t msg_type) |
daniele | 29:1a47b7151851 | 527 | { |
daniele | 29:1a47b7151851 | 528 | uint8_t buf_out[DHCPC_DATAGRAM_SIZE] = {0}; |
daniele | 29:1a47b7151851 | 529 | struct pico_dhcphdr *dh_out = (struct pico_dhcphdr *) buf_out; |
daniele | 29:1a47b7151851 | 530 | int sent = 0; |
daniele | 29:1a47b7151851 | 531 | int i = 0; |
daniele | 29:1a47b7151851 | 532 | struct pico_ip4 destination; |
daniele | 29:1a47b7151851 | 533 | uint16_t port = PICO_DHCPD_PORT; |
daniele | 29:1a47b7151851 | 534 | if(cli->state == DHCPSTATE_BOUND || cli->state == DHCPSTATE_RENEWING){ |
daniele | 29:1a47b7151851 | 535 | destination.addr = cli->server_id.addr; |
daniele | 29:1a47b7151851 | 536 | }else{ |
daniele | 29:1a47b7151851 | 537 | destination.addr = long_be(0xFFFFFFFF); |
daniele | 29:1a47b7151851 | 538 | } |
daniele | 29:1a47b7151851 | 539 | |
daniele | 29:1a47b7151851 | 540 | if(cli->device->eth == NULL){ |
daniele | 29:1a47b7151851 | 541 | pico_err = PICO_ERR_EOPNOTSUPP; |
daniele | 29:1a47b7151851 | 542 | if(cli->cb != NULL){ |
daniele | 29:1a47b7151851 | 543 | cli->cb(cli, PICO_DHCP_ERROR); |
daniele | 29:1a47b7151851 | 544 | } |
daniele | 29:1a47b7151851 | 545 | return -1; |
daniele | 29:1a47b7151851 | 546 | } |
daniele | 29:1a47b7151851 | 547 | memcpy(dh_out->hwaddr, &cli->device->eth->mac, PICO_HLEN_ETHER); |
daniele | 29:1a47b7151851 | 548 | dh_out->op = PICO_DHCP_OP_REQUEST; |
daniele | 29:1a47b7151851 | 549 | dh_out->htype = PICO_HTYPE_ETHER; |
daniele | 29:1a47b7151851 | 550 | dh_out->hlen = PICO_HLEN_ETHER; |
daniele | 29:1a47b7151851 | 551 | dh_out->xid = cli->xid; |
daniele | 29:1a47b7151851 | 552 | dh_out->secs = (msg_type == PICO_DHCP_MSG_REQUEST)?0:short_be((pico_tick - cli->start_time)/1000); |
daniele | 29:1a47b7151851 | 553 | dh_out->dhcp_magic = PICO_DHCPD_MAGIC_COOKIE; |
daniele | 29:1a47b7151851 | 554 | if (cli->state == DHCPSTATE_RENEWING) |
daniele | 29:1a47b7151851 | 555 | dh_out->ciaddr = cli->address.addr; |
daniele | 29:1a47b7151851 | 556 | |
daniele | 29:1a47b7151851 | 557 | /* Option: msg type, len 1 */ |
daniele | 29:1a47b7151851 | 558 | dh_out->options[i++] = PICO_DHCPOPT_MSGTYPE; |
daniele | 29:1a47b7151851 | 559 | dh_out->options[i++] = 1; |
daniele | 29:1a47b7151851 | 560 | dh_out->options[i++] = msg_type; |
daniele | 29:1a47b7151851 | 561 | |
daniele | 29:1a47b7151851 | 562 | if (( msg_type == PICO_DHCP_MSG_REQUEST) && ( cli->state != DHCPSTATE_RENEWING )) { |
daniele | 29:1a47b7151851 | 563 | dh_out->options[i++] = PICO_DHCPOPT_REQIP; |
daniele | 29:1a47b7151851 | 564 | dh_out->options[i++] = 4; |
daniele | 29:1a47b7151851 | 565 | dh_out->options[i++] = (long_be(cli->address.addr) & 0xFF000000) >> 24; |
daniele | 29:1a47b7151851 | 566 | dh_out->options[i++] = (long_be(cli->address.addr) & 0xFF0000) >> 16; |
daniele | 29:1a47b7151851 | 567 | dh_out->options[i++] = (long_be(cli->address.addr) & 0xFF00) >> 8; |
daniele | 29:1a47b7151851 | 568 | dh_out->options[i++] = (long_be(cli->address.addr) & 0xFF); |
daniele | 29:1a47b7151851 | 569 | dh_out->options[i++] = PICO_DHCPOPT_SERVERID; |
daniele | 29:1a47b7151851 | 570 | dh_out->options[i++] = 4; |
daniele | 29:1a47b7151851 | 571 | dh_out->options[i++] = (long_be(cli->server_id.addr) & 0xFF000000) >> 24; |
daniele | 29:1a47b7151851 | 572 | dh_out->options[i++] = (long_be(cli->server_id.addr) & 0xFF0000) >> 16; |
daniele | 29:1a47b7151851 | 573 | dh_out->options[i++] = (long_be(cli->server_id.addr) & 0xFF00) >> 8; |
daniele | 29:1a47b7151851 | 574 | dh_out->options[i++] = (long_be(cli->server_id.addr) & 0xFF); |
daniele | 29:1a47b7151851 | 575 | } |
daniele | 29:1a47b7151851 | 576 | |
daniele | 29:1a47b7151851 | 577 | /* Option: req list, len 4 */ |
daniele | 29:1a47b7151851 | 578 | dh_out->options[i++] = PICO_DHCPOPT_PARMLIST; |
daniele | 29:1a47b7151851 | 579 | dh_out->options[i++] = 7; |
daniele | 29:1a47b7151851 | 580 | dh_out->options[i++] = PICO_DHCPOPT_NETMASK; |
daniele | 29:1a47b7151851 | 581 | dh_out->options[i++] = PICO_DHCPOPT_BCAST; |
daniele | 29:1a47b7151851 | 582 | dh_out->options[i++] = PICO_DHCPOPT_TIME; |
daniele | 29:1a47b7151851 | 583 | dh_out->options[i++] = PICO_DHCPOPT_ROUTER; |
daniele | 29:1a47b7151851 | 584 | dh_out->options[i++] = PICO_DHCPOPT_HOSTNAME; |
daniele | 29:1a47b7151851 | 585 | dh_out->options[i++] = PICO_DHCPOPT_RENEWALTIME; |
daniele | 29:1a47b7151851 | 586 | dh_out->options[i++] = PICO_DHCPOPT_REBINDINGTIME; |
daniele | 29:1a47b7151851 | 587 | |
daniele | 29:1a47b7151851 | 588 | /* Option : max message size */ |
daniele | 29:1a47b7151851 | 589 | if( msg_type == PICO_DHCP_MSG_REQUEST || msg_type == PICO_DHCP_MSG_DISCOVER){ |
daniele | 29:1a47b7151851 | 590 | uint16_t dds = DHCPC_DATAGRAM_SIZE; |
daniele | 29:1a47b7151851 | 591 | dh_out->options[i++] = PICO_DHCPOPT_MAXMSGSIZE; |
daniele | 29:1a47b7151851 | 592 | dh_out->options[i++] = 2; |
daniele | 29:1a47b7151851 | 593 | dh_out->options[i++] = (dds & 0xFF00) >> 8; |
daniele | 29:1a47b7151851 | 594 | dh_out->options[i++] = (dds & 0xFF); |
daniele | 29:1a47b7151851 | 595 | } |
daniele | 29:1a47b7151851 | 596 | |
daniele | 29:1a47b7151851 | 597 | |
daniele | 29:1a47b7151851 | 598 | |
daniele | 29:1a47b7151851 | 599 | dh_out->options[i] = PICO_DHCPOPT_END; |
daniele | 29:1a47b7151851 | 600 | sent = pico_socket_sendto(cli->socket, buf_out, DHCPC_DATAGRAM_SIZE, &destination, port); |
daniele | 29:1a47b7151851 | 601 | if (sent < 0) { |
daniele | 29:1a47b7151851 | 602 | dbg("DHCPC: sendto failed: %s\n", strerror(pico_err)); |
daniele | 29:1a47b7151851 | 603 | if(cli->cb != NULL) |
daniele | 29:1a47b7151851 | 604 | cli->cb(cli, PICO_DHCP_ERROR); |
daniele | 29:1a47b7151851 | 605 | } |
daniele | 29:1a47b7151851 | 606 | |
daniele | 29:1a47b7151851 | 607 | |
daniele | 29:1a47b7151851 | 608 | //resend-timer : |
daniele | 29:1a47b7151851 | 609 | if(cli->timer_param_retransmit != NULL) |
daniele | 29:1a47b7151851 | 610 | cli->timer_param_retransmit->valid=0; |
daniele | 29:1a47b7151851 | 611 | |
daniele | 29:1a47b7151851 | 612 | cli->timer_param_retransmit = pico_zalloc(sizeof(struct dhcp_timer_param)); |
daniele | 29:1a47b7151851 | 613 | if(!cli->timer_param_retransmit){ |
daniele | 29:1a47b7151851 | 614 | if(cli->cb != NULL) |
daniele | 29:1a47b7151851 | 615 | pico_err = PICO_ERR_ENOMEM; |
daniele | 29:1a47b7151851 | 616 | cli->cb(cli, PICO_DHCP_ERROR); |
daniele | 29:1a47b7151851 | 617 | return -1; |
daniele | 29:1a47b7151851 | 618 | } |
daniele | 29:1a47b7151851 | 619 | cli->timer_param_retransmit->valid = 1; |
daniele | 29:1a47b7151851 | 620 | cli->timer_param_retransmit->cli = cli; |
daniele | 29:1a47b7151851 | 621 | cli->timer_param_retransmit->type = PICO_DHCP_EVENT_RETRANSMIT; |
daniele | 29:1a47b7151851 | 622 | pico_timer_add(4000, dhcp_timer_cb, cli->timer_param_retransmit); |
daniele | 29:1a47b7151851 | 623 | |
daniele | 29:1a47b7151851 | 624 | return 0; |
daniele | 29:1a47b7151851 | 625 | } |
daniele | 29:1a47b7151851 | 626 | |
daniele | 29:1a47b7151851 | 627 | //identifies type & does some preprocessing : checking if everything is valid |
daniele | 29:1a47b7151851 | 628 | static int pico_dhcp_verify_and_identify_type(uint8_t* data, int len, struct pico_dhcp_client_cookie *cli) |
daniele | 29:1a47b7151851 | 629 | { |
daniele | 29:1a47b7151851 | 630 | struct pico_dhcphdr *dhdr = (struct pico_dhcphdr *) data; |
daniele | 29:1a47b7151851 | 631 | uint8_t *nextopt, opt_data[20], opt_type; |
daniele | 29:1a47b7151851 | 632 | int opt_len = 20; |
daniele | 29:1a47b7151851 | 633 | |
daniele | 29:1a47b7151851 | 634 | if (dhdr->xid != cli->xid) |
daniele | 29:1a47b7151851 | 635 | return 0; |
daniele | 29:1a47b7151851 | 636 | |
daniele | 29:1a47b7151851 | 637 | if (!is_options_valid(dhdr->options, len - sizeof(struct pico_dhcphdr))) |
daniele | 29:1a47b7151851 | 638 | return 0; |
daniele | 29:1a47b7151851 | 639 | |
daniele | 29:1a47b7151851 | 640 | if( dhdr->dhcp_magic != PICO_DHCPD_MAGIC_COOKIE) |
daniele | 29:1a47b7151851 | 641 | return 0; |
daniele | 29:1a47b7151851 | 642 | |
daniele | 29:1a47b7151851 | 643 | opt_type = dhcp_get_next_option(dhdr->options, opt_data, &opt_len, &nextopt); |
daniele | 29:1a47b7151851 | 644 | while (opt_type != PICO_DHCPOPT_END) { |
daniele | 29:1a47b7151851 | 645 | /* parse interesting options here */ |
daniele | 29:1a47b7151851 | 646 | if (opt_type == PICO_DHCPOPT_MSGTYPE) { |
daniele | 29:1a47b7151851 | 647 | return *opt_data; |
daniele | 29:1a47b7151851 | 648 | } |
daniele | 29:1a47b7151851 | 649 | opt_len = 20; |
daniele | 29:1a47b7151851 | 650 | opt_type = dhcp_get_next_option(NULL, opt_data, &opt_len, &nextopt); |
daniele | 29:1a47b7151851 | 651 | } |
daniele | 29:1a47b7151851 | 652 | return 0; |
daniele | 29:1a47b7151851 | 653 | |
daniele | 29:1a47b7151851 | 654 | } |
daniele | 29:1a47b7151851 | 655 | |
daniele | 29:1a47b7151851 | 656 | static int init_cookie(struct pico_dhcp_client_cookie* cli) |
daniele | 29:1a47b7151851 | 657 | { |
daniele | 29:1a47b7151851 | 658 | uint8_t n = 3; |
daniele | 29:1a47b7151851 | 659 | uint16_t port = PICO_DHCP_CLIENT_PORT; |
daniele | 29:1a47b7151851 | 660 | struct pico_ip4 address, netmask; |
daniele | 29:1a47b7151851 | 661 | |
daniele | 29:1a47b7151851 | 662 | address.addr = long_be(0x00000000); |
daniele | 29:1a47b7151851 | 663 | netmask.addr = long_be(0x00000000); |
daniele | 29:1a47b7151851 | 664 | |
daniele | 29:1a47b7151851 | 665 | cli->state = DHCPSTATE_DISCOVER; |
daniele | 29:1a47b7151851 | 666 | cli->start_time = pico_tick; |
daniele | 29:1a47b7151851 | 667 | cli->attempt = 0; |
daniele | 29:1a47b7151851 | 668 | |
daniele | 29:1a47b7151851 | 669 | cli->socket = pico_socket_open(PICO_PROTO_IPV4, PICO_PROTO_UDP, &pico_dhcp_wakeup); |
daniele | 29:1a47b7151851 | 670 | if (!cli->socket) { |
daniele | 29:1a47b7151851 | 671 | dbg("DHCPC: error opening socket: %s\n", strerror(pico_err)); |
daniele | 29:1a47b7151851 | 672 | if(cli->cb != NULL) |
daniele | 29:1a47b7151851 | 673 | cli->cb(cli, PICO_DHCP_ERROR); |
daniele | 29:1a47b7151851 | 674 | return -1; |
daniele | 29:1a47b7151851 | 675 | } |
daniele | 29:1a47b7151851 | 676 | if (pico_socket_bind(cli->socket, &address, &port) != 0){ |
daniele | 29:1a47b7151851 | 677 | dbg("DHCPC: error binding socket: %s\n", strerror(pico_err)); |
daniele | 29:1a47b7151851 | 678 | pico_socket_close(cli->socket); |
daniele | 29:1a47b7151851 | 679 | if(cli->cb != NULL) |
daniele | 29:1a47b7151851 | 680 | cli->cb(cli, PICO_DHCP_ERROR); |
daniele | 29:1a47b7151851 | 681 | return -1; |
daniele | 29:1a47b7151851 | 682 | } |
daniele | 29:1a47b7151851 | 683 | cli->socket->dev = cli->device; |
daniele | 29:1a47b7151851 | 684 | |
daniele | 29:1a47b7151851 | 685 | if(pico_ipv4_link_add(cli->device, address, netmask) != 0){ |
daniele | 29:1a47b7151851 | 686 | dbg("DHCPC: error adding link: %s\n", strerror(pico_err)); |
daniele | 29:1a47b7151851 | 687 | if(cli->cb != NULL) |
daniele | 29:1a47b7151851 | 688 | cli->cb(cli, PICO_DHCP_ERROR); |
daniele | 29:1a47b7151851 | 689 | return -1; |
daniele | 29:1a47b7151851 | 690 | } |
daniele | 29:1a47b7151851 | 691 | |
daniele | 29:1a47b7151851 | 692 | /* attempt to generate a correct xid 3 times, then fail */ |
daniele | 29:1a47b7151851 | 693 | do { |
daniele | 29:1a47b7151851 | 694 | cli->xid = pico_rand(); |
daniele | 29:1a47b7151851 | 695 | } while (!cli->xid && --n); |
daniele | 29:1a47b7151851 | 696 | if (!cli->xid) { |
daniele | 29:1a47b7151851 | 697 | if(cli->cb != NULL) |
daniele | 29:1a47b7151851 | 698 | cli->cb(cli, PICO_DHCP_ERROR); |
daniele | 29:1a47b7151851 | 699 | return -1; |
daniele | 29:1a47b7151851 | 700 | } |
daniele | 29:1a47b7151851 | 701 | |
daniele | 29:1a47b7151851 | 702 | return 0; |
daniele | 29:1a47b7151851 | 703 | } |
daniele | 29:1a47b7151851 | 704 | |
daniele | 29:1a47b7151851 | 705 | static struct pico_dhcp_client_cookie *get_cookie_by_xid(uint32_t xid) |
daniele | 29:1a47b7151851 | 706 | { |
daniele | 29:1a47b7151851 | 707 | struct pico_dhcp_client_cookie test = { }, *cookie = NULL; |
daniele | 29:1a47b7151851 | 708 | |
daniele | 29:1a47b7151851 | 709 | if (!xid) |
daniele | 29:1a47b7151851 | 710 | return NULL; |
daniele | 29:1a47b7151851 | 711 | |
daniele | 29:1a47b7151851 | 712 | test.xid = xid; |
daniele | 29:1a47b7151851 | 713 | cookie = pico_tree_findKey(&DHCPCookies, &test); |
daniele | 29:1a47b7151851 | 714 | if (!cookie) |
daniele | 29:1a47b7151851 | 715 | return NULL; |
daniele | 29:1a47b7151851 | 716 | else |
daniele | 29:1a47b7151851 | 717 | return cookie; |
daniele | 29:1a47b7151851 | 718 | } |
daniele | 29:1a47b7151851 | 719 | |
daniele | 29:1a47b7151851 | 720 | void *pico_dhcp_get_identifier(uint32_t xid) |
daniele | 29:1a47b7151851 | 721 | { |
daniele | 29:1a47b7151851 | 722 | return (void *) get_cookie_by_xid(xid); |
daniele | 29:1a47b7151851 | 723 | } |
daniele | 29:1a47b7151851 | 724 | |
daniele | 29:1a47b7151851 | 725 | static uint32_t get_xid(uint8_t* data) |
daniele | 29:1a47b7151851 | 726 | { |
daniele | 29:1a47b7151851 | 727 | struct pico_dhcphdr *dhdr = (struct pico_dhcphdr *) data; |
daniele | 29:1a47b7151851 | 728 | return dhdr->xid; |
daniele | 29:1a47b7151851 | 729 | } |
daniele | 29:1a47b7151851 | 730 | |
daniele | 29:1a47b7151851 | 731 | #endif |