/*
- * Copyright (C) 2015-2018 "IoT.bzh"
+ * Copyright (C) 2015-2020 "IoT.bzh"
* Author José Bollo <jose.bollo@iot.bzh>
*
* Licensed under the Apache License, Version 2.0 (the "License");
#include "afb-fdev.h"
#include "afb-socket.h"
-#include "afb-systemd.h"
+
+#include "systemd.h"
#include "fdev.h"
#include "verbose.h"
/******************************************************************************/
+/**
+ * known types
+ */
+enum type {
+ /** type internet */
+ Type_Inet,
+
+ /** type systemd */
+ Type_Systemd,
+
+ /** type Unix */
+ Type_Unix
+};
+
+/**
+ * Structure for known entries
+ */
+struct entry
+{
+ /** the known prefix */
+ const char *prefix;
+
+ /** the type of the entry */
+ unsigned type: 2;
+
+ /** should not set SO_REUSEADDR for servers */
+ unsigned noreuseaddr: 1;
+
+ /** should not call listen for servers */
+ unsigned nolisten: 1;
+};
+
+/**
+ * The known entries with the default one at the first place
+ */
+static struct entry entries[] = {
+ {
+ .prefix = "tcp:",
+ .type = Type_Inet
+ },
+ {
+ .prefix = "sd:",
+ .type = Type_Systemd,
+ .noreuseaddr = 1,
+ .nolisten = 1
+ },
+ {
+ .prefix = "unix:",
+ .type = Type_Unix
+ }
+};
+
+/**
+ * It is possible to set explicit api name instead of using the
+ * default one.
+ */
+static const char as_api[] = "?as-api=";
+
+/******************************************************************************/
+
+/**
+ * open a unix domain socket for client or server
+ *
+ * @param spec the specification of the path (prefix with @ for abstract)
+ * @param server 0 for client, server otherwise
+ *
+ * @return the file descriptor number of the socket or -1 in case of error
+ */
static int open_unix(const char *spec, int server)
{
int fd, rc, abstract;
return fd;
}
-static int open_inet(const char *spec, int server)
+/**
+ * open a tcp socket for client or server
+ *
+ * @param spec the specification of the host:port/...
+ * @param server 0 for client, server otherwise
+ *
+ * @return the file descriptor number of the socket or -1 in case of error
+ */
+static int open_tcp(const char *spec, int server, int reuseaddr)
{
int rc, fd;
- const char *service, *host, *api;
+ const char *service, *host, *tail;
struct addrinfo hint, *rai, *iai;
/* scan the uri */
- api = strrchr(spec, '/');
- service = strrchr(spec, ':');
- if (api == NULL || service == NULL || api < service) {
+ tail = strchrnul(spec, '/');
+ service = strchr(spec, ':');
+ if (tail == NULL || service == NULL || tail < service) {
errno = EINVAL;
return -1;
}
host = strndupa(spec, service++ - spec);
- service = strndupa(service, api - service);
+ service = strndupa(service, tail - service);
/* get addr */
memset(&hint, 0, sizeof hint);
hint.ai_family = AF_INET;
hint.ai_socktype = SOCK_STREAM;
+ if (server) {
+ hint.ai_flags = AI_PASSIVE;
+ if (host[0] == 0 || (host[0] == '*' && host[1] == 0))
+ host = NULL;
+ }
rc = getaddrinfo(host, service, &hint, &rai);
if (rc != 0) {
errno = EINVAL;
fd = socket(iai->ai_family, iai->ai_socktype, iai->ai_protocol);
if (fd >= 0) {
if (server) {
+ if (reuseaddr) {
+ rc = 1;
+ setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &rc, sizeof rc);
+ }
rc = bind(fd, iai->ai_addr, iai->ai_addrlen);
} else {
rc = connect(fd, iai->ai_addr, iai->ai_addrlen);
return -1;
}
+/**
+ * open a systemd socket for server
+ *
+ * @param spec the specification of the systemd name
+ *
+ * @return the file descriptor number of the socket or -1 in case of error
+ */
static int open_systemd(const char *spec)
{
#if defined(NO_SYSTEMD_ACTIVATION)
errno = EAFNOSUPPORT;
- fd = -1;
+ return -1;
#else
- return afb_systemd_fds_for(spec);
+ return systemd_fds_for(spec);
#endif
}
/******************************************************************************/
-enum type {
- Type_Inet,
- Type_Systemd,
- Type_Unix
-};
-
-struct entry
+/**
+ * Get the entry of the uri by searching to its prefix
+ *
+ * @param uri the searched uri
+ * @param offset where to store the prefix length
+ * @param scheme the default scheme to use if none is set in uri (can be NULL)
+ *
+ * @return the found entry or the default one
+ */
+static struct entry *get_entry(const char *uri, int *offset, const char *scheme)
{
- const char *prefix;
- unsigned type: 2;
- unsigned noreuseaddr: 1;
- unsigned nolisten: 1;
-};
-
-static struct entry entries[] = { /* default at first place */
- {
- .prefix = "tcp:",
- .type = Type_Inet
- },
- {
- .prefix = "sd:",
- .type = Type_Systemd,
- .noreuseaddr = 1,
- .nolisten = 1
- },
- {
- .prefix = "unix:",
- .type = Type_Unix
+ int len, i, deflen;
+
+ /* search as prefix of URI */
+ i = (int)(sizeof entries / sizeof * entries);
+ while (i) {
+ i--;
+ len = (int)strlen(entries[i].prefix);
+ if (!strncmp(uri, entries[i].prefix, len))
+ goto end; /* found */
}
-};
-
-/******************************************************************************/
-/* get the entry of the uri by searching to its prefix */
-static struct entry *get_entry(const char *uri, int *offset)
-{
- int l, search = 1, i = (int)(sizeof entries / sizeof * entries);
+ /* not a prefix of uri */
+ len = 0;
- while (search) {
- if (!i) {
- l = 0;
- search = 0;
- } else {
+ /* search default scheme if given and valid */
+ if (scheme && *scheme) {
+ deflen = (int)strlen(scheme);
+ deflen += (scheme[deflen - 1] != ':'); /* add virtual trailing colon */
+ i = (int)(sizeof entries / sizeof * entries);
+ while (i) {
i--;
- l = (int)strlen(entries[i].prefix);
- search = strncmp(uri, entries[i].prefix, l);
+ if (deflen == (int)strlen(entries[i].prefix)
+ && !strncmp(scheme, entries[i].prefix, deflen - 1))
+ goto end; /* found */
}
}
- *offset = l;
+end:
+ *offset = len;
return &entries[i];
}
-static int open_any(const char *uri, int server)
+/**
+ * open socket for client or server
+ *
+ * @param uri the specification of the socket
+ * @param server 0 for client, server otherwise
+ * @param scheme the default scheme to use if none is set in uri (can be NULL)
+ *
+ * @return the file descriptor number of the socket or -1 in case of error
+ */
+static int open_uri(const char *uri, int server, const char *scheme)
{
- int fd, rc, offset;
+ int fd, offset;
struct entry *e;
+ const char *api;
/* search for the entry */
- e = get_entry(uri, &offset);
+ e = get_entry(uri, &offset, scheme);
/* get the names */
-
uri += offset;
+ api = strstr(uri, as_api);
+ if (api)
+ uri = strndupa(uri, api - uri);
/* open the socket */
switch (e->type) {
fd = open_unix(uri, server);
break;
case Type_Inet:
- fd = open_inet(uri, server);
+ fd = open_tcp(uri, server, !e->noreuseaddr);
break;
case Type_Systemd:
- fd = open_systemd(uri);
+ if (server)
+ fd = open_systemd(uri);
+ else {
+ errno = EINVAL;
+ fd = -1;
+ }
break;
default:
errno = EAFNOSUPPORT;
fcntl(fd, F_SETFD, FD_CLOEXEC);
fcntl(fd, F_SETFL, O_NONBLOCK);
if (server) {
- if (!e->noreuseaddr) {
- rc = 1;
- setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &rc, sizeof rc);
- }
if (!e->nolisten)
listen(fd, BACKLOG);
}
return fd;
}
-struct fdev *afb_socket_open(const char *uri, int server)
+/**
+ * open socket for client or server
+ *
+ * @param uri the specification of the socket
+ * @param server 0 for client, server otherwise
+ * @param scheme the default scheme to use if none is set in uri (can be NULL)
+ *
+ * @return the file descriptor number of the socket or -1 in case of error
+ */
+int afb_socket_open_scheme(const char *uri, int server, const char *scheme)
{
- int fd;
- struct fdev *fdev;
-
- fd = open_any(uri, server);
+ int fd = open_uri(uri, server, scheme);
if (fd < 0)
- goto error;
+ ERROR("can't open %s socket for %s", server ? "server" : "client", uri);
+ return fd;
+}
- fdev = afb_fdev_create(fd);
- if (!fdev)
- goto error2;
+/**
+ * open socket for client or server
+ *
+ * @param uri the specification of the socket
+ * @param server 0 for client, server otherwise
+ * @param scheme the default scheme to use if none is set in uri (can be NULL)
+ *
+ * @return the fdev of the socket or NULL in case of error
+ */
+struct fdev *afb_socket_open_fdev_scheme(const char *uri, int server, const char *scheme)
+{
+ struct fdev *fdev;
+ int fd;
+ fd = afb_socket_open_scheme(uri, server, scheme);
+ if (fd < 0)
+ fdev = NULL;
+ else {
+ fdev = afb_fdev_create(fd);
+ if (!fdev) {
+ close(fd);
+ ERROR("can't make %s socket for %s", server ? "server" : "client", uri);
+ }
+ }
return fdev;
-
-error2:
- close(fd);
-error:
- ERROR("can't make %s socket for %s", server ? "server" : "client", uri);
- return NULL;
}
+/**
+ * Get the api name of the uri
+ *
+ * @param uri the specification of the socket
+ *
+ * @return the api name or NULL if none can be deduced
+ */
+const char *afb_socket_api(const char *uri)
+{
+ int offset;
+ const char *api;
+ struct entry *entry;
+
+ entry = get_entry(uri, &offset, NULL);
+ uri += offset;
+ uri += (entry->type == Type_Unix && *uri == '@');
+ api = strstr(uri, as_api);
+ if (api)
+ api += sizeof as_api - 1;
+ else {
+ api = strrchr(uri, '/');
+ if (api)
+ api++;
+ else
+ api = uri;
+ if (strchr(api, ':'))
+ api = NULL;
+ }
+ return api;
+}