static vsf_err_t vsfip_dhcpc_init_msg(struct vsfip_dhcpc_t *dhcpc, uint8_t op) { struct vsfip_netif_t *netif = dhcpc->netif; struct vsfip_buffer_t *buf; struct vsfip_dhcphead_t *head; dhcpc->outbuffer = VSFIP_UDPBUF_GET(sizeof(struct vsfip_dhcphead_t)); if (NULL == dhcpc->outbuffer) { return VSFERR_FAIL; } buf = dhcpc->outbuffer; head = (struct vsfip_dhcphead_t *)buf->app.buffer; memset(head, 0, sizeof(struct vsfip_dhcphead_t)); head->op = DHCP_TOSERVER; head->htype = netif->drv->hwtype; head->hlen = netif->macaddr.size; head->xid = dhcpc->xid; // shift right 10-bit for div 1000 head->secs = 0; memcpy(head->chaddr, netif->macaddr.addr.s_addr_buf, netif->macaddr.size); head->magic = SYS_TO_BE_U32(DHCP_MAGIC); dhcpc->optlen = 0; vsfip_dhcp_append_opt(buf, &dhcpc->optlen, DHCPOPT_MSGTYPE, DHCPOPT_MSGTYPE_LEN, &op); { uint16_t tmp16 = SYS_TO_BE_U16(576); // RFC 2132 9.10, message size MUST be >= 576 vsfip_dhcp_append_opt(buf, &dhcpc->optlen, DHCPOPT_MAXMSGSIZE, DHCPOPT_MAXMSGSIZE_LEN, (uint8_t *)&tmp16); } { uint8_t requestlist[] = {DHCPOPT_SUBNETMASK, DHCPOPT_ROUTER, DHCPOPT_DNSSERVER, DHCPOPT_BROADCAST}; vsfip_dhcp_append_opt(buf, &dhcpc->optlen, DHCPOPT_PARAMLIST, sizeof(requestlist), requestlist); } #ifdef VSFIP_CFG_HOSTNAME vsfip_dhcp_append_opt(buf, &dhcpc->optlen, DHCPOPT_HOSTNAME, strlen(VSFIP_CFG_HOSTNAME), (uint8_t *)VSFIP_CFG_HOSTNAME); #endif return VSFERR_NONE; }
static struct vsfip_ipmac_assoc* vsfip_dhcpd_get_assoc(struct vsfip_netif_t *netif, uint8_t *hwaddr) { struct vsfip_dhcpd_t *dhcpd = netif->dhcp.dhcpd; uint8_t macsize = netif->macaddr.size; struct vsfip_ipmac_assoc *assoc = NULL; uint32_t i; for (i = 0; i < dimof(dhcpd->assoc); i++) { if (dhcpd->assoc[i].mac.size == macsize) { // allocated assoc if (!memcmp(dhcpd->assoc[i].mac.addr.s_addr_buf, hwaddr, macsize)) { return &dhcpd->assoc[i]; } } else if (NULL == assoc) { assoc = &dhcpd->assoc[i]; } } if (assoc != NULL) { uint32_t addr = netif->ipaddr.addr.s_addr & ~netif->netmask.addr.s_addr; addr = BE_TO_SYS_U32(addr) + dhcpd->alloc_idx; addr = SYS_TO_BE_U32(addr); if (addr & netif->netmask.addr.s_addr) { // no more address available return NULL; } dhcpd->alloc_idx++; assoc->ip.size = netif->ipaddr.size; assoc->ip.addr.s_addr = addr | (netif->netmask.addr.s_addr & netif->ipaddr.addr.s_addr); assoc->mac.size = netif->macaddr.size; memcpy(assoc->mac.addr.s_addr_buf, hwaddr, netif->macaddr.size); } return assoc; }
static void vsfip_dhcpd_input(void *param, struct vsfip_buffer_t *buf) { struct vsfip_dhcpd_t *dhcpd = (struct vsfip_dhcpd_t *)param; struct vsfip_netif_t *netif = dhcpd->netif; struct vsfip_dhcphead_t *head; struct vsfip_ipmac_assoc *assoc; uint8_t optlen, op; uint8_t *optptr; if (netif != buf->netif) { goto exit; } head = (struct vsfip_dhcphead_t *)buf->app.buffer; if ((head->op != DHCP_TOSERVER) || (head->magic != SYS_TO_BE_U32(DHCP_MAGIC)) || (head->htype != VSFIP_ETH_HWTYPE) || (head->hlen != 6)) { goto exit; } optlen = vsfip_dhcp_get_opt(buf, DHCPOPT_MSGTYPE, &optptr); if (optlen != DHCPOPT_MSGTYPE_LEN) { goto exit; } switch (optptr[0]) { case DHCPOP_DISCOVER: assoc = vsfip_dhcpd_get_assoc(netif, head->chaddr); if (NULL == assoc) { goto exit; } op = DHCPOP_OFFER; common_reply: head->op = DHCP_TOCLIENT; head->secs = 0; head->flags = 0; head->yiaddr = assoc->ip.addr.s_addr; head->siaddr = netif->ipaddr.addr.s_addr; buf->app.size = sizeof(*head); dhcpd->optlen = 0; vsfip_dhcp_append_opt(buf, &dhcpd->optlen, DHCPOPT_MSGTYPE, DHCPOPT_MSGTYPE_LEN, &op); vsfip_dhcp_append_opt(buf, &dhcpd->optlen, DHCPOPT_SERVERID, netif->ipaddr.size, netif->ipaddr.addr.s_addr_buf); { uint32_t lease_time = SYS_TO_BE_U32(0x80000000); vsfip_dhcp_append_opt(buf, &dhcpd->optlen, DHCPOPT_LEASE_TIME, netif->ipaddr.size, (uint8_t *)&lease_time); } vsfip_dhcp_append_opt(buf, &dhcpd->optlen, DHCPOPT_SUBNETMASK, netif->netmask.size, netif->netmask.addr.s_addr_buf); vsfip_dhcp_append_opt(buf, &dhcpd->optlen, DHCPOPT_ROUTER, netif->ipaddr.size, netif->ipaddr.addr.s_addr_buf); vsfip_dhcp_append_opt(buf, &dhcpd->optlen, DHCPOPT_DNSSERVER, netif->ipaddr.size, netif->ipaddr.addr.s_addr_buf); #ifdef VSFIP_CFG_HOSTNAME vsfip_dhcp_append_opt(buf, &dhcpd->optlen, DHCPOPT_HOSTNAME, strlen(VSFIP_CFG_HOSTNAME), (uint8_t *)VSFIP_CFG_HOSTNAME); #endif #ifdef VSFIP_CFG_DOMAIN vsfip_dhcp_append_opt(buf, &dhcpd->optlen, DHCPOPT_DOMAIN, strlen(VSFIP_CFG_DOMAIN), (uint8_t *)VSFIP_CFG_DOMAIN); #endif vsfip_dhcp_end_opt(buf, &dhcpd->optlen); dhcpd->sockaddr.sin_addr.addr.s_addr = assoc->ip.addr.s_addr; vsfip_netif_arp_add_assoc(dhcpd->netif, assoc->mac.size, assoc->mac.addr.s_addr_buf, assoc->ip.size, assoc->ip.addr.s_addr_buf); vsfip_udp_async_send(dhcpd->so, &dhcpd->sockaddr, buf); dhcpd->so->remote_sockaddr.sin_addr.addr.s_addr = VSFIP_IPADDR_ANY; return; case DHCPOP_REQUEST: assoc = vsfip_dhcpd_get_assoc(netif, head->chaddr); if (NULL == assoc) { goto exit; } optlen = vsfip_dhcp_get_opt(buf, DHCPOPT_REQIP, &optptr); if ((4 == optlen) && (*(uint32_t *)optptr != assoc->ip.addr.s_addr)) { op = DHCPOP_NAK; } else { op = DHCPOP_ACK; } goto common_reply; } exit: vsfip_buffer_release(buf); }
static void vsfip_dhcpc_input(void *param, struct vsfip_buffer_t *buf) { struct vsfip_dhcpc_t *dhcpc = (struct vsfip_dhcpc_t *)param; struct vsfip_netif_t *netif = dhcpc->netif; struct vsfip_dhcphead_t *head; uint8_t optlen; uint8_t *optptr; head = (struct vsfip_dhcphead_t *)buf->app.buffer; if ((head->op != DHCP_TOCLIENT) || (head->magic != SYS_TO_BE_U32(DHCP_MAGIC)) || memcmp(head->chaddr, netif->macaddr.addr.s_addr_buf, netif->macaddr.size) || (head->xid != dhcpc->xid)) { goto exit; } optlen = vsfip_dhcp_get_opt(buf, DHCPOPT_MSGTYPE, &optptr); if (optlen != DHCPOPT_MSGTYPE_LEN) { goto exit; } switch (optptr[0]) { case DHCPOP_OFFER: dhcpc->ipaddr.size = 4; dhcpc->ipaddr.addr.s_addr = head->yiaddr; vsfsm_post_evt(&dhcpc->sm, VSFIP_DHCP_EVT_SEND_REQUEST); break; case DHCPOP_ACK: optlen = vsfip_dhcp_get_opt(buf, DHCPOPT_LEASE_TIME, &optptr); dhcpc->leasetime = (4 == optlen) ? GET_BE_U32(optptr) : 0; optlen = vsfip_dhcp_get_opt(buf, DHCPOPT_RENEW_TIME, &optptr); dhcpc->renew_time = (4 == optlen) ? GET_BE_U32(optptr) : 0; optlen = vsfip_dhcp_get_opt(buf, DHCPOPT_REBINDING_TIME, &optptr); dhcpc->rebinding_time = (4 == optlen) ? GET_BE_U32(optptr) : 0; optlen = vsfip_dhcp_get_opt(buf, DHCPOPT_SUBNETMASK, &optptr); dhcpc->netmask.size = optlen; dhcpc->netmask.addr.s_addr = (4 == optlen) ? *(uint32_t *)optptr : 0; optlen = vsfip_dhcp_get_opt(buf, DHCPOPT_ROUTER, &optptr); dhcpc->gw.size = optlen; dhcpc->gw.addr.s_addr = (4 == optlen) ? *(uint32_t *)optptr : 0; optlen = vsfip_dhcp_get_opt(buf, DHCPOPT_DNSSERVER, &optptr); dhcpc->dns[0].size = dhcpc->dns[1].size = 0; if (optlen >= 4) { dhcpc->dns[0].size = 4; dhcpc->dns[0].addr.s_addr = *(uint32_t *)optptr; if (optlen >= 8) { dhcpc->dns[1].size = 4; dhcpc->dns[1].addr.s_addr = *(uint32_t *)(optptr + 4); } } vsfsm_post_evt(&dhcpc->sm, VSFIP_DHCP_EVT_READY); break; } exit: vsfip_buffer_release(buf); }