From 424a87321427d255ac287f08a649e7808ae1641c Mon Sep 17 00:00:00 2001 From: Tom Gundersen Date: Wed, 21 May 2014 16:46:14 +0200 Subject: [PATCH] sd-dhcp-client: factor out dhcp packet allocation --- TODO | 1 + src/libsystemd-network/sd-dhcp-client.c | 66 ++++++++++++------------- 2 files changed, 34 insertions(+), 33 deletions(-) diff --git a/TODO b/TODO index e5ba3d284..2aff98edf 100644 --- a/TODO +++ b/TODO @@ -714,6 +714,7 @@ Features: - make operstates to wait for configurable? * dhcp: + - figure out how much we can increase Maximum Message Size - export timezone information - FORCERENEW diff --git a/src/libsystemd-network/sd-dhcp-client.c b/src/libsystemd-network/sd-dhcp-client.c index ba44069f6..67593c42c 100644 --- a/src/libsystemd-network/sd-dhcp-client.c +++ b/src/libsystemd-network/sd-dhcp-client.c @@ -256,35 +256,45 @@ static sd_dhcp_client *client_stop(sd_dhcp_client *client, int error) { return client; } -static int client_message_init(sd_dhcp_client *client, DHCPMessage *message, - uint8_t type, size_t optlen, size_t *optoffset) { +static int client_message_init(sd_dhcp_client *client, DHCPPacket **ret, + uint8_t type, size_t *_optlen, size_t *_optoffset) { + _cleanup_free_ DHCPPacket *packet; + size_t optlen, optoffset, size; be16_t max_size; int r; assert(client); assert(client->secs); - assert(message); - assert(optoffset); + assert(ret); + assert(_optlen); + assert(_optoffset); assert(type == DHCP_DISCOVER || type == DHCP_REQUEST); - r = dhcp_message_init(message, BOOTREQUEST, client->xid, type, - optlen, optoffset); + optlen = DHCP_MIN_OPTIONS_SIZE; + size = sizeof(DHCPPacket) + optlen; + + packet = malloc0(size); + if (!packet) + return -ENOMEM; + + r = dhcp_message_init(&packet->dhcp, BOOTREQUEST, client->xid, type, + optlen, &optoffset); if (r < 0) return r; /* Although 'secs' field is a SHOULD in RFC 2131, certain DHCP servers refuse to issue an DHCP lease if 'secs' is set to zero */ - message->secs = htobe16(client->secs); + packet->dhcp.secs = htobe16(client->secs); /* RFC2132 section 4.1.1: The client MUST include its hardware address in the ’chaddr’ field, if necessary for delivery of DHCP reply messages. */ - memcpy(&message->chaddr, &client->client_id.mac_addr, ETH_ALEN); + memcpy(&packet->dhcp.chaddr, &client->client_id.mac_addr, ETH_ALEN); /* Some DHCP servers will refuse to issue an DHCP lease if the Client Identifier option is not set */ - r = dhcp_option_append(message, optlen, optoffset, 0, + r = dhcp_option_append(&packet->dhcp, optlen, &optoffset, 0, DHCP_OPTION_CLIENT_IDENTIFIER, sizeof(client->client_id), &client->client_id); if (r < 0) @@ -299,7 +309,7 @@ static int client_message_init(sd_dhcp_client *client, DHCPMessage *message, it MUST include that list in any subsequent DHCPREQUEST messages. */ - r = dhcp_option_append(message, optlen, optoffset, 0, + r = dhcp_option_append(&packet->dhcp, optlen, &optoffset, 0, DHCP_OPTION_PARAMETER_REQUEST_LIST, client->req_opts_size, client->req_opts); if (r < 0) @@ -313,14 +323,18 @@ static int client_message_init(sd_dhcp_client *client, DHCPMessage *message, than the defined default size unless the Maximum Messge Size option is explicitely set */ - max_size = htobe16(DHCP_IP_UDP_SIZE + DHCP_MESSAGE_SIZE + - DHCP_MIN_OPTIONS_SIZE); - r = dhcp_option_append(message, optlen, optoffset, 0, + max_size = htobe16(size); + r = dhcp_option_append(&packet->dhcp, optlen, &optoffset, 0, DHCP_OPTION_MAXIMUM_MESSAGE_SIZE, 2, &max_size); if (r < 0) return r; + *_optlen = optlen; + *_optoffset = optoffset; + *ret = packet; + packet = NULL; + return 0; } @@ -335,7 +349,7 @@ static int dhcp_client_send_raw(sd_dhcp_client *client, DHCPPacket *packet, static int client_send_discover(sd_dhcp_client *client) { _cleanup_free_ DHCPPacket *discover = NULL; - size_t optoffset, optlen, len; + size_t optoffset, optlen; usec_t time_now; int r; @@ -354,15 +368,8 @@ static int client_send_discover(sd_dhcp_client *client) { * must always be strictly positive to deal with broken servers */ client->secs = ((time_now - client->start_time) / USEC_PER_SEC) ? : 1; - optlen = DHCP_MIN_OPTIONS_SIZE; - len = sizeof(DHCPPacket) + optlen; - - discover = malloc0(len); - if (!discover) - return -ENOMEM; - - r = client_message_init(client, &discover->dhcp, DHCP_DISCOVER, - optlen, &optoffset); + r = client_message_init(client, &discover, DHCP_DISCOVER, + &optlen, &optoffset); if (r < 0) return r; @@ -398,18 +405,11 @@ static int client_send_discover(sd_dhcp_client *client) { static int client_send_request(sd_dhcp_client *client) { _cleanup_free_ DHCPPacket *request; - size_t optoffset, optlen, len; + size_t optoffset, optlen; int r; - optlen = DHCP_MIN_OPTIONS_SIZE; - len = sizeof(DHCPPacket) + optlen; - - request = malloc0(len); - if (!request) - return -ENOMEM; - - r = client_message_init(client, &request->dhcp, DHCP_REQUEST, - optlen, &optoffset); + r = client_message_init(client, &request, DHCP_REQUEST, + &optlen, &optoffset); if (r < 0) return r; -- 2.30.2