capabilities support
[platform/upstream/libwebsockets.git] / lib / lws-plat-unix.c
index e26b041..ae73dfc 100644 (file)
@@ -3,7 +3,9 @@
 #include <pwd.h>
 #include <grp.h>
 
+#ifdef LWS_WITH_PLUGINS
 #include <dlfcn.h>
+#endif
 #include <dirent.h>
 
 
@@ -33,7 +35,7 @@ lws_send_pipe_choked(struct lws *wsi)
        if (wsi->trunc_len)
                return 1;
 
-       fds.fd = wsi->sock;
+       fds.fd = wsi->desc.sockfd;
        fds.events = POLLOUT;
        fds.revents = 0;
 
@@ -54,14 +56,6 @@ lws_poll_listen_fd(struct lws_pollfd *fd)
        return poll(fd, 1, 0);
 }
 
-/**
- * lws_cancel_service_pt() - Cancel servicing of pending socket activity
- *                             on one thread
- * @wsi:       Cancel service on the thread this wsi is serviced by
- *
- *     This function let a call to lws_service() waiting for a timeout
- *     immediately return.
- */
 LWS_VISIBLE void
 lws_cancel_service_pt(struct lws *wsi)
 {
@@ -72,13 +66,6 @@ lws_cancel_service_pt(struct lws *wsi)
                lwsl_err("Cannot write to dummy pipe");
 }
 
-/**
- * lws_cancel_service() - Cancel ALL servicing of pending socket activity
- * @context:   Websocket context
- *
- *     This function let a call to lws_service() waiting for a timeout
- *     immediately return.
- */
 LWS_VISIBLE void
 lws_cancel_service(struct lws_context *context)
 {
@@ -113,10 +100,10 @@ LWS_VISIBLE void lwsl_emit_syslog(int level, const char *line)
        syslog(syslog_level, "%s", line);
 }
 
-LWS_VISIBLE int
-lws_plat_service_tsi(struct lws_context *context, int timeout_ms, int tsi)
+LWS_VISIBLE LWS_EXTERN int
+_lws_plat_service_tsi(struct lws_context *context, int timeout_ms, int tsi)
 {
-       struct lws_context_per_thread *pt = &context->pt[tsi];
+       struct lws_context_per_thread *pt;
        int n = -1, m, c;
        char buf;
 
@@ -125,11 +112,16 @@ lws_plat_service_tsi(struct lws_context *context, int timeout_ms, int tsi)
        if (!context || !context->vhost_list)
                return 1;
 
+       pt = &context->pt[tsi];
+
+       lws_stats_atomic_bump(context, pt, LWSSTATS_C_SERVICE_ENTRY, 1);
+
        if (timeout_ms < 0)
                goto faked_service;
 
        lws_libev_run(context, tsi);
        lws_libuv_run(context, tsi);
+       lws_libevent_run(context, tsi);
 
        if (!context->service_tid_detected) {
                struct lws _lws;
@@ -143,7 +135,17 @@ lws_plat_service_tsi(struct lws_context *context, int timeout_ms, int tsi)
        }
        context->service_tid = context->service_tid_detected;
 
-       timeout_ms = lws_service_adjust_timeout(context, timeout_ms, tsi);
+       /*
+        * is there anybody with pending stuff that needs service forcing?
+        */
+       if (!lws_service_adjust_timeout(context, 1, tsi)) {
+               /* -1 timeout means just do forced service */
+               _lws_plat_service_tsi(context, -1, pt->tid);
+               /* still somebody left who wants forced service? */
+               if (!lws_service_adjust_timeout(context, 1, pt->tid))
+                       /* yes... come back again quickly */
+                       timeout_ms = 0;
+       }
 
        n = poll(pt->fds, pt->fds_count, timeout_ms);
 
@@ -194,9 +196,15 @@ faked_service:
 }
 
 LWS_VISIBLE int
+lws_plat_check_connection_error(struct lws *wsi)
+{
+       return 0;
+}
+
+LWS_VISIBLE int
 lws_plat_service(struct lws_context *context, int timeout_ms)
 {
-       return lws_plat_service_tsi(context, timeout_ms, 0);
+       return _lws_plat_service_tsi(context, timeout_ms, 0);
 }
 
 LWS_VISIBLE int
@@ -222,7 +230,7 @@ lws_plat_set_socket_options(struct lws_vhost *vhost, int fd)
 #if defined(__APPLE__) || \
     defined(__FreeBSD__) || defined(__FreeBSD_kernel__) || \
     defined(__NetBSD__) || \
-        defined(__CYGWIN__) || defined(__OpenBSD__)
+        defined(__CYGWIN__) || defined(__OpenBSD__) || defined (__sun)
 
                /*
                 * didn't find a way to set these per-socket, need to
@@ -249,10 +257,13 @@ lws_plat_set_socket_options(struct lws_vhost *vhost, int fd)
 
        /* Disable Nagle */
        optval = 1;
-#if !defined(__APPLE__) && \
-    !defined(__FreeBSD__) && !defined(__FreeBSD_kernel__) && \
-    !defined(__NetBSD__) && \
-    !defined(__OpenBSD__)
+#if defined (__sun)
+       if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (const void *)&optval, optlen) < 0)
+               return 1;
+#elif !defined(__APPLE__) && \
+      !defined(__FreeBSD__) && !defined(__FreeBSD_kernel__) &&        \
+      !defined(__NetBSD__) && \
+      !defined(__OpenBSD__)
        if (setsockopt(fd, SOL_TCP, TCP_NODELAY, (const void *)&optval, optlen) < 0)
                return 1;
 #else
@@ -268,9 +279,29 @@ lws_plat_set_socket_options(struct lws_vhost *vhost, int fd)
        return 0;
 }
 
+#if defined(LWS_HAVE_SYS_CAPABILITY_H) && defined(LWS_HAVE_LIBCAP)
+static void
+_lws_plat_apply_caps(int mode, cap_value_t *cv, int count)
+{
+       cap_t caps = cap_get_proc();
+
+       if (!count)
+               return;
+
+       cap_set_flag(caps, mode, count, cv, CAP_SET);
+       cap_set_proc(caps);
+       prctl(PR_SET_KEEPCAPS, 1, 0, 0, 0);
+       cap_free(caps);
+}
+#endif
+
 LWS_VISIBLE void
 lws_plat_drop_app_privileges(struct lws_context_creation_info *info)
 {
+#if defined(LWS_HAVE_SYS_CAPABILITY_H) && defined(LWS_HAVE_LIBCAP)
+       int n;
+#endif
+
        if (info->gid != -1)
                if (setgid(info->gid))
                        lwsl_warn("setgid: %s\n", strerror(LWS_ERRNO));
@@ -279,11 +310,25 @@ lws_plat_drop_app_privileges(struct lws_context_creation_info *info)
                struct passwd *p = getpwuid(info->uid);
 
                if (p) {
+
+#if defined(LWS_HAVE_SYS_CAPABILITY_H) && defined(LWS_HAVE_LIBCAP)
+                       _lws_plat_apply_caps(CAP_PERMITTED, info->caps, info->count_caps);
+#endif
+
                        initgroups(p->pw_name, info->gid);
                        if (setuid(info->uid))
                                lwsl_warn("setuid: %s\n", strerror(LWS_ERRNO));
                        else
                                lwsl_notice("Set privs to user '%s'\n", p->pw_name);
+
+#if defined(LWS_HAVE_SYS_CAPABILITY_H) && defined(LWS_HAVE_LIBCAP)
+                       _lws_plat_apply_caps(CAP_EFFECTIVE, info->caps, info->count_caps);
+
+                       if (info->count_caps)
+                               for (n = 0; n < info->count_caps; n++)
+                                       lwsl_notice("   RETAINING CAPABILITY %d\n", (int)info->caps[n]);
+#endif
+
                } else
                        lwsl_warn("getpwuid: unable to find uid %d", info->uid);
        }
@@ -331,7 +376,7 @@ lws_plat_plugins_init(struct lws_context * context, const char * const *d)
 
                        lwsl_notice("   %s\n", namelist[i]->d_name);
 
-                       snprintf(path, sizeof(path) - 1, "%s/%s", *d,
+                       lws_snprintf(path, sizeof(path) - 1, "%s/%s", *d,
                                 namelist[i]->d_name);
                        l = dlopen(path, RTLD_NOW);
                        if (!l) {
@@ -341,7 +386,7 @@ lws_plat_plugins_init(struct lws_context * context, const char * const *d)
                                goto bail;
                        }
                        /* we could open it, can we get his init function? */
-                       m = snprintf(path, sizeof(path) - 1, "init_%s",
+                       m = lws_snprintf(path, sizeof(path) - 1, "init_%s",
                                     namelist[i]->d_name + 3 /* snip lib... */);
                        path[m - 3] = '\0'; /* snip the .so */
                        initfunc = dlsym(l, path);
@@ -406,7 +451,7 @@ lws_plat_plugins_destroy(struct lws_context * context)
 
        while (plugin) {
                p = plugin;
-               m = snprintf(path, sizeof(path) - 1, "destroy_%s", plugin->name + 3);
+               m = lws_snprintf(path, sizeof(path) - 1, "destroy_%s", plugin->name + 3);
                path[m - 3] = '\0';
                func = dlsym(plugin->l, path);
                if (!func) {
@@ -446,7 +491,9 @@ sigabrt_handler(int x)
 LWS_VISIBLE int
 lws_plat_context_early_init(void)
 {
+#if !defined(LWS_AVOID_SIGPIPE_IGN)
        signal(SIGPIPE, SIG_IGN);
+#endif
 
 //     signal(SIGABRT, sigabrt_handler);
 
@@ -473,11 +520,16 @@ lws_plat_context_late_destroy(struct lws_context *context)
                lws_free(context->lws_lookup);
 
        while (m--) {
-               close(pt->dummy_pipe_fds[0]);
-               close(pt->dummy_pipe_fds[1]);
+               if (pt->dummy_pipe_fds[0])
+                       close(pt->dummy_pipe_fds[0]);
+               if (pt->dummy_pipe_fds[1])
+                       close(pt->dummy_pipe_fds[1]);
                pt++;
        }
-       close(context->fd_random);
+       if (!context->fd_random)
+               lwsl_err("ZERO RANDOM FD\n");
+       if (context->fd_random != LWS_INVALID_FILE)
+               close(context->fd_random);
 }
 
 /* cast a struct sockaddr_in6 * into addr for ipv6 */
@@ -538,7 +590,7 @@ lws_interface_to_sa(int ipv6, const char *ifname, struct sockaddr_in *addr,
        freeifaddrs(ifr);
 
        if (rc == -1) {
-               /* check if bind to IP adddress */
+               /* check if bind to IP address */
 #ifdef LWS_USE_IPV6
                if (inet_pton(AF_INET6, ifname, &addr6->sin6_addr) == 1)
                        rc = 0;
@@ -558,6 +610,7 @@ lws_plat_insert_socket_into_fds(struct lws_context *context, struct lws *wsi)
 
        lws_libev_io(wsi, LWS_EV_START | LWS_EV_READ);
        lws_libuv_io(wsi, LWS_EV_START | LWS_EV_READ);
+       lws_libevent_io(wsi, LWS_EV_START | LWS_EV_READ);
 
        pt->fds[pt->fds_count++].revents = 0;
 }
@@ -570,6 +623,7 @@ lws_plat_delete_socket_from_fds(struct lws_context *context,
 
        lws_libev_io(wsi, LWS_EV_STOP | LWS_EV_READ | LWS_EV_WRITE);
        lws_libuv_io(wsi, LWS_EV_STOP | LWS_EV_READ | LWS_EV_WRITE);
+       lws_libevent_io(wsi, LWS_EV_STOP | LWS_EV_READ | LWS_EV_WRITE);
 
        pt->fds_count--;
 }
@@ -596,65 +650,109 @@ lws_plat_inet_ntop(int af, const void *src, char *dst, int cnt)
        return inet_ntop(af, src, dst, cnt);
 }
 
-static lws_filefd_type
-_lws_plat_file_open(struct lws *wsi, const char *filename,
-                   unsigned long *filelen, int flags)
+LWS_VISIBLE int
+lws_plat_inet_pton(int af, const char *src, void *dst)
+{
+       return inet_pton(af, src, dst);
+}
+
+LWS_VISIBLE lws_fop_fd_t
+_lws_plat_file_open(const struct lws_plat_file_ops *fops, const char *filename,
+                   const char *vpath, lws_fop_flags_t *flags)
 {
        struct stat stat_buf;
-       int ret = open(filename, flags, 0664);
+       int ret = open(filename, (*flags) & LWS_FOP_FLAGS_MASK, 0664);
+       lws_fop_fd_t fop_fd;
 
        if (ret < 0)
-               return LWS_INVALID_FILE;
+               return NULL;
 
-       if (fstat(ret, &stat_buf) < 0) {
-               close(ret);
-               return LWS_INVALID_FILE;
-       }
-       *filelen = stat_buf.st_size;
-       return ret;
+       if (fstat(ret, &stat_buf) < 0)
+               goto bail;
+
+       fop_fd = malloc(sizeof(*fop_fd));
+       if (!fop_fd)
+               goto bail;
+
+       fop_fd->fops = fops;
+       fop_fd->flags = *flags;
+       fop_fd->fd = ret;
+       fop_fd->filesystem_priv = NULL; /* we don't use it */
+       fop_fd->len = stat_buf.st_size;
+       fop_fd->pos = 0;
+
+       return fop_fd;
+
+bail:
+       close(ret);
+       return NULL;
 }
 
-static int
-_lws_plat_file_close(struct lws *wsi, lws_filefd_type fd)
+LWS_VISIBLE int
+_lws_plat_file_close(lws_fop_fd_t *fop_fd)
 {
+       int fd = (*fop_fd)->fd;
+
+       free(*fop_fd);
+       *fop_fd = NULL;
+
        return close(fd);
 }
 
-unsigned long
-_lws_plat_file_seek_cur(struct lws *wsi, lws_filefd_type fd, long offset)
+LWS_VISIBLE lws_fileofs_t
+_lws_plat_file_seek_cur(lws_fop_fd_t fop_fd, lws_fileofs_t offset)
 {
-       return lseek(fd, offset, SEEK_CUR);
+       lws_fileofs_t r;
+
+       if (offset > 0 && offset > fop_fd->len - fop_fd->pos)
+               offset = fop_fd->len - fop_fd->pos;
+
+       if ((lws_fileofs_t)fop_fd->pos + offset < 0)
+               offset = -fop_fd->pos;
+
+       r = lseek(fop_fd->fd, offset, SEEK_CUR);
+
+       if (r >= 0)
+               fop_fd->pos = r;
+       else
+               lwsl_err("error seeking from cur %ld, offset %ld\n",
+                        (long)fop_fd->pos, (long)offset);
+
+       return r;
 }
 
-static int
-_lws_plat_file_read(struct lws *wsi, lws_filefd_type fd, unsigned long *amount,
-                   unsigned char *buf, unsigned long len)
+LWS_VISIBLE int
+_lws_plat_file_read(lws_fop_fd_t fop_fd, lws_filepos_t *amount,
+                   uint8_t *buf, lws_filepos_t len)
 {
        long n;
 
-       n = read((int)fd, buf, len);
+       n = read((int)fop_fd->fd, buf, len);
        if (n == -1) {
                *amount = 0;
                return -1;
        }
-
+       fop_fd->pos += n;
+       lwsl_debug("%s: read %ld of req %ld, pos %ld, len %ld\n", __func__, n,
+                  (long)len, (long)fop_fd->pos, (long)fop_fd->len);
        *amount = n;
 
        return 0;
 }
 
-static int
-_lws_plat_file_write(struct lws *wsi, lws_filefd_type fd, unsigned long *amount,
-                    unsigned char *buf, unsigned long len)
+LWS_VISIBLE int
+_lws_plat_file_write(lws_fop_fd_t fop_fd, lws_filepos_t *amount,
+                    uint8_t *buf, lws_filepos_t len)
 {
        long n;
 
-       n = write((int)fd, buf, len);
+       n = write((int)fop_fd->fd, buf, len);
        if (n == -1) {
                *amount = 0;
                return -1;
        }
 
+       fop_fd->pos += n;
        *amount = n;
 
        return 0;
@@ -677,8 +775,8 @@ lws_plat_init(struct lws_context *context,
                return 1;
        }
 
-       lwsl_notice(" mem: platform fd map: %5u bytes\n",
-                   sizeof(struct lws *) * context->max_fds);
+       lwsl_notice(" mem: platform fd map: %5lu bytes\n",
+                   (unsigned long)(sizeof(struct lws *) * context->max_fds));
        fd = open(SYSTEM_RANDOM_FILEPATH, O_RDONLY);
 
        context->fd_random = fd;
@@ -689,7 +787,8 @@ lws_plat_init(struct lws_context *context,
        }
 
        if (!lws_libev_init_fd_table(context) &&
-           !lws_libuv_init_fd_table(context)) {
+           !lws_libuv_init_fd_table(context) &&
+           !lws_libevent_init_fd_table(context)) {
                /* otherwise libev handled it instead */
 
                while (n--) {
@@ -707,12 +806,6 @@ lws_plat_init(struct lws_context *context,
                }
        }
 
-       context->fops.open      = _lws_plat_file_open;
-       context->fops.close     = _lws_plat_file_close;
-       context->fops.seek_cur  = _lws_plat_file_seek_cur;
-       context->fops.read      = _lws_plat_file_read;
-       context->fops.write     = _lws_plat_file_write;
-
 #ifdef LWS_WITH_PLUGINS
        if (info->plugin_dirs)
                lws_plat_plugins_init(context, info->plugin_dirs);