int rc;
char buf[256];
+ /* If we can't find the socket node, then it's definitely not in use. If
+ * we get some other error, then this socket is weird.
+ */
if (stat(sun->sun_path, &st))
return (errno == ENOENT ? UNUSED : USED);
+
+ /* If it's not a socket, then something weird is going on. If we're just
+ * probing quickly to find a spare port, then existence is sufficient to
+ * discourage us now.
+ */
if (!S_ISSOCK(st.st_mode) || quickp)
return (USED);
+
+ /* The socket's definitely there, but is anyone actually still holding it
+ * open? The only way I know to discover this is to trundle through
+ * `/proc/net/unix'. If there's no entry, then the socket must be stale.
+ */
rc = USED;
if ((fp = fopen("/proc/net/unix", "r")) == 0)
goto done;
rc = STALE;
done:
if (fp) fclose(fp);
+
+ /* All done. */
return (rc);
}
+/* Encode SA as a Unix-domain address SUN, and return whether it's currently
+ * in use.
+ */
+static int encode_single_inet_addr(const struct sockaddr *sa,
+ struct sockaddr_un *sun,
+ int quickp)
+{
+ char buf[ADDRBUFSZ];
+ int rc;
+
+ snprintf(sun->sun_path, sizeof(sun->sun_path), "%s/%s", sockdir,
+ present_sockaddr(sa, 0, buf, sizeof(buf)));
+ if ((rc = unix_socket_status(sun, quickp)) == USED) return (USED);
+ else if (rc == STALE) unlink(sun->sun_path);
+ return (UNUSED);
+}
+
/* Convert the IP address SA to a Unix-domain address SUN. Fail if the
* address seems already taken. If DESPARATEP then try cleaning up stale old
* sockets.
{
address waddr;
struct sockaddr_un wsun;
- int rc;
- char buf[ADDRBUFSZ];
int port = port_from_sockaddr(sa);
- snprintf(sun->sun_path, sizeof(sun->sun_path), "%s/%s", sockdir,
- present_sockaddr(sa, 0, buf, sizeof(buf)));
- if ((rc = unix_socket_status(sun, !desperatep)) == USED) return (-1);
- else if (rc == STALE) unlink(sun->sun_path);
+ /* First, look for an exact match. Only look quickly unless we're
+ * desperate. If the socket is in use, we fail here. (This could get
+ * racy. Let's not worry about that for now.)
+ */
+ if (encode_single_inet_addr(sa, sun, !desperatep) == USED)
+ return (-1);
+ /* Next, check the corresponding wildcard address, so as to avoid
+ * inadvertant collisions with listeners. Do this in the same way.
+ */
wildcard_address(sa->sa_family, &waddr.sa);
port_to_sockaddr(&waddr.sa, port);
- snprintf(wsun.sun_path, sizeof(wsun.sun_path), "%s/%s", sockdir,
- present_sockaddr(&waddr.sa, 0, buf, sizeof(buf)));
- if ((rc = unix_socket_status(&wsun, !desperatep)) == USED) return (-1);
- else if (rc == STALE) unlink(wsun.sun_path);
+ if (encode_single_inet_addr(&waddr.sa, &wsun, !desperatep) == USED)
+ return (-1);
+ /* All is well. */
return (0);
}
address addr;
int port = port_from_sockaddr(sa);
char buf[ADDRBUFSZ];
- int rc;
D( fprintf(stderr, "noip(%d): encode %s (%s)", getpid(),
present_sockaddr(sa, 0, buf, sizeof(buf)),
(f&ENCF_FRESH) ? "FRESH" : "EXISTING"); )
+
+ /* Start making the Unix-domain address. */
sun->sun_family = AF_UNIX;
+
if (port || !(f&ENCF_FRESH)) {
- snprintf(sun->sun_path, sizeof(sun->sun_path), "%s/%s", sockdir,
- present_sockaddr(sa, 0, buf, sizeof(buf)));
- rc = unix_socket_status(sun, 0);
- if (rc == STALE) unlink(sun->sun_path);
- if (rc != USED && !(f&ENCF_FRESH)) {
- wildcard_address(sa->sa_family, &addr.sa);
- port_to_sockaddr(&addr.sa, port);
- snprintf(sun->sun_path, sizeof(sun->sun_path), "%s/%s", sockdir,
- present_sockaddr(&addr.sa, 0, buf, sizeof(buf)));
- if (unix_socket_status(sun, 0) == STALE) unlink(sun->sun_path);
- }
+
+ /* Try the address as given. If it's in use, or we don't necessarily
+ * want an existing socket, then we're done.
+ */
+ if (encode_single_inet_addr(sa, sun, 0) == USED || (f&ENCF_FRESH))
+ goto found;
+
+ /* We're looking for a socket which already exists. Try the
+ * corresponding wildcard address.
+ */
+ wildcard_address(sa->sa_family, &addr.sa);
+ port_to_sockaddr(&addr.sa, port);
+ encode_single_inet_addr(&addr.sa, sun, 0);
+
} else {
+ /* We want a fresh new socket. */
+
+ /* Make a copy of the given address, because we're going to mangle it. */
copy_sockaddr(&addr.sa, sa);
+
+ /* Try a few random-ish port numbers to see if any of them is spare. */
for (i = 0; i < 10; i++) {
port_to_sockaddr(&addr.sa, randrange(minautoport, maxautoport));
if (!encode_unused_inet_addr(&addr.sa, sun, 0)) goto found;
}
+
+ /* Things must be getting tight. Work through all of the autoport range
+ * to see if we can find a spare one. The first time, just do it the
+ * quick way; if that doesn't work, then check harder for stale sockets.
+ */
for (desperatep = 0; desperatep < 2; desperatep++) {
for (i = minautoport; i <= maxautoport; i++) {
port_to_sockaddr(&addr.sa, i);
if (!encode_unused_inet_addr(&addr.sa, sun, 0)) goto found;
}
}
+
+ /* We failed to find any free ports. */
errno = EADDRINUSE;
D( fprintf(stderr, " -- can't resolve\n"); )
return (-1);
- found:;
}
+
+ /* Success. */
+found:
D( fprintf(stderr, " -> `%s'\n", sun->sun_path); )
return (0);
}