X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=blobdiff_plain;f=src%2Flibsystemd-network%2Fsd-dhcp-lease.c;h=3e43ab635f00ab787e8c7890f27f1ae38e2040ec;hb=d3152a09ac5804ec8603daee12f98cf03523cce0;hp=61822abd267b06ca127a269cff05778ade3e8124;hpb=0b21bde96d6d33313bac42ae1d5802a83a4e7b2f;p=elogind.git diff --git a/src/libsystemd-network/sd-dhcp-lease.c b/src/libsystemd-network/sd-dhcp-lease.c index 61822abd2..3e43ab635 100644 --- a/src/libsystemd-network/sd-dhcp-lease.c +++ b/src/libsystemd-network/sd-dhcp-lease.c @@ -126,7 +126,10 @@ int sd_dhcp_lease_get_router(sd_dhcp_lease *lease, struct in_addr *addr) { assert_return(lease, -EINVAL); assert_return(addr, -EINVAL); - addr->s_addr = lease->router; + if (lease->router != INADDR_ANY) + addr->s_addr = lease->router; + else + return -ENOENT; return 0; } @@ -297,17 +300,59 @@ int dhcp_lease_new(sd_dhcp_lease **ret) { if (!lease) return -ENOMEM; + lease->router = INADDR_ANY; lease->n_ref = REFCNT_INIT; *ret = lease; return 0; } +static void serialize_addresses(FILE *f, const char *key, struct in_addr *addresses, size_t size) { + unsigned i; + + assert(key); + assert(addresses); + assert(size); + + fputs("DNS=", f); + + for (i = 0; i < size; i++) + fprintf(f, "%s%s", inet_ntoa(addresses[i]), + (i < (size - 1)) ? " ": ""); + + fputs("\n", f); +} + +static int deserialize_addresses(struct in_addr **addresses, size_t *size, const char *string) { + char *word, *state; + size_t len; + + FOREACH_WORD(word, len, string, state) { + struct in_addr *new_addresses; + int r; + + new_addresses = realloc(*addresses, (*size + 1) * sizeof(struct in_addr)); + if (!new_addresses) + return -ENOMEM; + else + *addresses = new_addresses; + + r = inet_aton(word, &(new_addresses[*size])); + if (r < 0) + continue; + + (*size)++; + } + + return 0; +} + int dhcp_lease_save(sd_dhcp_lease *lease, const char *lease_file) { _cleanup_free_ char *temp_path = NULL; _cleanup_fclose_ FILE *f = NULL; - char buf[INET_ADDRSTRLEN]; struct in_addr address; + struct in_addr *addresses; + size_t addresses_size; const char *string; uint16_t mtu; int r; @@ -325,71 +370,40 @@ int dhcp_lease_save(sd_dhcp_lease *lease, const char *lease_file) { if (r < 0) goto finish; - string = inet_ntop(AF_INET, &address, buf, INET_ADDRSTRLEN); - if (!string) { - r = -errno; - goto finish; - } - fprintf(f, "# This is private data. Do not parse.\n" - "ADDRESS=%s\n", string); - - r = sd_dhcp_lease_get_router(lease, &address); - if (r < 0) - goto finish; - - string = inet_ntop(AF_INET, &address, buf, INET_ADDRSTRLEN); - if (!string) { - r = -errno; - goto finish; - } - - fprintf(f, - "ROUTER=%s\n", string); + "ADDRESS=%s\n", inet_ntoa(address)); r = sd_dhcp_lease_get_netmask(lease, &address); if (r < 0) goto finish; - string = inet_ntop(AF_INET, &address, buf, INET_ADDRSTRLEN); - if (!string) { - r = -errno; - goto finish; - } + fprintf(f, "NETMASK=%s\n", inet_ntoa(address)); - fprintf(f, - "NETMASK=%s\n", string); + r = sd_dhcp_lease_get_router(lease, &address); + if (r >= 0) + fprintf(f, "ROUTER=%s\n", inet_ntoa(address)); r = sd_dhcp_lease_get_server_identifier(lease, &address); - if (r >= 0) { - string = inet_ntop(AF_INET, &address, buf, INET_ADDRSTRLEN); - if (!string) { - r = -errno; - goto finish; - } - - fprintf(f, - "SERVER_ADDRESS=%s\n", string); - } + if (r >= 0) + fprintf(f, "SERVER_ADDRESS=%s\n", + inet_ntoa(address)); r = sd_dhcp_lease_get_next_server(lease, &address); - if (r >= 0) { - string = inet_ntop(AF_INET, &address, buf, INET_ADDRSTRLEN); - if (!string) { - r = -errno; - goto finish; - } - - fprintf(f, - "NEXT_SERVER=%s\n", string); - } + if (r >= 0) + fprintf(f, "NEXT_SERVER=%s\n", inet_ntoa(address)); r = sd_dhcp_lease_get_mtu(lease, &mtu); if (r >= 0) fprintf(f, "MTU=%" PRIu16 "\n", mtu); -/* TODO: DNS. See resolv.conf writing in network-manager.c */ + r = sd_dhcp_lease_get_dns(lease, &addresses, &addresses_size); + if (r >= 0) + serialize_addresses(f, "DNS", addresses, addresses_size); + + r = sd_dhcp_lease_get_ntp(lease, &addresses, &addresses_size); + if (r >= 0) + serialize_addresses(f, "NTP", addresses, addresses_size); r = sd_dhcp_lease_get_domainname(lease, &string); if (r >= 0) @@ -424,7 +438,7 @@ int dhcp_lease_load(const char *lease_file, sd_dhcp_lease **ret) { _cleanup_dhcp_lease_unref_ sd_dhcp_lease *lease = NULL; _cleanup_free_ char *address = NULL, *router = NULL, *netmask = NULL, *server_address = NULL, *next_server = NULL, - *mtu = NULL; + *dns = NULL, *ntp = NULL, *mtu = NULL; struct in_addr addr; int r; @@ -441,6 +455,8 @@ int dhcp_lease_load(const char *lease_file, sd_dhcp_lease **ret) { "NETMASK", &netmask, "SERVER_IDENTIFIER", &server_address, "NEXT_SERVER", &next_server, + "DNS", &dns, + "NTP", &ntp, "MTU", &mtu, "DOMAINNAME", &lease->domainname, "HOSTNAME", &lease->hostname, @@ -460,11 +476,13 @@ int dhcp_lease_load(const char *lease_file, sd_dhcp_lease **ret) { lease->address = addr.s_addr; - r = inet_pton(AF_INET, router, &addr); - if (r < 0) - return r; + if (router) { + r = inet_pton(AF_INET, router, &addr); + if (r < 0) + return r; - lease->router = addr.s_addr; + lease->router = addr.s_addr; + } r = inet_pton(AF_INET, netmask, &addr); if (r < 0) @@ -488,6 +506,18 @@ int dhcp_lease_load(const char *lease_file, sd_dhcp_lease **ret) { lease->next_server = addr.s_addr; } + if (dns) { + r = deserialize_addresses(&lease->dns, &lease->dns_size, dns); + if (r < 0) + return r; + } + + if (ntp) { + r = deserialize_addresses(&lease->ntp, &lease->ntp_size, dns); + if (r < 0) + return r; + } + if (mtu) { uint16_t u; if (sscanf(mtu, "%" SCNu16, &u) > 0)