src/common/compat/socket.h

   1 /*
   2  * Copyright (C) 2011 - David Goulet <dgoulet@efficios.com>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License, version 2 only,
   6  * as published by the Free Software Foundation.
   7  *
   8  * This program is distributed in the hope that it will be useful, but WITHOUT
   9  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  10  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
  11  * more details.
  12  *
  13  * You should have received a copy of the GNU General Public License along
  14  * with this program; if not, write to the Free Software Foundation, Inc.,
  15  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  16  */
  17
  18 #ifndef _COMPAT_SOCKET_H
  19 #define _COMPAT_SOCKET_H
  20
  21 #include <sys/socket.h>
  22 #include <sys/un.h>
  23
  24 #include <common/macros.h>
  25
  26 #ifdef __linux__
  27
  28 #define LTTNG_SOCK_CREDS SCM_CREDENTIALS
  29
  30 typedef struct ucred lttng_sock_cred;
  31
  32 #define LTTNG_SOCK_SET_UID_CRED(c, u) LTTNG_REF(c)->uid = u
  33 #define LTTNG_SOCK_SET_GID_CRED(c, g) LTTNG_REF(c)->gid = g
  34 #define LTTNG_SOCK_SET_PID_CRED(c, p) LTTNG_REF(c)->pid = p
  35
  36 #define LTTNG_SOCK_GET_UID_CRED(c) LTTNG_REF(c)->uid
  37 #define LTTNG_SOCK_GET_GID_CRED(c) LTTNG_REF(c)->gid
  38 #define LTTNG_SOCK_GET_PID_CRED(c) LTTNG_REF(c)->pid
  39
  40 #elif (defined(__FreeBSD__) || defined(__CYGWIN__) || defined(__sun__))
  41
  42 struct lttng_sock_cred {
  43         uid_t uid;
  44         gid_t gid;
  45 };
  46
  47 typedef struct lttng_sock_cred lttng_sock_cred;
  48
  49 #define LTTNG_SOCK_GET_UID_CRED(c) LTTNG_REF(c)->uid
  50 #define LTTNG_SOCK_GET_GID_CRED(c) LTTNG_REF(c)->gid
  51 #define LTTNG_SOCK_GET_PID_CRED(c) -1
  52
  53 #else
  54 #error "Please add support for your OS."
  55 #endif /* __linux__ , __FreeBSD__ */
  56
  57
  58 #ifdef __sun__
  59
  60 # ifndef CMSG_ALIGN
  61 #  ifdef _CMSG_DATA_ALIGN
  62 #   define CMSG_ALIGN(len) _CMSG_DATA_ALIGN(len)
  63 #  else
  64     /* aligning to sizeof (long) is assumed to be portable (fd.o#40235) */
  65 #   define CMSG_ALIGN(len) (((len) + sizeof (long) - 1) & ~(sizeof (long) - 1))
  66 #  endif
  67 #  ifndef CMSG_SPACE
  68 #    define CMSG_SPACE(len) (CMSG_ALIGN (sizeof (struct cmsghdr)) + CMSG_ALIGN (len))
  69 #  endif
  70 #  ifndef CMSG_LEN
  71 #    define CMSG_LEN(len) (CMSG_ALIGN (sizeof (struct cmsghdr)) + (len))
  72 #  endif
  73 # endif
  74
  75
  76 #include <ucred.h>
  77 static int
  78 getpeereid(int s, uid_t *euid, gid_t *gid)
  79 {
  80         int ret = 0;
  81         ucred_t *ucred = NULL;
  82
  83         ret = getpeerucred(s, &ucred);
  84         if (ret == -1) {
  85                 goto end;
  86         }
  87
  88         ret = ucred_geteuid(ucred);
  89         if (ret == -1) {
  90                 goto free;
  91         }
  92         *euid = ret;
  93
  94         ret = ucred_getrgid(ucred);
  95         if (ret == -1) {
  96                 goto free;
  97         }
  98         *gid = ret;
  99         ret = 0;
 100 free:
 101         ucred_free(ucred);
 102 end:
 103         return ret;
 104 }
 105
 106 #endif /* __sun__ */
 107
 108 #endif /* _COMPAT_SOCKET_H */