bsd-misc.c

   1 /*
   2 **
   3 ** OpenBSD replacement 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 <fcntl.h>
  48 #ifdef HAVE_STDDEF_H
  49 #include <stddef.h>
  50 #endif
  51
  52 #include "xmalloc.h"
  53 #include "ssh.h"
  54 #include "bsd-misc.h"
  55 #include "random.h"
  56
  57 #ifndef HAVE_ARC4RANDOM
  58
  59 typedef struct
  60 {
  61         unsigned int s[256];
  62         int i;
  63         int j;
  64 } rc4_t;
  65
  66 void rc4_key(rc4_t *r, unsigned char *key, int len);
  67 void rc4_getbytes(rc4_t *r, unsigned char *buffer, int len);
  68
  69 static rc4_t *rc4 = NULL;
  70
  71 void rc4_key(rc4_t *r, unsigned char *key, int len)
  72 {
  73         int t;
  74
  75         for(r->i = 0; r->i < 256; r->i++)
  76                 r->s[r->i] = r->i;
  77
  78         r->j = 0;
  79         for(r->i = 0; r->i < 256; r->i++)
  80         {
  81                 r->j = (r->j + r->s[r->i] + key[r->i % len]) % 256;
  82                 t = r->s[r->i];
  83                 r->s[r->i] = r->s[r->j];
  84                 r->s[r->j] = t;
  85         }
  86         r->i = r->j = 0;
  87 }
  88
  89 void rc4_getbytes(rc4_t *r, unsigned char *buffer, int len)
  90 {
  91         int t;
  92         int c;
  93
  94         c = 0;
  95         while(c < len)
  96         {
  97                 r->i = (r->i + 1) % 256;
  98                 r->j = (r->j + r->s[r->i]) % 256;
  99                 t = r->s[r->i];
 100                 r->s[r->i] = r->s[r->j];
 101                 r->s[r->j] = t;
 102
 103                 t = (r->s[r->i] + r->s[r->j]) % 256;
 104
 105                 buffer[c] = r->s[t];
 106                 c++;
 107         }
 108 }
 109
 110 unsigned int arc4random(void)
 111 {
 112         unsigned int r;
 113
 114         if (rc4 == NULL)
 115                 arc4random_stir();
 116
 117         rc4_getbytes(rc4, (unsigned char *)&r, sizeof(r));
 118
 119         return(r);
 120 }
 121
 122 void arc4random_stir(void)
 123 {
 124         unsigned char rand_buf[32];
 125
 126         if (rc4 == NULL)
 127                 rc4 = xmalloc(sizeof(*rc4));
 128
 129         get_random_bytes(rand_buf, sizeof(rand_buf));
 130         rc4_key(rc4, rand_buf, sizeof(rand_buf));
 131         memset(rand_buf, 0, sizeof(rand_buf));
 132 }
 133 #endif /* !HAVE_ARC4RANDOM */
 134
 135 #ifndef HAVE_SETPROCTITLE
 136 void setproctitle(const char *fmt, ...)
 137 {
 138         /* FIXME */
 139 }
 140 #endif /* !HAVE_SETPROCTITLE */
 141
 142 #ifndef HAVE_SETENV
 143 int setenv(const char *name, const char *value, int overwrite)
 144 {
 145         char *env_string;
 146         int result;
 147
 148         /* Don't overwrite existing env. var if overwrite is 0 */
 149         if (!overwrite && (getenv(name) != NULL))
 150                 return(0);
 151
 152         env_string = xmalloc(strlen(name) + strlen(value) + 2);
 153         sprintf(env_string, "%s=%s", name, value);
 154
 155         result = putenv(env_string);
 156
 157         xfree(env_string);
 158
 159         return(result);
 160 }
 161 #endif /* !HAVE_SETENV */
 162
 163 #ifndef HAVE_SETLOGIN
 164 int setlogin(const char *name)
 165 {
 166         return(0);
 167 }
 168 #endif /* !HAVE_SETLOGIN */
 169
 170 #ifndef HAVE_INNETGR
 171 int innetgr(const char *netgroup, const char *host,
 172             const char *user, const char *domain)
 173 {
 174         return(0);
 175 }
 176 #endif /* HAVE_INNETGR */
 177
 178 #if !defined(HAVE_SETEUID) && defined(HAVE_SETREUID)
 179 int seteuid(uid_t euid)
 180 {
 181         return(setreuid(-1,euid));
 182 }
 183 #endif /* !defined(HAVE_SETEUID) && defined(HAVE_SETREUID) */