bsd-misc.c

   1 /*
   2 **
   3 ** OpenBSD emulation routines
   4 **
   5 ** Damien Miller <djm@ibs.com.au>
   6 **
   7 ** Copyright 1999 Damien Miller
   8 ** Copyright 1999 Internet Business Solutions
   9 **
  10 ** Permission is hereby granted, free of charge, to any person
  11 ** obtaining a copy of this software and associated documentation
  12 ** files (the "Software"), to deal in the Software without
  13 ** restriction, including without limitation the rights to use, copy,
  14 ** modify, merge, publish, distribute, sublicense, and/or sell copies
  15 ** of the Software, and to permit persons to whom the Software is
  16 ** furnished to do so, subject to the following conditions:
  17 **
  18 ** The above copyright notice and this permission notice shall be
  19 ** included in all copies or substantial portions of the Software.
  20 **
  21 ** THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY
  22 ** KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE
  23 ** WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE
  24 ** AND NONINFRINGEMENT.  IN NO EVENT SHALL DAMIEN MILLER OR INTERNET
  25 ** BUSINESS SOLUTIONS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  26 ** LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
  27 ** ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE
  28 ** OR OTHER DEALINGS IN THE SOFTWARE.
  29 **
  30 ** Except as contained in this notice, the name of Internet Business
  31 ** Solutions shall not be used in advertising or otherwise to promote
  32 ** the sale, use or other dealings in this Software without prior
  33 ** written authorization from Internet Business Solutions.
  34 **
  35 */
  36
  37 #include "config.h"
  38
  39 #include <stdio.h>
  40 #include <stdlib.h>
  41 #include <string.h>
  42 #include <errno.h>
  43 #include <unistd.h>
  44
  45 #include <sys/types.h>
  46 #include <sys/stat.h>
  47 #include <sys/socket.h>
  48 #include <sys/un.h>
  49 #include <fcntl.h>
  50 #ifdef HAVE_STDDEF_H
  51 #include <stddef.h>
  52 #endif
  53
  54 #include "xmalloc.h"
  55 #include "ssh.h"
  56 #include "bsd-misc.h"
  57
  58 #ifndef offsetof
  59 #define offsetof(type, member) ((size_t) &((type *)0)->member)
  60 #endif
  61
  62 #ifndef HAVE_ARC4RANDOM
  63
  64 typedef struct
  65 {
  66         unsigned int s[256];
  67         int i;
  68         int j;
  69 } rc4_t;
  70
  71 void get_random_bytes(unsigned char *buf, int len);
  72 void rc4_key(rc4_t *r, unsigned char *key, int len);
  73 void rc4_getbytes(rc4_t *r, unsigned char *buffer, int len);
  74
  75 static rc4_t *rc4 = NULL;
  76
  77 void rc4_key(rc4_t *r, unsigned char *key, int len)
  78 {
  79         int t;
  80
  81         for(r->i = 0; r->i < 256; r->i++)
  82                 r->s[r->i] = r->i;
  83
  84         r->j = 0;
  85         for(r->i = 0; r->i < 256; r->i++)
  86         {
  87                 r->j = (r->j + r->s[r->i] + key[r->i % len]) % 256;
  88                 t = r->s[r->i];
  89                 r->s[r->i] = r->s[r->j];
  90                 r->s[r->j] = t;
  91         }
  92         r->i = r->j = 0;
  93 }
  94
  95 void rc4_getbytes(rc4_t *r, unsigned char *buffer, int len)
  96 {
  97         int t;
  98         int c;
  99
 100         c = 0;
 101         while(c < len)
 102         {
 103                 r->i = (r->i + 1) % 256;
 104                 r->j = (r->j + r->s[r->i]) % 256;
 105                 t = r->s[r->i];
 106                 r->s[r->i] = r->s[r->j];
 107                 r->s[r->j] = t;
 108
 109                 t = (r->s[r->i] + r->s[r->j]) % 256;
 110
 111                 buffer[c] = r->s[t];
 112                 c++;
 113         }
 114 }
 115
 116 unsigned int arc4random(void)
 117 {
 118         unsigned int r;
 119
 120         if (rc4 == NULL)
 121                 arc4random_stir();
 122
 123         rc4_getbytes(rc4, (unsigned char *)&r, sizeof(r));
 124
 125         return(r);
 126 }
 127
 128 void arc4random_stir(void)
 129 {
 130         unsigned char rand_buf[32];
 131
 132         if (rc4 == NULL)
 133                 rc4 = xmalloc(sizeof(*rc4));
 134
 135         get_random_bytes(rand_buf, sizeof(rand_buf));
 136         rc4_key(rc4, rand_buf, sizeof(rand_buf));
 137 }
 138
 139 void get_random_bytes(unsigned char *buf, int len)
 140 {
 141         static int random_pool;
 142         int c;
 143 #ifdef HAVE_EGD
 144         char egd_message[2] = { 0x02, 0x00 };
 145         struct sockaddr_un addr;
 146         int addr_len;
 147
 148         memset(&addr, '\0', sizeof(addr));
 149         addr.sun_family = AF_UNIX;
 150
 151         /* FIXME: compile time check? */
 152         if (sizeof(RANDOM_POOL) > sizeof(addr.sun_path))
 153                 fatal("Random pool path is too long");
 154
 155         strcpy(addr.sun_path, RANDOM_POOL);
 156
 157         addr_len = offsetof(struct sockaddr_un, sun_path) + sizeof(RANDOM_POOL);
 158
 159         random_pool = socket(AF_UNIX, SOCK_STREAM, 0);
 160
 161         if (random_pool == -1)
 162                 fatal("Couldn't create AF_UNIX socket: %s", strerror(errno));
 163
 164         if (connect(random_pool, (struct sockaddr*)&addr, addr_len) == -1)
 165                 fatal("Couldn't connect to EGD socket \"%s\": %s", addr.sun_path, strerror(errno));
 166
 167         if (len > 255)
 168                 fatal("Too many bytes to read from EGD");
 169
 170         /* Send blocking read request to EGD */
 171         egd_message[1] = len;
 172
 173         c = atomicio(write, random_pool, egd_message, sizeof(egd_message));
 174         if (c == -1)
 175                 fatal("Couldn't write to EGD socket \"%s\": %s", RANDOM_POOL, strerror(errno));
 176
 177 #else /* HAVE_EGD */
 178
 179         random_pool = open(RANDOM_POOL, O_RDONLY);
 180         if (random_pool == -1)
 181                 fatal("Couldn't open random pool \"%s\": %s", RANDOM_POOL, strerror(errno));
 182
 183 #endif /* HAVE_EGD */
 184
 185         c = atomicio(read, random_pool, buf, len);
 186         if (c <= 0)
 187                 fatal("Couldn't read from random pool \"%s\": %s", RANDOM_POOL, strerror(errno));
 188
 189         close(random_pool);
 190 }
 191 #endif /* !HAVE_ARC4RANDOM */
 192
 193 #ifndef HAVE_SETPROCTITLE
 194 void setproctitle(const char *fmt, ...)
 195 {
 196         /* FIXME */
 197 }
 198 #endif /* !HAVE_SETPROCTITLE */
 199
 200 #ifndef HAVE_SETENV
 201 int setenv(const char *name, const char *value, int overwrite)
 202 {
 203         char *env_string;
 204         int result;
 205
 206         /* Don't overwrite existing env. var if overwrite is 0 */
 207         if (!overwrite && (getenv(name) != NULL))
 208                 return(0);
 209
 210         env_string = xmalloc(strlen(name) + strlen(value) + 2);
 211         sprintf(env_string, "%s=%s", name, value);
 212
 213         result = putenv(env_string);
 214
 215         xfree(env_string);
 216
 217         return(result);
 218 }
 219 #endif /* !HAVE_SETENV */
 220
 221 #ifndef HAVE_SETLOGIN
 222 int setlogin(const char *name)
 223 {
 224         return(0);
 225 }
 226 #endif /* !HAVE_SETLOGIN */
 227
 228 #ifndef HAVE_INNETGR
 229 int innetgr(const char *netgroup, const char *host,
 230             const char *user, const char *domain)
 231 {
 232         return(0);
 233 }
 234 #endif /* HAVE_INNETGR */
 235
 236 #if !defined(HAVE_SETEUID) && defined(HAVE_SETREUID)
 237 int seteuid(uid_t euid)
 238 {
 239         return(setreuid(-1,euid));
 240 }
 241 #endif /* !defined(HAVE_SETEUID) && defined(HAVE_SETREUID) */