X-Git-Url: http://git.samba.org/samba.git/?a=blobdiff_plain;f=lib%2Ftsocket%2Ftsocket_bsd.c;h=bc7cfe3fe9e24aad603be171b79c3d6a0814d150;hb=4423aa59abda50c8b71815f922ea03e2009f9e50;hp=c44ba4e714fe2735bd787003e4b468d918a0b850;hpb=c42d9c4ec410e205091784cd97cbceb5572609d8;p=samba.git diff --git a/lib/tsocket/tsocket_bsd.c b/lib/tsocket/tsocket_bsd.c index c44ba4e714f..bc7cfe3fe9e 100644 --- a/lib/tsocket/tsocket_bsd.c +++ b/lib/tsocket/tsocket_bsd.c @@ -190,6 +190,7 @@ static ssize_t tsocket_bsd_pending(int fd) static const struct tsocket_address_ops tsocket_address_bsd_ops; struct tsocket_address_bsd { + socklen_t sa_socklen; union { struct sockaddr sa; struct sockaddr_in in; @@ -261,6 +262,8 @@ int _tsocket_address_bsd_from_sockaddr(TALLOC_CTX *mem_ctx, memcpy(&bsda->u.ss, sa, sa_socklen); + bsda->sa_socklen = sa_socklen; + *_addr = addr; return 0; } @@ -271,42 +274,61 @@ ssize_t tsocket_address_bsd_sockaddr(const struct tsocket_address *addr, { struct tsocket_address_bsd *bsda = talloc_get_type(addr->private_data, struct tsocket_address_bsd); - ssize_t rlen = 0; if (!bsda) { errno = EINVAL; return -1; } - switch (bsda->u.sa.sa_family) { - case AF_UNIX: - rlen = sizeof(struct sockaddr_un); - break; - case AF_INET: - rlen = sizeof(struct sockaddr_in); - break; -#ifdef HAVE_IPV6 - case AF_INET6: - rlen = sizeof(struct sockaddr_in6); - break; -#endif - default: - errno = EAFNOSUPPORT; - return -1; - } - - if (sa_socklen < rlen) { + if (sa_socklen < bsda->sa_socklen) { errno = EINVAL; return -1; } - if (sa_socklen > sizeof(struct sockaddr_storage)) { + if (sa_socklen > bsda->sa_socklen) { memset(sa, 0, sa_socklen); - sa_socklen = sizeof(struct sockaddr_storage); + sa_socklen = bsda->sa_socklen; } memcpy(sa, &bsda->u.ss, sa_socklen); - return rlen; + return sa_socklen; +} + +bool tsocket_address_is_inet(const struct tsocket_address *addr, const char *fam) +{ + struct tsocket_address_bsd *bsda = talloc_get_type(addr->private_data, + struct tsocket_address_bsd); + + if (!bsda) { + return false; + } + + switch (bsda->u.sa.sa_family) { + case AF_INET: + if (strcasecmp(fam, "ip") == 0) { + return true; + } + + if (strcasecmp(fam, "ipv4") == 0) { + return true; + } + + return false; +#ifdef HAVE_IPV6 + case AF_INET6: + if (strcasecmp(fam, "ip") == 0) { + return true; + } + + if (strcasecmp(fam, "ipv6") == 0) { + return true; + } + + return false; +#endif + } + + return false; } int _tsocket_address_inet_from_strings(TALLOC_CTX *mem_ctx, @@ -481,6 +503,23 @@ int tsocket_address_inet_set_port(struct tsocket_address *addr, return 0; } +bool tsocket_address_is_unix(const struct tsocket_address *addr) +{ + struct tsocket_address_bsd *bsda = talloc_get_type(addr->private_data, + struct tsocket_address_bsd); + + if (!bsda) { + return false; + } + + switch (bsda->u.sa.sa_family) { + case AF_UNIX: + return true; + } + + return false; +} + int _tsocket_address_unix_from_path(TALLOC_CTX *mem_ctx, const char *path, struct tsocket_address **_addr, @@ -588,7 +627,7 @@ static struct tsocket_address *tsocket_address_bsd_copy(const struct tsocket_add ret = _tsocket_address_bsd_from_sockaddr(mem_ctx, &bsda->u.sa, - sizeof(bsda->u.ss), + bsda->sa_socklen, ©, location); if (ret != 0) { @@ -827,8 +866,6 @@ static void tdgram_bsd_recvfrom_handler(void *private_data) struct tdgram_bsd *bsds = tdgram_context_data(dgram, struct tdgram_bsd); struct tsocket_address_bsd *bsda; ssize_t ret; - struct sockaddr *sa = NULL; - socklen_t sa_socklen = 0; int err; bool retry; @@ -862,18 +899,10 @@ static void tdgram_bsd_recvfrom_handler(void *private_data) } ZERO_STRUCTP(bsda); + bsda->sa_socklen = sizeof(bsda->u.ss); - sa = &bsda->u.sa; - sa_socklen = sizeof(bsda->u.ss); - /* - * for unix sockets we can't use the size of sockaddr_storage - * we would get EINVAL - */ - if (bsda->u.sa.sa_family == AF_UNIX) { - sa_socklen = sizeof(bsda->u.un); - } - - ret = recvfrom(bsds->fd, state->buf, state->len, 0, sa, &sa_socklen); + ret = recvfrom(bsds->fd, state->buf, state->len, 0, + &bsda->u.sa, &bsda->sa_socklen); err = tsocket_bsd_error_from_errno(ret, errno, &retry); if (retry) { /* retry later */ @@ -1019,14 +1048,7 @@ static void tdgram_bsd_sendto_handler(void *private_data) struct tsocket_address_bsd); sa = &bsda->u.sa; - sa_socklen = sizeof(bsda->u.ss); - /* - * for unix sockets we can't use the size of sockaddr_storage - * we would get EINVAL - */ - if (bsda->u.sa.sa_family == AF_UNIX) { - sa_socklen = sizeof(bsda->u.un); - } + sa_socklen = bsda->sa_socklen; } ret = sendto(bsds->fd, state->buf, state->len, 0, sa, sa_socklen); @@ -1085,6 +1107,7 @@ static struct tevent_req *tdgram_bsd_disconnect_send(TALLOC_CTX *mem_ctx, goto post; } + TALLOC_FREE(bsds->fde); ret = close(bsds->fd); bsds->fd = -1; err = tsocket_bsd_error_from_errno(ret, errno, &dummy); @@ -1152,7 +1175,6 @@ static int tdgram_bsd_dgram_socket(const struct tsocket_address *local, bool do_ipv6only = false; bool is_inet = false; int sa_fam = lbsda->u.sa.sa_family; - socklen_t sa_socklen = sizeof(lbsda->u.ss); if (remote) { rbsda = talloc_get_type_abort(remote->private_data, @@ -1169,11 +1191,6 @@ static int tdgram_bsd_dgram_socket(const struct tsocket_address *local, do_reuseaddr = true; do_bind = true; } - /* - * for unix sockets we can't use the size of sockaddr_storage - * we would get EINVAL - */ - sa_socklen = sizeof(lbsda->u.un); break; case AF_INET: if (lbsda->u.in.sin_port != 0) { @@ -1184,7 +1201,6 @@ static int tdgram_bsd_dgram_socket(const struct tsocket_address *local, do_bind = true; } is_inet = true; - sa_socklen = sizeof(rbsda->u.in); break; #ifdef HAVE_IPV6 case AF_INET6: @@ -1198,7 +1214,6 @@ static int tdgram_bsd_dgram_socket(const struct tsocket_address *local, do_bind = true; } is_inet = true; - sa_socklen = sizeof(rbsda->u.in6); do_ipv6only = true; break; #endif @@ -1211,12 +1226,10 @@ static int tdgram_bsd_dgram_socket(const struct tsocket_address *local, sa_fam = rbsda->u.sa.sa_family; switch (sa_fam) { case AF_INET: - sa_socklen = sizeof(rbsda->u.in); do_ipv6only = false; break; #ifdef HAVE_IPV6 case AF_INET6: - sa_socklen = sizeof(rbsda->u.in6); do_ipv6only = true; break; #endif @@ -1225,12 +1238,12 @@ static int tdgram_bsd_dgram_socket(const struct tsocket_address *local, fd = socket(sa_fam, SOCK_DGRAM, 0); if (fd < 0) { - return fd; + return -1; } fd = tsocket_bsd_common_prepare_fd(fd, true); if (fd < 0) { - return fd; + return -1; } dgram = tdgram_context_create(mem_ctx, @@ -1258,7 +1271,7 @@ static int tdgram_bsd_dgram_socket(const struct tsocket_address *local, int saved_errno = errno; talloc_free(dgram); errno = saved_errno; - return ret; + return -1; } } #endif @@ -1272,7 +1285,7 @@ static int tdgram_bsd_dgram_socket(const struct tsocket_address *local, int saved_errno = errno; talloc_free(dgram); errno = saved_errno; - return ret; + return -1; } } @@ -1285,17 +1298,17 @@ static int tdgram_bsd_dgram_socket(const struct tsocket_address *local, int saved_errno = errno; talloc_free(dgram); errno = saved_errno; - return ret; + return -1; } } if (do_bind) { - ret = bind(fd, &lbsda->u.sa, sa_socklen); + ret = bind(fd, &lbsda->u.sa, lbsda->sa_socklen); if (ret == -1) { int saved_errno = errno; talloc_free(dgram); errno = saved_errno; - return ret; + return -1; } } @@ -1306,12 +1319,12 @@ static int tdgram_bsd_dgram_socket(const struct tsocket_address *local, return -1; } - ret = connect(fd, &rbsda->u.sa, sa_socklen); + ret = connect(fd, &rbsda->u.sa, rbsda->sa_socklen); if (ret == -1) { int saved_errno = errno; talloc_free(dgram); errno = saved_errno; - return ret; + return -1; } } @@ -1651,7 +1664,7 @@ static void tstream_bsd_readv_handler(void *private_data) uint8_t *base; base = (uint8_t *)state->vector[0].iov_base; base += ret; - state->vector[0].iov_base = base; + state->vector[0].iov_base = (void *)base; state->vector[0].iov_len -= ret; break; } @@ -1811,7 +1824,7 @@ static void tstream_bsd_writev_handler(void *private_data) uint8_t *base; base = (uint8_t *)state->vector[0].iov_base; base += ret; - state->vector[0].iov_base = base; + state->vector[0].iov_base = (void *)base; state->vector[0].iov_len -= ret; break; } @@ -1882,6 +1895,7 @@ static struct tevent_req *tstream_bsd_disconnect_send(TALLOC_CTX *mem_ctx, goto post; } + TALLOC_FREE(bsds->fde); ret = close(bsds->fd); bsds->fd = -1; err = tsocket_bsd_error_from_errno(ret, errno, &dummy); @@ -1999,7 +2013,6 @@ static struct tevent_req * tstream_bsd_connect_send(TALLOC_CTX *mem_ctx, bool do_ipv6only = false; bool is_inet = false; int sa_fam = lbsda->u.sa.sa_family; - socklen_t sa_socklen = sizeof(rbsda->u.ss); req = tevent_req_create(mem_ctx, &state, struct tstream_bsd_connect_state); @@ -2023,11 +2036,6 @@ static struct tevent_req * tstream_bsd_connect_send(TALLOC_CTX *mem_ctx, do_reuseaddr = true; do_bind = true; } - /* - * for unix sockets we can't use the size of sockaddr_storage - * we would get EINVAL - */ - sa_socklen = sizeof(rbsda->u.un); break; case AF_INET: if (lbsda->u.in.sin_port != 0) { @@ -2038,7 +2046,6 @@ static struct tevent_req * tstream_bsd_connect_send(TALLOC_CTX *mem_ctx, do_bind = true; } is_inet = true; - sa_socklen = sizeof(rbsda->u.in); break; #ifdef HAVE_IPV6 case AF_INET6: @@ -2052,7 +2059,6 @@ static struct tevent_req * tstream_bsd_connect_send(TALLOC_CTX *mem_ctx, do_bind = true; } is_inet = true; - sa_socklen = sizeof(rbsda->u.in6); do_ipv6only = true; break; #endif @@ -2065,12 +2071,10 @@ static struct tevent_req * tstream_bsd_connect_send(TALLOC_CTX *mem_ctx, sa_fam = rbsda->u.sa.sa_family; switch (sa_fam) { case AF_INET: - sa_socklen = sizeof(rbsda->u.in); do_ipv6only = false; break; #ifdef HAVE_IPV6 case AF_INET6: - sa_socklen = sizeof(rbsda->u.in6); do_ipv6only = true; break; #endif @@ -2114,7 +2118,7 @@ static struct tevent_req * tstream_bsd_connect_send(TALLOC_CTX *mem_ctx, } if (do_bind) { - ret = bind(state->fd, &lbsda->u.sa, sa_socklen); + ret = bind(state->fd, &lbsda->u.sa, lbsda->sa_socklen); if (ret == -1) { tevent_req_error(req, errno); goto post; @@ -2126,7 +2130,7 @@ static struct tevent_req * tstream_bsd_connect_send(TALLOC_CTX *mem_ctx, goto post; } - ret = connect(state->fd, &rbsda->u.sa, sa_socklen); + ret = connect(state->fd, &rbsda->u.sa, rbsda->sa_socklen); err = tsocket_bsd_error_from_errno(ret, errno, &retry); if (retry) { /* retry later */