Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 1 | /* |
| 2 | ** |
Damien Miller | 4e61b79 | 2000-01-29 20:13:40 +1100 | [diff] [blame^] | 3 | ** OpenBSD replacement routines |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 4 | ** |
| 5 | ** Damien Miller <djm@ibs.com.au> |
| 6 | ** |
Damien Miller | f20c2aa | 2000-01-06 20:28:41 +1100 | [diff] [blame] | 7 | ** Copyright 1999 Damien Miller |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 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 | |
Damien Miller | 6b85a7f | 2000-01-02 11:45:33 +1100 | [diff] [blame] | 37 | #include "config.h" |
| 38 | |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 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> |
Damien Miller | 58fc473 | 1999-11-19 12:05:01 +1100 | [diff] [blame] | 47 | #include <sys/socket.h> |
| 48 | #include <sys/un.h> |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 49 | #include <fcntl.h> |
Damien Miller | 1fa154b | 2000-01-23 10:32:03 +1100 | [diff] [blame] | 50 | #ifdef HAVE_STDDEF_H |
| 51 | #include <stddef.h> |
| 52 | #endif |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 53 | |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 54 | #include "xmalloc.h" |
Damien Miller | ab18c41 | 1999-11-11 10:40:23 +1100 | [diff] [blame] | 55 | #include "ssh.h" |
Damien Miller | 753331e | 1999-12-30 01:29:35 +1100 | [diff] [blame] | 56 | #include "bsd-misc.h" |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 57 | |
Damien Miller | 58fc473 | 1999-11-19 12:05:01 +1100 | [diff] [blame] | 58 | #ifndef offsetof |
| 59 | #define offsetof(type, member) ((size_t) &((type *)0)->member) |
| 60 | #endif |
| 61 | |
Damien Miller | e413cba | 1999-10-28 14:12:54 +1000 | [diff] [blame] | 62 | #ifndef HAVE_ARC4RANDOM |
| 63 | |
Damien Miller | 037a0dc | 1999-12-07 15:38:31 +1100 | [diff] [blame] | 64 | typedef struct |
| 65 | { |
| 66 | unsigned int s[256]; |
| 67 | int i; |
| 68 | int j; |
| 69 | } rc4_t; |
| 70 | |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 71 | void get_random_bytes(unsigned char *buf, int len); |
Damien Miller | 037a0dc | 1999-12-07 15:38:31 +1100 | [diff] [blame] | 72 | void rc4_key(rc4_t *r, unsigned char *key, int len); |
| 73 | void rc4_getbytes(rc4_t *r, unsigned char *buffer, int len); |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 74 | |
| 75 | static rc4_t *rc4 = NULL; |
| 76 | |
Damien Miller | 037a0dc | 1999-12-07 15:38:31 +1100 | [diff] [blame] | 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 | |
Damien Miller | 3d112ef | 1999-10-28 13:20:30 +1000 | [diff] [blame] | 116 | unsigned int arc4random(void) |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 117 | { |
Damien Miller | 3d112ef | 1999-10-28 13:20:30 +1000 | [diff] [blame] | 118 | unsigned int r; |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 119 | |
| 120 | if (rc4 == NULL) |
| 121 | arc4random_stir(); |
| 122 | |
Damien Miller | 3d112ef | 1999-10-28 13:20:30 +1000 | [diff] [blame] | 123 | rc4_getbytes(rc4, (unsigned char *)&r, sizeof(r)); |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 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 | { |
Damien Miller | 58fc473 | 1999-11-19 12:05:01 +1100 | [diff] [blame] | 141 | static int random_pool; |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 142 | int c; |
Damien Miller | ab18c41 | 1999-11-11 10:40:23 +1100 | [diff] [blame] | 143 | #ifdef HAVE_EGD |
| 144 | char egd_message[2] = { 0x02, 0x00 }; |
Damien Miller | 58fc473 | 1999-11-19 12:05:01 +1100 | [diff] [blame] | 145 | struct sockaddr_un addr; |
| 146 | int addr_len; |
| 147 | |
| 148 | memset(&addr, '\0', sizeof(addr)); |
| 149 | addr.sun_family = AF_UNIX; |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 150 | |
Damien Miller | 58fc473 | 1999-11-19 12:05:01 +1100 | [diff] [blame] | 151 | /* FIXME: compile time check? */ |
| 152 | if (sizeof(RANDOM_POOL) > sizeof(addr.sun_path)) |
| 153 | fatal("Random pool path is too long"); |
| 154 | |
Damien Miller | d71b12e | 1999-11-22 15:24:34 +1100 | [diff] [blame] | 155 | strcpy(addr.sun_path, RANDOM_POOL); |
Damien Miller | 58fc473 | 1999-11-19 12:05:01 +1100 | [diff] [blame] | 156 | |
| 157 | addr_len = offsetof(struct sockaddr_un, sun_path) + sizeof(RANDOM_POOL); |
| 158 | |
| 159 | random_pool = socket(AF_UNIX, SOCK_STREAM, 0); |
| 160 | |
Damien Miller | ab18c41 | 1999-11-11 10:40:23 +1100 | [diff] [blame] | 161 | if (random_pool == -1) |
Damien Miller | 58fc473 | 1999-11-19 12:05:01 +1100 | [diff] [blame] | 162 | fatal("Couldn't create AF_UNIX socket: %s", strerror(errno)); |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 163 | |
Damien Miller | 58fc473 | 1999-11-19 12:05:01 +1100 | [diff] [blame] | 164 | if (connect(random_pool, (struct sockaddr*)&addr, addr_len) == -1) |
Damien Miller | a75cb96 | 1999-11-22 13:55:36 +1100 | [diff] [blame] | 165 | fatal("Couldn't connect to EGD socket \"%s\": %s", addr.sun_path, strerror(errno)); |
Damien Miller | 58fc473 | 1999-11-19 12:05:01 +1100 | [diff] [blame] | 166 | |
Damien Miller | ab18c41 | 1999-11-11 10:40:23 +1100 | [diff] [blame] | 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; |
Damien Miller | 037a0dc | 1999-12-07 15:38:31 +1100 | [diff] [blame] | 172 | |
| 173 | c = atomicio(write, random_pool, egd_message, sizeof(egd_message)); |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 174 | if (c == -1) |
Damien Miller | ab18c41 | 1999-11-11 10:40:23 +1100 | [diff] [blame] | 175 | fatal("Couldn't write to EGD socket \"%s\": %s", RANDOM_POOL, strerror(errno)); |
Damien Miller | 58fc473 | 1999-11-19 12:05:01 +1100 | [diff] [blame] | 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 | |
Damien Miller | ab18c41 | 1999-11-11 10:40:23 +1100 | [diff] [blame] | 183 | #endif /* HAVE_EGD */ |
| 184 | |
Damien Miller | 037a0dc | 1999-12-07 15:38:31 +1100 | [diff] [blame] | 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)); |
Damien Miller | ab18c41 | 1999-11-11 10:40:23 +1100 | [diff] [blame] | 188 | |
| 189 | close(random_pool); |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 190 | } |
Damien Miller | e413cba | 1999-10-28 14:12:54 +1000 | [diff] [blame] | 191 | #endif /* !HAVE_ARC4RANDOM */ |
Damien Miller | d4a8b7e | 1999-10-27 13:42:43 +1000 | [diff] [blame] | 192 | |
Damien Miller | e413cba | 1999-10-28 14:12:54 +1000 | [diff] [blame] | 193 | #ifndef HAVE_SETPROCTITLE |
| 194 | void setproctitle(const char *fmt, ...) |
| 195 | { |
| 196 | /* FIXME */ |
| 197 | } |
| 198 | #endif /* !HAVE_SETPROCTITLE */ |
Damien Miller | d770252 | 1999-11-22 16:11:05 +1100 | [diff] [blame] | 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 */ |
Damien Miller | e72b7af | 1999-12-30 15:08:44 +1100 | [diff] [blame] | 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) */ |