lib/libc/gen/utxdb.c

   1 /*-
   2  * Copyright (c) 2010 Ed Schouten <ed@FreeBSD.org>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  24  * SUCH DAMAGE.
  25  */
  26
  27 #include <sys/cdefs.h>
  28 __FBSDID("$FreeBSD$");
  29
  30 #include "namespace.h"
  31 #include <sys/endian.h>
  32 #include <sys/param.h>
  33 #include <sys/time.h>
  34 #include <stdlib.h>
  35 #include <string.h>
  36 #include <utmpx.h>
  37 #include "utxdb.h"
  38 #include "un-namespace.h"
  39
  40 #define UTOF_STRING(ut, fu, field) do { \
  41         strncpy((fu)->fu_ ## field, (ut)->ut_ ## field,         \
  42             MIN(sizeof (fu)->fu_ ## field, sizeof (ut)->ut_ ## field)); \
  43 } while (0)
  44 #define UTOF_ID(ut, fu) do { \
  45         memcpy((fu)->fu_id, (ut)->ut_id,                                \
  46             MIN(sizeof (fu)->fu_id, sizeof (ut)->ut_id));               \
  47 } while (0)
  48 #define UTOF_PID(ut, fu) do { \
  49         (fu)->fu_pid = htobe32((ut)->ut_pid);                           \
  50 } while (0)
  51 #define UTOF_TYPE(ut, fu) do { \
  52         (fu)->fu_type = (ut)->ut_type;                                  \
  53 } while (0)
  54 #define UTOF_TV(fu) do { \
  55         struct timeval tv;                                              \
  56         gettimeofday(&tv, NULL);                                        \
  57         (fu)->fu_tv = htobe64((uint64_t)tv.tv_sec * 1000000 +           \
  58             (uint64_t)tv.tv_usec);                                      \
  59 } while (0)
  60
  61 void
  62 utx_to_futx(const struct utmpx *ut, struct futx *fu)
  63 {
  64
  65         memset(fu, 0, sizeof *fu);
  66
  67         switch (ut->ut_type) {
  68         case BOOT_TIME:
  69         case OLD_TIME:
  70         case NEW_TIME:
  71         /* Extension: shutdown time. */
  72         case SHUTDOWN_TIME:
  73                 break;
  74         case USER_PROCESS:
  75                 UTOF_ID(ut, fu);
  76                 UTOF_STRING(ut, fu, user);
  77                 UTOF_STRING(ut, fu, line);
  78                 /* Extension: host name. */
  79                 UTOF_STRING(ut, fu, host);
  80                 UTOF_PID(ut, fu);
  81                 break;
  82         case INIT_PROCESS:
  83                 UTOF_ID(ut, fu);
  84                 UTOF_PID(ut, fu);
  85                 break;
  86         case LOGIN_PROCESS:
  87                 UTOF_ID(ut, fu);
  88                 UTOF_STRING(ut, fu, user);
  89                 UTOF_STRING(ut, fu, line);
  90                 UTOF_PID(ut, fu);
  91                 break;
  92         case DEAD_PROCESS:
  93                 UTOF_ID(ut, fu);
  94                 UTOF_PID(ut, fu);
  95                 break;
  96         default:
  97                 fu->fu_type = EMPTY;
  98                 return;
  99         }
 100
 101         UTOF_TYPE(ut, fu);
 102         UTOF_TV(fu);
 103 }
 104
 105 #define FTOU_STRING(fu, ut, field) do { \
 106         strncpy((ut)->ut_ ## field, (fu)->fu_ ## field,         \
 107             MIN(sizeof (ut)->ut_ ## field - 1, sizeof (fu)->fu_ ## field)); \
 108 } while (0)
 109 #define FTOU_ID(fu, ut) do { \
 110         memcpy((ut)->ut_id, (fu)->fu_id,                                \
 111             MIN(sizeof (ut)->ut_id, sizeof (fu)->fu_id));               \
 112 } while (0)
 113 #define FTOU_PID(fu, ut) do { \
 114         (ut)->ut_pid = be32toh((fu)->fu_pid);                           \
 115 } while (0)
 116 #define FTOU_TYPE(fu, ut) do { \
 117         (ut)->ut_type = (fu)->fu_type;                                  \
 118 } while (0)
 119 #define FTOU_TV(fu, ut) do { \
 120         uint64_t t;                                                     \
 121         t = be64toh((fu)->fu_tv);                                       \
 122         (ut)->ut_tv.tv_sec = t / 1000000;                               \
 123         (ut)->ut_tv.tv_usec = t % 1000000;                              \
 124 } while (0)
 125
 126 struct utmpx *
 127 futx_to_utx(const struct futx *fu)
 128 {
 129 #ifdef __NO_TLS
 130         static struct utmpx *ut;
 131 #else
 132         static _Thread_local struct utmpx *ut;
 133 #endif
 134
 135         if (ut == NULL) {
 136                 ut = calloc(1, sizeof *ut);
 137                 if (ut == NULL)
 138                         return (NULL);
 139         } else
 140                 memset(ut, 0, sizeof *ut);
 141
 142         switch (fu->fu_type) {
 143         case BOOT_TIME:
 144         case OLD_TIME:
 145         case NEW_TIME:
 146         /* Extension: shutdown time. */
 147         case SHUTDOWN_TIME:
 148                 break;
 149         case USER_PROCESS:
 150                 FTOU_ID(fu, ut);
 151                 FTOU_STRING(fu, ut, user);
 152                 FTOU_STRING(fu, ut, line);
 153                 /* Extension: host name. */
 154                 FTOU_STRING(fu, ut, host);
 155                 FTOU_PID(fu, ut);
 156                 break;
 157         case INIT_PROCESS:
 158                 FTOU_ID(fu, ut);
 159                 FTOU_PID(fu, ut);
 160                 break;
 161         case LOGIN_PROCESS:
 162                 FTOU_ID(fu, ut);
 163                 FTOU_STRING(fu, ut, user);
 164                 FTOU_STRING(fu, ut, line);
 165                 FTOU_PID(fu, ut);
 166                 break;
 167         case DEAD_PROCESS:
 168                 FTOU_ID(fu, ut);
 169                 FTOU_PID(fu, ut);
 170                 break;
 171         default:
 172                 ut->ut_type = EMPTY;
 173                 return (ut);
 174         }
 175
 176         FTOU_TYPE(fu, ut);
 177         FTOU_TV(fu, ut);
 178         return (ut);
 179 }