[CRIU] [PATCH 05/78] scm: Move fd-sending core code AS IS into include/common/
Cyrill Gorcunov
gorcunov at openvz.org
Mon Nov 7 08:35:50 PST 2016
From: Pavel Emelyanov <xemul at virtuozzo.com>
Signed-off-by: Pavel Emelyanov <xemul at virtuozzo.com>
---
criu/pie/util-fd.c | 147 +-------------------------------------------
include/common/scm-code.c | 151 ++++++++++++++++++++++++++++++++++++++++++++++
2 files changed, 152 insertions(+), 146 deletions(-)
create mode 100644 include/common/scm-code.c
diff --git a/criu/pie/util-fd.c b/criu/pie/util-fd.c
index f3f9c2e39527..c9813602213f 100644
--- a/criu/pie/util-fd.c
+++ b/criu/pie/util-fd.c
@@ -21,149 +21,4 @@
#include "common/bug.h"
-static void scm_fdset_init_chunk(struct scm_fdset *fdset, int nr_fds, bool with_flags)
-{
- struct cmsghdr *cmsg;
-
- fdset->hdr.msg_controllen = CMSG_LEN(sizeof(int) * nr_fds);
-
- cmsg = CMSG_FIRSTHDR(&fdset->hdr);
- cmsg->cmsg_len = fdset->hdr.msg_controllen;
-
- fdset->iov.iov_len = with_flags ? (sizeof(struct fd_opts) * nr_fds) : 1;
-}
-
-static int *scm_fdset_init(struct scm_fdset *fdset, struct sockaddr_un *saddr,
- int saddr_len, bool with_flags)
-{
- struct cmsghdr *cmsg;
-
- BUILD_BUG_ON(sizeof(fdset->msg_buf) < (CMSG_SPACE(sizeof(int) * CR_SCM_MAX_FD)));
-
- fdset->iov.iov_base = fdset->opts;
-
- fdset->hdr.msg_iov = &fdset->iov;
- fdset->hdr.msg_iovlen = 1;
- fdset->hdr.msg_name = (struct sockaddr *)saddr;
- fdset->hdr.msg_namelen = saddr_len;
-
- fdset->hdr.msg_control = &fdset->msg_buf;
- fdset->hdr.msg_controllen = CMSG_LEN(sizeof(int) * CR_SCM_MAX_FD);
-
- cmsg = CMSG_FIRSTHDR(&fdset->hdr);
- cmsg->cmsg_len = fdset->hdr.msg_controllen;
- cmsg->cmsg_level = SOL_SOCKET;
- cmsg->cmsg_type = SCM_RIGHTS;
-
- return (int *)CMSG_DATA(cmsg);
-}
-
-int send_fds(int sock, struct sockaddr_un *saddr, int len,
- int *fds, int nr_fds, bool with_flags)
-{
- struct scm_fdset fdset;
- int *cmsg_data;
- int i, min_fd, ret;
-
- cmsg_data = scm_fdset_init(&fdset, saddr, len, with_flags);
- for (i = 0; i < nr_fds; i += min_fd) {
- min_fd = min(CR_SCM_MAX_FD, nr_fds - i);
- scm_fdset_init_chunk(&fdset, min_fd, with_flags);
- builtin_memcpy(cmsg_data, &fds[i], sizeof(int) * min_fd);
-
- if (with_flags) {
- int j;
-
- for (j = 0; j < min_fd; j++) {
- int flags, fd = fds[i + j];
- struct fd_opts *p = fdset.opts + j;
- struct f_owner_ex owner_ex;
- u32 v[2];
-
- flags = __sys(fcntl)(fd, F_GETFD, 0);
- if (flags < 0) {
- pr_err("fcntl(%d, F_GETFD) -> %d\n", fd, flags);
- return -1;
- }
-
- p->flags = (char)flags;
-
- ret = __sys(fcntl)(fd, F_GETOWN_EX, (long)&owner_ex);
- if (ret) {
- pr_err("fcntl(%d, F_GETOWN_EX) -> %d\n", fd, ret);
- return -1;
- }
-
- /*
- * Simple case -- nothing is changed.
- */
- if (owner_ex.pid == 0) {
- p->fown.pid = 0;
- continue;
- }
-
- ret = __sys(fcntl)(fd, F_GETOWNER_UIDS, (long)&v);
- if (ret) {
- pr_err("fcntl(%d, F_GETOWNER_UIDS) -> %d\n", fd, ret);
- return -1;
- }
-
- p->fown.uid = v[0];
- p->fown.euid = v[1];
- p->fown.pid_type = owner_ex.type;
- p->fown.pid = owner_ex.pid;
- }
- }
-
- ret = __sys(sendmsg)(sock, &fdset.hdr, 0);
- if (ret <= 0)
- return ret ? : -1;
- }
-
- return 0;
-}
-
-int recv_fds(int sock, int *fds, int nr_fds, struct fd_opts *opts)
-{
- struct scm_fdset fdset;
- struct cmsghdr *cmsg;
- int *cmsg_data;
- int ret;
- int i, min_fd;
-
- cmsg_data = scm_fdset_init(&fdset, NULL, 0, opts != NULL);
- for (i = 0; i < nr_fds; i += min_fd) {
- min_fd = min(CR_SCM_MAX_FD, nr_fds - i);
- scm_fdset_init_chunk(&fdset, min_fd, opts != NULL);
-
- ret = __sys(recvmsg)(sock, &fdset.hdr, 0);
- if (ret <= 0)
- return ret ? : -1;
-
- cmsg = CMSG_FIRSTHDR(&fdset.hdr);
- if (!cmsg || cmsg->cmsg_type != SCM_RIGHTS)
- return -EINVAL;
- if (fdset.hdr.msg_flags & MSG_CTRUNC)
- return -ENFILE;
-
- min_fd = (cmsg->cmsg_len - sizeof(struct cmsghdr)) / sizeof(int);
- /*
- * In case if kernel screwed the recipient, most probably
- * the caller stack frame will be overwriten, just scream
- * and exit.
- *
- * FIXME Need to sanitize util.h to be able to include it
- * into files which do not have glibc and a couple of
- * sys_write_ helpers. Meawhile opencoded BUG_ON here.
- */
- BUG_ON(min_fd > CR_SCM_MAX_FD);
-
- if (unlikely(min_fd <= 0))
- return -1;
- builtin_memcpy(&fds[i], cmsg_data, sizeof(int) * min_fd);
- if (opts)
- builtin_memcpy(opts + i, fdset.opts, sizeof(struct fd_opts) * min_fd);
- }
-
- return 0;
-}
+#include "common/scm-code.c"
diff --git a/include/common/scm-code.c b/include/common/scm-code.c
new file mode 100644
index 000000000000..5884f6974561
--- /dev/null
+++ b/include/common/scm-code.c
@@ -0,0 +1,151 @@
+#ifndef __sys
+#error "The __sys macro is required"
+#endif
+
+static void scm_fdset_init_chunk(struct scm_fdset *fdset, int nr_fds, bool with_flags)
+{
+ struct cmsghdr *cmsg;
+
+ fdset->hdr.msg_controllen = CMSG_LEN(sizeof(int) * nr_fds);
+
+ cmsg = CMSG_FIRSTHDR(&fdset->hdr);
+ cmsg->cmsg_len = fdset->hdr.msg_controllen;
+
+ fdset->iov.iov_len = with_flags ? (sizeof(struct fd_opts) * nr_fds) : 1;
+}
+
+static int *scm_fdset_init(struct scm_fdset *fdset, struct sockaddr_un *saddr,
+ int saddr_len, bool with_flags)
+{
+ struct cmsghdr *cmsg;
+
+ BUILD_BUG_ON(sizeof(fdset->msg_buf) < (CMSG_SPACE(sizeof(int) * CR_SCM_MAX_FD)));
+
+ fdset->iov.iov_base = fdset->opts;
+
+ fdset->hdr.msg_iov = &fdset->iov;
+ fdset->hdr.msg_iovlen = 1;
+ fdset->hdr.msg_name = (struct sockaddr *)saddr;
+ fdset->hdr.msg_namelen = saddr_len;
+
+ fdset->hdr.msg_control = &fdset->msg_buf;
+ fdset->hdr.msg_controllen = CMSG_LEN(sizeof(int) * CR_SCM_MAX_FD);
+
+ cmsg = CMSG_FIRSTHDR(&fdset->hdr);
+ cmsg->cmsg_len = fdset->hdr.msg_controllen;
+ cmsg->cmsg_level = SOL_SOCKET;
+ cmsg->cmsg_type = SCM_RIGHTS;
+
+ return (int *)CMSG_DATA(cmsg);
+}
+
+int send_fds(int sock, struct sockaddr_un *saddr, int len,
+ int *fds, int nr_fds, bool with_flags)
+{
+ struct scm_fdset fdset;
+ int *cmsg_data;
+ int i, min_fd, ret;
+
+ cmsg_data = scm_fdset_init(&fdset, saddr, len, with_flags);
+ for (i = 0; i < nr_fds; i += min_fd) {
+ min_fd = min(CR_SCM_MAX_FD, nr_fds - i);
+ scm_fdset_init_chunk(&fdset, min_fd, with_flags);
+ builtin_memcpy(cmsg_data, &fds[i], sizeof(int) * min_fd);
+
+ if (with_flags) {
+ int j;
+
+ for (j = 0; j < min_fd; j++) {
+ int flags, fd = fds[i + j];
+ struct fd_opts *p = fdset.opts + j;
+ struct f_owner_ex owner_ex;
+ uint32_t v[2];
+
+ flags = __sys(fcntl)(fd, F_GETFD, 0);
+ if (flags < 0) {
+ pr_err("fcntl(%d, F_GETFD) -> %d\n", fd, flags);
+ return -1;
+ }
+
+ p->flags = (char)flags;
+
+ ret = __sys(fcntl)(fd, F_GETOWN_EX, (long)&owner_ex);
+ if (ret) {
+ pr_err("fcntl(%d, F_GETOWN_EX) -> %d\n", fd, ret);
+ return -1;
+ }
+
+ /*
+ * Simple case -- nothing is changed.
+ */
+ if (owner_ex.pid == 0) {
+ p->fown.pid = 0;
+ continue;
+ }
+
+ ret = __sys(fcntl)(fd, F_GETOWNER_UIDS, (long)&v);
+ if (ret) {
+ pr_err("fcntl(%d, F_GETOWNER_UIDS) -> %d\n", fd, ret);
+ return -1;
+ }
+
+ p->fown.uid = v[0];
+ p->fown.euid = v[1];
+ p->fown.pid_type = owner_ex.type;
+ p->fown.pid = owner_ex.pid;
+ }
+ }
+
+ ret = __sys(sendmsg)(sock, &fdset.hdr, 0);
+ if (ret <= 0)
+ return ret ? : -1;
+ }
+
+ return 0;
+}
+
+int recv_fds(int sock, int *fds, int nr_fds, struct fd_opts *opts)
+{
+ struct scm_fdset fdset;
+ struct cmsghdr *cmsg;
+ int *cmsg_data;
+ int ret;
+ int i, min_fd;
+
+ cmsg_data = scm_fdset_init(&fdset, NULL, 0, opts != NULL);
+ for (i = 0; i < nr_fds; i += min_fd) {
+ min_fd = min(CR_SCM_MAX_FD, nr_fds - i);
+ scm_fdset_init_chunk(&fdset, min_fd, opts != NULL);
+
+ ret = __sys(recvmsg)(sock, &fdset.hdr, 0);
+ if (ret <= 0)
+ return ret ? : -1;
+
+ cmsg = CMSG_FIRSTHDR(&fdset.hdr);
+ if (!cmsg || cmsg->cmsg_type != SCM_RIGHTS)
+ return -EINVAL;
+ if (fdset.hdr.msg_flags & MSG_CTRUNC)
+ return -ENFILE;
+
+ min_fd = (cmsg->cmsg_len - sizeof(struct cmsghdr)) / sizeof(int);
+ /*
+ * In case if kernel screwed the recipient, most probably
+ * the caller stack frame will be overwriten, just scream
+ * and exit.
+ *
+ * FIXME Need to sanitize util.h to be able to include it
+ * into files which do not have glibc and a couple of
+ * sys_write_ helpers. Meawhile opencoded BUG_ON here.
+ */
+ BUG_ON(min_fd > CR_SCM_MAX_FD);
+
+ if (unlikely(min_fd <= 0))
+ return -1;
+ builtin_memcpy(&fds[i], cmsg_data, sizeof(int) * min_fd);
+ if (opts)
+ builtin_memcpy(opts + i, fdset.opts, sizeof(struct fd_opts) * min_fd);
+ }
+
+ return 0;
+}
+
--
2.7.4
More information about the CRIU
mailing list