X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=blobdiff_plain;f=src%2Flibsystemd-network%2Fsd-dhcp-lease.c;h=648e6e2159b9f45a11ef775cc4a9613b0a148e10;hb=109731eb250705e055b98b844d7c0dfb74bcf728;hp=0529b6d8fa3cb4f26bfac8aadd3a9253ba9b32c3;hpb=fe8db0c5ee3365a2fc80ee7ebffa238f9a0a2ae2;p=elogind.git diff --git a/src/libsystemd-network/sd-dhcp-lease.c b/src/libsystemd-network/sd-dhcp-lease.c index 0529b6d8f..648e6e215 100644 --- a/src/libsystemd-network/sd-dhcp-lease.c +++ b/src/libsystemd-network/sd-dhcp-lease.c @@ -72,6 +72,20 @@ int sd_dhcp_lease_get_dns(sd_dhcp_lease *lease, struct in_addr **addr, size_t *a return 0; } +int sd_dhcp_lease_get_ntp(sd_dhcp_lease *lease, struct in_addr **addr, size_t *addr_size) { + assert_return(lease, -EINVAL); + assert_return(addr, -EINVAL); + assert_return(addr_size, -EINVAL); + + if (lease->ntp_size) { + *addr_size = lease->ntp_size; + *addr = lease->ntp; + } else + return -ENOENT; + + return 0; +} + int sd_dhcp_lease_get_domainname(sd_dhcp_lease *lease, const char **domainname) { assert_return(lease, -EINVAL); assert_return(domainname, -EINVAL); @@ -96,6 +110,18 @@ int sd_dhcp_lease_get_hostname(sd_dhcp_lease *lease, const char **hostname) { return 0; } +int sd_dhcp_lease_get_root_path(sd_dhcp_lease *lease, const char **root_path) { + assert_return(lease, -EINVAL); + assert_return(root_path, -EINVAL); + + if (lease->root_path) + *root_path = lease->root_path; + else + return -ENOENT; + + return 0; +} + int sd_dhcp_lease_get_router(sd_dhcp_lease *lease, struct in_addr *addr) { assert_return(lease, -EINVAL); assert_return(addr, -EINVAL); @@ -114,6 +140,24 @@ int sd_dhcp_lease_get_netmask(sd_dhcp_lease *lease, struct in_addr *addr) { return 0; } +int sd_dhcp_lease_get_server_identifier(sd_dhcp_lease *lease, struct in_addr *addr) { + assert_return(lease, -EINVAL); + assert_return(addr, -EINVAL); + + addr->s_addr = lease->server_address; + + return 0; +} + +int sd_dhcp_lease_get_next_server(sd_dhcp_lease *lease, struct in_addr *addr) { + assert_return(lease, -EINVAL); + assert_return(addr, -EINVAL); + + addr->s_addr = lease->next_server; + + return 0; +} + sd_dhcp_lease *sd_dhcp_lease_ref(sd_dhcp_lease *lease) { if (lease) assert_se(REFCNT_INC(lease->n_ref) >= 2); @@ -167,18 +211,24 @@ int dhcp_lease_parse_options(uint8_t code, uint8_t len, const uint8_t *option, case DHCP_OPTION_DOMAIN_NAME_SERVER: if (len && !(len % 4)) { - unsigned i; - lease->dns_size = len / 4; free(lease->dns); - lease->dns = new0(struct in_addr, lease->dns_size); + lease->dns = newdup(struct in_addr, option, lease->dns_size); if (!lease->dns) return -ENOMEM; + } + + break; - for (i = 0; i < lease->dns_size; i++) { - memcpy(&lease->dns[i].s_addr, option + 4 * i, 4); - } + case DHCP_OPTION_NTP_SERVER: + if (len && !(len % 4)) { + lease->ntp_size = len / 4; + + free(lease->ntp); + lease->ntp = newdup(struct in_addr, option, lease->ntp_size); + if (!lease->ntp) + return -ENOMEM; } break; @@ -212,6 +262,14 @@ int dhcp_lease_parse_options(uint8_t code, uint8_t len, const uint8_t *option, break; + case DHCP_OPTION_ROOT_PATH: + if (len >= 1) { + free(lease->root_path); + lease->root_path = strndup((const char *)option, len); + } + + break; + case DHCP_OPTION_RENEWAL_T1_TIME: if (len == 4) { memcpy(&val, option, 4); @@ -233,7 +291,7 @@ int dhcp_lease_parse_options(uint8_t code, uint8_t len, const uint8_t *option, } int dhcp_lease_new(sd_dhcp_lease **ret) { - _cleanup_dhcp_lease_unref_ sd_dhcp_lease *lease = NULL; + sd_dhcp_lease *lease; lease = new0(sd_dhcp_lease, 1); if (!lease) @@ -242,7 +300,44 @@ int dhcp_lease_new(sd_dhcp_lease **ret) { lease->n_ref = REFCNT_INIT; *ret = lease; - lease = NULL; + 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; + + r = inet_aton(word, &(new_addresses[*size])); + if (r < 0) + continue; + + *addresses = new_addresses; + (*size)++; + } return 0; } @@ -250,8 +345,9 @@ int dhcp_lease_new(sd_dhcp_lease **ret) { 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; @@ -259,10 +355,6 @@ int dhcp_lease_save(sd_dhcp_lease *lease, const char *lease_file) { assert(lease); assert(lease_file); - r = mkdir_safe_label("/run/systemd/network/leases", 0755, 0, 0); - if (r < 0) - goto finish; - r = fopen_temporary(lease_file, &f, &temp_path); if (r < 0) goto finish; @@ -273,47 +365,42 @@ 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); + "ADDRESS=%s\n", inet_ntoa(address)); 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); + fprintf(f, "ROUTER=%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_server_identifier(lease, &address); + if (r >= 0) + fprintf(f, "SERVER_ADDRESS=%s\n", + inet_ntoa(address)); + + r = sd_dhcp_lease_get_next_server(lease, &address); + 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) @@ -323,6 +410,10 @@ int dhcp_lease_save(sd_dhcp_lease *lease, const char *lease_file) { if (r >= 0) fprintf(f, "HOSTNAME=%s\n", string); + r = sd_dhcp_lease_get_root_path(lease, &string); + if (r >= 0) + fprintf(f, "ROOT_PATH=%s\n", string); + r = 0; fflush(f); @@ -343,7 +434,8 @@ finish: 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, - *mtu = NULL; + *server_address = NULL, *next_server = NULL, + *dns = NULL, *ntp = NULL, *mtu = NULL; struct in_addr addr; int r; @@ -358,9 +450,14 @@ int dhcp_lease_load(const char *lease_file, sd_dhcp_lease **ret) { "ADDRESS", &address, "ROUTER", &router, "NETMASK", &netmask, + "SERVER_IDENTIFIER", &server_address, + "NEXT_SERVER", &next_server, + "DNS", &dns, + "NTP", &ntp, "MTU", &mtu, "DOMAINNAME", &lease->domainname, "HOSTNAME", &lease->hostname, + "ROOT_PATH", &lease->root_path, NULL); if (r < 0) { if (r == -ENOENT) @@ -388,6 +485,34 @@ int dhcp_lease_load(const char *lease_file, sd_dhcp_lease **ret) { lease->subnet_mask = addr.s_addr; + if (server_address) { + r = inet_pton(AF_INET, server_address, &addr); + if (r < 0) + return r; + + lease->server_address = addr.s_addr; + } + + if (next_server) { + r = inet_pton(AF_INET, next_server, &addr); + if (r < 0) + return r; + + 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) @@ -399,3 +524,29 @@ int dhcp_lease_load(const char *lease_file, sd_dhcp_lease **ret) { return 0; } + +int dhcp_lease_set_default_subnet_mask(sd_dhcp_lease *lease) { + uint32_t address; + + assert(lease); + assert(lease->address != INADDR_ANY); + + address = be32toh(lease->address); + + /* fall back to the default subnet masks based on address class */ + + if ((address >> 31) == 0x0) + /* class A, leading bits: 0 */ + lease->subnet_mask = htobe32(0xff000000); + else if ((address >> 30) == 0x2) + /* class B, leading bits 10 */ + lease->subnet_mask = htobe32(0xffff0000); + else if ((address >> 29) == 0x6) + /* class C, leading bits 110 */ + lease->subnet_mask = htobe32(0xffffff00); + else + /* class D or E, no default mask. give up */ + return -ERANGE; + + return 0; +}