/* * File : dhcp_server_raw.c * A simple DHCP server implementation * * This file is part of RT-Thread RTOS * COPYRIGHT (C) 2011-2018, Shanghai Real-Thread Technology Co., Ltd * http://www.rt-thread.com * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License along * with this program; if not, write to the Free Software Foundation, Inc., * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. * * Change Logs: * Date Author Notes * 2014-04-01 Ren.Haibo the first version * 2018-06-12 aozima ignore DHCP_OPTION_SERVER_ID. */ #include #include #include #include #include #include #include #include #include #include #if (LWIP_VERSION) < 0x02000000U #error "not support old LWIP" #endif #if !LWIP_IPV4 #error "must enable IPV4" #endif #if (LWIP_VERSION) >= 0x02000000U #include #endif /* DHCP server option */ /* allocated client ip range */ #ifndef DHCPD_CLIENT_IP_MIN #define DHCPD_CLIENT_IP_MIN 2 #endif #ifndef DHCPD_CLIENT_IP_MAX #define DHCPD_CLIENT_IP_MAX 254 #endif /* the DHCP server address */ #ifndef DHCPD_SERVER_IP #define DHCPD_SERVER_IP "192.168.169.1" #endif #define DHCP_DEBUG_PRINTF #ifdef DHCP_DEBUG_PRINTF #define DEBUG_PRINTF rt_kprintf("[DHCP] "); rt_kprintf #else #define DEBUG_PRINTF(...) #endif /* DHCP_DEBUG_PRINTF */ /* we need some routines in the DHCP of lwIP */ #undef LWIP_DHCP #define LWIP_DHCP 1 #include /** Mac address length */ #define DHCP_MAX_HLEN 6 /** dhcp default live time */ #define DHCP_DEFAULT_LIVE_TIME 0x80510100 /** Minimum length for request before packet is parsed */ #define DHCP_MIN_REQUEST_LEN 44 #define LWIP_NETIF_LOCK(...) #define LWIP_NETIF_UNLOCK(...) /** * The dhcp client node struct. */ struct dhcp_client_node { struct dhcp_client_node *next; u8_t chaddr[DHCP_MAX_HLEN]; ip4_addr_t ipaddr; u32_t lease_end; }; /** * The dhcp server struct. */ struct dhcp_server { struct dhcp_server *next; struct netif *netif; struct udp_pcb *pcb; struct dhcp_client_node *node_list; ip4_addr_t start; ip4_addr_t end; ip4_addr_t current; }; static u8_t *dhcp_server_option_find(u8_t *buf, u16_t len, u8_t option); /** * The dhcp server struct list. */ static struct dhcp_server *lw_dhcp_server; /** * Find a dhcp client node by mac address * * @param dhcpserver The dhcp server * @param chaddr Mac address * @param hlen Mac address length * @return dhcp client node */ static struct dhcp_client_node * dhcp_client_find_by_mac(struct dhcp_server *dhcpserver, const u8_t *chaddr, u8_t hlen) { struct dhcp_client_node *node; for (node = dhcpserver->node_list; node != NULL; node = node->next) { if (memcmp(node->chaddr, chaddr, hlen) == 0) { return node; } } return NULL; } /** * Find a dhcp client node by ip address * * @param dhcpserver The dhcp server * @param chaddr Mac address * @param hlen Mac address length * @return dhcp client node */ static struct dhcp_client_node * dhcp_client_find_by_ip(struct dhcp_server *dhcpserver, const ip4_addr_t *ip) { struct dhcp_client_node *node; for (node = dhcpserver->node_list; node != NULL; node = node->next) { if (ip4_addr_cmp(&node->ipaddr, ip)) { return node; } } return NULL; } /** * Find a dhcp client node by ip address * * @param dhcpserver The dhcp server * @param chaddr Mac address * @param hlen Mac address length * @return dhcp client node */ static struct dhcp_client_node * dhcp_client_find(struct dhcp_server *dhcpserver, struct dhcp_msg *msg, u8_t *opt_buf, u16_t len) { u8_t *opt; //u32_t ipaddr; struct dhcp_client_node *node; node = dhcp_client_find_by_mac(dhcpserver, msg->chaddr, msg->hlen); if (node != NULL) { return node; } opt = dhcp_server_option_find(opt_buf, len, DHCP_OPTION_REQUESTED_IP); if (opt != NULL) { node = dhcp_client_find_by_ip(dhcpserver, (ip4_addr_t *)(&opt[2])); if (node != NULL) { return node; } } return NULL; } /** * Find a dhcp client node by ip address * * @param dhcpserver The dhcp server * @param chaddr Mac address * @param hlen Mac address length * @return dhcp client node */ static struct dhcp_client_node * dhcp_client_alloc(struct dhcp_server *dhcpserver, struct dhcp_msg *msg, u8_t *opt_buf, u16_t len) { u8_t *opt; u32_t ipaddr; struct dhcp_client_node *node; node = dhcp_client_find_by_mac(dhcpserver, msg->chaddr, msg->hlen); if (node != NULL) { return node; } opt = dhcp_server_option_find(opt_buf, len, DHCP_OPTION_REQUESTED_IP); if (opt != NULL) { node = dhcp_client_find_by_ip(dhcpserver, (ip4_addr_t *)(&opt[2])); if (node != NULL) { return node; } } dhcp_alloc_again: node = dhcp_client_find_by_ip(dhcpserver, &dhcpserver->current); if (node != NULL) { ipaddr = (ntohl(dhcpserver->current.addr) + 1); if (ipaddr > ntohl(dhcpserver->end.addr)) { ipaddr = ntohl(dhcpserver->start.addr); } dhcpserver->current.addr = htonl(ipaddr); goto dhcp_alloc_again; } node = (struct dhcp_client_node *)mem_malloc(sizeof(struct dhcp_client_node)); if (node == NULL) { return NULL; } SMEMCPY(node->chaddr, msg->chaddr, msg->hlen); node->ipaddr = dhcpserver->current; node->next = dhcpserver->node_list; dhcpserver->node_list = node; return node; } /** * find option from buffer. * * @param buf The buffer to find option * @param len The buffer length * @param option Which option to find * @return dhcp option buffer */ static u8_t * dhcp_server_option_find(u8_t *buf, u16_t len, u8_t option) { u8_t *end = buf + len; while ((buf < end) && (*buf != DHCP_OPTION_END)) { if (*buf == option) { return buf; } buf += (buf[1] + 2); } return NULL; } /** * If an incoming DHCP message is in response to us, then trigger the state machine */ static void dhcp_server_recv(void *arg, struct udp_pcb *pcb, struct pbuf *p, const ip_addr_t *recv_addr, u16_t port) { struct dhcp_server *dhcp_server = (struct dhcp_server *)arg; struct dhcp_msg *msg; struct pbuf *q; u8_t *opt_buf; u8_t *opt; struct dhcp_client_node *node; u8_t msg_type; u16_t length; ip_addr_t addr = *recv_addr; u32_t tmp; LWIP_DEBUGF(DHCP_DEBUG | LWIP_DBG_TRACE, ("[%s:%d] %c%c recv %d\n", __FUNCTION__, __LINE__, dhcp_server->netif->name[0], dhcp_server->netif->name[1], p->tot_len)); /* prevent warnings about unused arguments */ LWIP_UNUSED_ARG(pcb); LWIP_UNUSED_ARG(addr); LWIP_UNUSED_ARG(port); if (p->len < DHCP_MIN_REQUEST_LEN) { LWIP_DEBUGF(DHCP_DEBUG | LWIP_DBG_TRACE | LWIP_DBG_LEVEL_WARNING, ("DHCP request message or pbuf too short\n")); pbuf_free(p); return; } q = pbuf_alloc(PBUF_TRANSPORT, 1500, PBUF_RAM); if (q == NULL) { LWIP_DEBUGF(DHCP_DEBUG | LWIP_DBG_TRACE | LWIP_DBG_LEVEL_WARNING, ("pbuf_alloc dhcp_msg failed!\n")); pbuf_free(p); return; } if (q->tot_len < p->tot_len) { LWIP_DEBUGF(DHCP_DEBUG | LWIP_DBG_TRACE | LWIP_DBG_LEVEL_WARNING, ("pbuf_alloc dhcp_msg too small %d:%d\n", q->tot_len, p->tot_len)); pbuf_free(p); return; } pbuf_copy(q, p); pbuf_free(p); msg = (struct dhcp_msg *)q->payload; if (msg->op != DHCP_BOOTREQUEST) { LWIP_DEBUGF(DHCP_DEBUG | LWIP_DBG_TRACE | LWIP_DBG_LEVEL_WARNING, ("not a DHCP request message, but type %"U16_F"\n", (u16_t)msg->op)); goto free_pbuf_and_return; } if (msg->cookie != PP_HTONL(DHCP_MAGIC_COOKIE)) { LWIP_DEBUGF(DHCP_DEBUG | LWIP_DBG_TRACE | LWIP_DBG_LEVEL_WARNING, ("bad DHCP_MAGIC_COOKIE!\n")); goto free_pbuf_and_return; } if (msg->hlen > DHCP_MAX_HLEN) { goto free_pbuf_and_return; } opt_buf = (u8_t *)msg + DHCP_OPTIONS_OFS; length = q->tot_len - DHCP_OPTIONS_OFS; opt = dhcp_server_option_find(opt_buf, length, DHCP_OPTION_MESSAGE_TYPE); if (opt) { msg_type = *(opt + 2); if (msg_type == DHCP_DISCOVER) { node = dhcp_client_alloc(dhcp_server, msg, opt_buf, length); if (node == NULL) { goto free_pbuf_and_return; } node->lease_end = DHCP_DEFAULT_LIVE_TIME; /* create dhcp offer and send */ msg->op = DHCP_BOOTREPLY; msg->hops = 0; msg->secs = 0; SMEMCPY(&msg->siaddr, &(dhcp_server->netif->ip_addr), 4); msg->sname[0] = '\0'; msg->file[0] = '\0'; msg->cookie = PP_HTONL(DHCP_MAGIC_COOKIE); SMEMCPY(&msg->yiaddr, &node->ipaddr, 4); opt_buf = (u8_t *)msg + DHCP_OPTIONS_OFS; /* add msg type */ *opt_buf++ = DHCP_OPTION_MESSAGE_TYPE; *opt_buf++ = 1; *opt_buf++ = DHCP_OFFER; /* add server id */ *opt_buf++ = DHCP_OPTION_SERVER_ID; *opt_buf++ = 4; SMEMCPY(opt_buf, &(dhcp_server->netif->ip_addr), 4); opt_buf += 4; /* add_lease_time */ *opt_buf++ = DHCP_OPTION_LEASE_TIME; *opt_buf++ = 4; tmp = PP_HTONL(DHCP_DEFAULT_LIVE_TIME); SMEMCPY(opt_buf, &tmp, 4); opt_buf += 4; /* add config */ *opt_buf++ = DHCP_OPTION_SUBNET_MASK; *opt_buf++ = 4; SMEMCPY(opt_buf, &ip_2_ip4(&dhcp_server->netif->netmask)->addr, 4); opt_buf += 4; *opt_buf++ = DHCP_OPTION_DNS_SERVER; *opt_buf++ = 4; #ifdef DHCP_DNS_SERVER_IP { ip_addr_t dns_addr; ipaddr_aton(DHCP_DNS_SERVER_IP, &dns_addr); SMEMCPY(opt_buf, &ip_2_ip4(&dns_addr)->addr, 4); } #else /* default use gatewary dns server */ SMEMCPY(opt_buf, &(dhcp_server->netif->ip_addr), 4); #endif /* DHCP_DNS_SERVER_IP */ opt_buf += 4; *opt_buf++ = DHCP_OPTION_ROUTER; *opt_buf++ = 4; SMEMCPY(opt_buf, &ip_2_ip4(&dhcp_server->netif->ip_addr)->addr, 4); opt_buf += 4; /* add option end */ *opt_buf++ = DHCP_OPTION_END; length = (u32_t)opt_buf - (u32_t)msg; if (length < q->tot_len) { pbuf_realloc(q, length); } ip_2_ip4(&addr)->addr = INADDR_BROADCAST; udp_sendto_if(pcb, q, &addr, port, dhcp_server->netif); } else { if (1) { if (msg_type == DHCP_REQUEST) { node = dhcp_client_find(dhcp_server, msg, opt_buf, length); if (node != NULL) { /* Send ack */ node->lease_end = DHCP_DEFAULT_LIVE_TIME; /* create dhcp offer and send */ msg->op = DHCP_BOOTREPLY; msg->hops = 0; msg->secs = 0; SMEMCPY(&msg->siaddr, &(dhcp_server->netif->ip_addr), 4); msg->sname[0] = '\0'; msg->file[0] = '\0'; msg->cookie = PP_HTONL(DHCP_MAGIC_COOKIE); SMEMCPY(&msg->yiaddr, &node->ipaddr, 4); opt_buf = (u8_t *)msg + DHCP_OPTIONS_OFS; /* add msg type */ *opt_buf++ = DHCP_OPTION_MESSAGE_TYPE; *opt_buf++ = 1; *opt_buf++ = DHCP_ACK; /* add server id */ *opt_buf++ = DHCP_OPTION_SERVER_ID; *opt_buf++ = 4; SMEMCPY(opt_buf, &(dhcp_server->netif->ip_addr), 4); opt_buf += 4; /* add_lease_time */ *opt_buf++ = DHCP_OPTION_LEASE_TIME; *opt_buf++ = 4; tmp = PP_HTONL(DHCP_DEFAULT_LIVE_TIME); SMEMCPY(opt_buf, &tmp, 4); opt_buf += 4; /* add config */ *opt_buf++ = DHCP_OPTION_SUBNET_MASK; *opt_buf++ = 4; SMEMCPY(opt_buf, &ip_2_ip4(&dhcp_server->netif->netmask)->addr, 4); opt_buf += 4; *opt_buf++ = DHCP_OPTION_DNS_SERVER; *opt_buf++ = 4; #ifdef DHCP_DNS_SERVER_IP { ip_addr_t dns_addr; ipaddr_aton(DHCP_DNS_SERVER_IP, &dns_addr); SMEMCPY(opt_buf, &ip_2_ip4(&dns_addr)->addr, 4); } #else /* default use gatewary dns server */ SMEMCPY(opt_buf, &(dhcp_server->netif->ip_addr), 4); #endif /* DHCP_DNS_SERVER_IP */ opt_buf += 4; *opt_buf++ = DHCP_OPTION_ROUTER; *opt_buf++ = 4; SMEMCPY(opt_buf, &ip_2_ip4(&dhcp_server->netif->ip_addr)->addr, 4); opt_buf += 4; /* add option end */ *opt_buf++ = DHCP_OPTION_END; length = (u32_t)opt_buf - (u32_t)msg; if (length < q->tot_len) { pbuf_realloc(q, length); } ip_2_ip4(&addr)->addr = INADDR_BROADCAST; udp_sendto_if(pcb, q, &addr, port, dhcp_server->netif); } else { /* Send no ack */ /* create dhcp offer and send */ msg->op = DHCP_BOOTREPLY; msg->hops = 0; msg->secs = 0; SMEMCPY(&msg->siaddr, &(dhcp_server->netif->ip_addr), 4); msg->sname[0] = '\0'; msg->file[0] = '\0'; msg->cookie = PP_HTONL(DHCP_MAGIC_COOKIE); memset(&msg->yiaddr, 0, 4); opt_buf = (u8_t *)msg + DHCP_OPTIONS_OFS; /* add msg type */ *opt_buf++ = DHCP_OPTION_MESSAGE_TYPE; *opt_buf++ = 1; *opt_buf++ = DHCP_NAK; /* add server id */ *opt_buf++ = DHCP_OPTION_SERVER_ID; *opt_buf++ = 4; SMEMCPY(opt_buf, &(dhcp_server->netif->ip_addr), 4); opt_buf += 4; /* add option end */ *opt_buf++ = DHCP_OPTION_END; length = (u32_t)opt_buf - (u32_t)msg; if (length < q->tot_len) { pbuf_realloc(q, length); } ip_2_ip4(&addr)->addr = INADDR_BROADCAST; udp_sendto_if(pcb, q, &addr, port, dhcp_server->netif); } } else if (msg_type == DHCP_RELEASE) { struct dhcp_client_node *node_prev = NULL; for (node = dhcp_server->node_list; node != NULL; node = node->next) { if (memcmp(node->chaddr, msg->chaddr, msg->hlen) == 0) { if (node == dhcp_server->node_list) { dhcp_server->node_list = node->next; } else { node_prev->next = node->next; } break; } node_prev = node; node = node->next; } if (node != NULL) { mem_free(node); } } else if (msg_type == DHCP_DECLINE) { ; } else if (msg_type == DHCP_INFORM) { ; } } } } free_pbuf_and_return: pbuf_free(q); } /** * start dhcp server for a netif * * @param netif The netif which use dhcp server * @param start The Start IP address * @param end The netif which use dhcp server * @return lwIP error code * - ERR_OK - No error * - ERR_MEM - Out of memory */ err_t dhcp_server_start(struct netif *netif, ip4_addr_t *start, ip4_addr_t *end) { struct dhcp_server *dhcp_server; /* If this netif alreday use the dhcp server. */ for (dhcp_server = lw_dhcp_server; dhcp_server != NULL; dhcp_server = dhcp_server->next) { if (dhcp_server->netif == netif) { dhcp_server->start = *start; dhcp_server->end = *end; dhcp_server->current = *start; return ERR_OK; } } dhcp_server = NULL; LWIP_DEBUGF(DHCP_DEBUG | LWIP_DBG_TRACE, ("dhcp_server_start(): starting new DHCP server\n")); dhcp_server = (struct dhcp_server *)mem_malloc(sizeof(struct dhcp_server)); if (dhcp_server == NULL) { LWIP_DEBUGF(DHCP_DEBUG | LWIP_DBG_TRACE, ("dhcp_server_start(): could not allocate dhcp\n")); return ERR_MEM; } /* clear data structure */ memset(dhcp_server, 0, sizeof(struct dhcp_server)); /* store this dhcp server to list */ dhcp_server->next = lw_dhcp_server; lw_dhcp_server = dhcp_server; dhcp_server->netif = netif; dhcp_server->node_list = NULL; dhcp_server->start = *start; dhcp_server->end = *end; dhcp_server->current = *start; /* allocate UDP PCB */ dhcp_server->pcb = udp_new(); if (dhcp_server->pcb == NULL) { LWIP_DEBUGF(DHCP_DEBUG | LWIP_DBG_TRACE, ("dhcp_server_start(): could not obtain pcb\n")); return ERR_MEM; } ip_set_option(dhcp_server->pcb, SOF_BROADCAST); /* set up local and remote port for the pcb */ udp_bind(dhcp_server->pcb, IP_ADDR_ANY, DHCP_SERVER_PORT); //udp_connect(dhcp_server->pcb, IP_ADDR_ANY, DHCP_CLIENT_PORT); /* set up the recv callback and argument */ udp_recv(dhcp_server->pcb, dhcp_server_recv, dhcp_server); LWIP_DEBUGF(DHCP_DEBUG | LWIP_DBG_TRACE, ("dhcp_server_start(): starting DHCP server\n")); return ERR_OK; } void dhcpd_start(const char *netif_name) { struct netif *netif = netif_list; err_t res; DEBUG_PRINTF("%s: %s\r\n", __FUNCTION__, netif_name); LWIP_NETIF_LOCK(); if (strlen(netif_name) > sizeof(netif->name)) { DEBUG_PRINTF("network interface name too long!\r\n"); goto _exit; } while (netif != RT_NULL) { if (strncmp(netif_name, netif->name, sizeof(netif->name)) == 0) break; netif = netif->next; if (netif == RT_NULL) { DEBUG_PRINTF("network interface: %s not found!\r\n", netif_name); break; } } if (netif == RT_NULL) { goto _exit; } if (1) { extern void set_if(const char *netif_name, const char *ip_addr, const char *gw_addr, const char *nm_addr); dhcp_stop(netif); set_if(netif_name, DHCPD_SERVER_IP, "0.0.0.0", "255.255.255.0"); netif_set_up(netif); } { char str_tmp[4 * 4 + 4] = DHCPD_SERVER_IP; char *p = str_tmp; ip4_addr_t ip_start, ip_end; p = strchr(str_tmp, '.'); if (p) { p = strchr(p + 1, '.'); if (p) { p = strchr(p + 1, '.'); } } if (!p) { DEBUG_PRINTF("DHCPD_SERVER_IP: %s error!\r\n", str_tmp); goto _exit; } p = p + 1; /* move to xxx.xxx.xxx.^ */ sprintf(p, "%d", DHCPD_CLIENT_IP_MIN); ip4addr_aton(str_tmp, &ip_start); DEBUG_PRINTF("ip_start: [%s]\r\n", str_tmp); sprintf(p, "%d", DHCPD_CLIENT_IP_MAX); ip4addr_aton(str_tmp, &ip_end); DEBUG_PRINTF("ip_start: [%s]\r\n", str_tmp); res = dhcp_server_start(netif, &ip_start, &ip_end); if (res != 0) { DEBUG_PRINTF("dhcp_server_start res: %s.\r\n", res); } } _exit: LWIP_NETIF_UNLOCK(); return; }