1 /* $OpenBSD: misc.c,v 1.85 2011/03/29 18:54:17 stevesk Exp $ */ 2 /* 3 * Copyright (c) 2000 Markus Friedl. All rights reserved. 4 * Copyright (c) 2005,2006 Damien Miller. All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 18 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25 */ 26 27 #include "includes.h" 28 29 #include <sys/types.h> 30 #include <sys/ioctl.h> 31 #include <sys/socket.h> 32 #include <sys/param.h> 33 34 #include <stdarg.h> 35 #include <stdio.h> 36 #include <stdlib.h> 37 #include <string.h> 38 #include <time.h> 39 #include <unistd.h> 40 41 #include <netinet/in.h> 42 #include <netinet/in_systm.h> 43 #include <netinet/ip.h> 44 #include <netinet/tcp.h> 45 46 #include <errno.h> 47 #include <fcntl.h> 48 #include <netdb.h> 49 #ifdef HAVE_PATHS_H 50 # include <paths.h> 51 #include <pwd.h> 52 #endif 53 #ifdef SSH_TUN_OPENBSD 54 #include <net/if.h> 55 #endif 56 57 #include "xmalloc.h" 58 #include "misc.h" 59 #include "log.h" 60 #include "ssh.h" 61 62 /* remove newline at end of string */ 63 char * 64 chop(char *s) 65 { 66 char *t = s; 67 while (*t) { 68 if (*t == '\n' || *t == '\r') { 69 *t = '\0'; 70 return s; 71 } 72 t++; 73 } 74 return s; 75 76 } 77 78 /* set/unset filedescriptor to non-blocking */ 79 int 80 set_nonblock(int fd) 81 { 82 int val; 83 84 val = fcntl(fd, F_GETFL, 0); 85 if (val < 0) { 86 error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno)); 87 return (-1); 88 } 89 if (val & O_NONBLOCK) { 90 debug3("fd %d is O_NONBLOCK", fd); 91 return (0); 92 } 93 debug2("fd %d setting O_NONBLOCK", fd); 94 val |= O_NONBLOCK; 95 if (fcntl(fd, F_SETFL, val) == -1) { 96 debug("fcntl(%d, F_SETFL, O_NONBLOCK): %s", fd, 97 strerror(errno)); 98 return (-1); 99 } 100 return (0); 101 } 102 103 int 104 unset_nonblock(int fd) 105 { 106 int val; 107 108 val = fcntl(fd, F_GETFL, 0); 109 if (val < 0) { 110 error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno)); 111 return (-1); 112 } 113 if (!(val & O_NONBLOCK)) { 114 debug3("fd %d is not O_NONBLOCK", fd); 115 return (0); 116 } 117 debug("fd %d clearing O_NONBLOCK", fd); 118 val &= ~O_NONBLOCK; 119 if (fcntl(fd, F_SETFL, val) == -1) { 120 debug("fcntl(%d, F_SETFL, ~O_NONBLOCK): %s", 121 fd, strerror(errno)); 122 return (-1); 123 } 124 return (0); 125 } 126 127 const char * 128 ssh_gai_strerror(int gaierr) 129 { 130 if (gaierr == EAI_SYSTEM) 131 return strerror(errno); 132 return gai_strerror(gaierr); 133 } 134 135 /* disable nagle on socket */ 136 void 137 set_nodelay(int fd) 138 { 139 int opt; 140 socklen_t optlen; 141 142 optlen = sizeof opt; 143 if (getsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, &optlen) == -1) { 144 debug("getsockopt TCP_NODELAY: %.100s", strerror(errno)); 145 return; 146 } 147 if (opt == 1) { 148 debug2("fd %d is TCP_NODELAY", fd); 149 return; 150 } 151 opt = 1; 152 debug2("fd %d setting TCP_NODELAY", fd); 153 if (setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &opt, sizeof opt) == -1) 154 error("setsockopt TCP_NODELAY: %.100s", strerror(errno)); 155 } 156 157 /* Characters considered whitespace in strsep calls. */ 158 #define WHITESPACE " \t\r\n" 159 #define QUOTE "\"" 160 161 /* return next token in configuration line */ 162 char * 163 strdelim(char **s) 164 { 165 char *old; 166 int wspace = 0; 167 168 if (*s == NULL) 169 return NULL; 170 171 old = *s; 172 173 *s = strpbrk(*s, WHITESPACE QUOTE "="); 174 if (*s == NULL) 175 return (old); 176 177 if (*s[0] == '\"') { 178 memmove(*s, *s + 1, strlen(*s)); /* move nul too */ 179 /* Find matching quote */ 180 if ((*s = strpbrk(*s, QUOTE)) == NULL) { 181 return (NULL); /* no matching quote */ 182 } else { 183 *s[0] = '\0'; 184 *s += strspn(*s + 1, WHITESPACE) + 1; 185 return (old); 186 } 187 } 188 189 /* Allow only one '=' to be skipped */ 190 if (*s[0] == '=') 191 wspace = 1; 192 *s[0] = '\0'; 193 194 /* Skip any extra whitespace after first token */ 195 *s += strspn(*s + 1, WHITESPACE) + 1; 196 if (*s[0] == '=' && !wspace) 197 *s += strspn(*s + 1, WHITESPACE) + 1; 198 199 return (old); 200 } 201 202 struct passwd * 203 pwcopy(struct passwd *pw) 204 { 205 struct passwd *copy = xcalloc(1, sizeof(*copy)); 206 207 copy->pw_name = xstrdup(pw->pw_name); 208 copy->pw_passwd = pw->pw_passwd ? xstrdup(pw->pw_passwd) : NULL; 209 #ifdef HAVE_PW_GECOS_IN_PASSWD 210 copy->pw_gecos = xstrdup(pw->pw_gecos); 211 #endif 212 copy->pw_uid = pw->pw_uid; 213 copy->pw_gid = pw->pw_gid; 214 #ifdef HAVE_PW_EXPIRE_IN_PASSWD 215 copy->pw_expire = pw->pw_expire; 216 #endif 217 #ifdef HAVE_PW_CHANGE_IN_PASSWD 218 copy->pw_change = pw->pw_change; 219 #endif 220 #ifdef HAVE_PW_CLASS_IN_PASSWD 221 copy->pw_class = xstrdup(pw->pw_class); 222 #endif 223 copy->pw_dir = xstrdup(pw->pw_dir); 224 copy->pw_shell = xstrdup(pw->pw_shell); 225 return copy; 226 } 227 228 /* 229 * Convert ASCII string to TCP/IP port number. 230 * Port must be >=0 and <=65535. 231 * Return -1 if invalid. 232 */ 233 int 234 a2port(const char *s) 235 { 236 long long port; 237 const char *errstr; 238 239 port = strtonum(s, 0, 65535, &errstr); 240 if (errstr != NULL) 241 return -1; 242 return (int)port; 243 } 244 245 int 246 a2tun(const char *s, int *remote) 247 { 248 const char *errstr = NULL; 249 char *sp, *ep; 250 int tun; 251 252 if (remote != NULL) { 253 *remote = SSH_TUNID_ANY; 254 sp = xstrdup(s); 255 if ((ep = strchr(sp, ':')) == NULL) { 256 xfree(sp); 257 return (a2tun(s, NULL)); 258 } 259 ep[0] = '\0'; ep++; 260 *remote = a2tun(ep, NULL); 261 tun = a2tun(sp, NULL); 262 xfree(sp); 263 return (*remote == SSH_TUNID_ERR ? *remote : tun); 264 } 265 266 if (strcasecmp(s, "any") == 0) 267 return (SSH_TUNID_ANY); 268 269 tun = strtonum(s, 0, SSH_TUNID_MAX, &errstr); 270 if (errstr != NULL) 271 return (SSH_TUNID_ERR); 272 273 return (tun); 274 } 275 276 #define SECONDS 1 277 #define MINUTES (SECONDS * 60) 278 #define HOURS (MINUTES * 60) 279 #define DAYS (HOURS * 24) 280 #define WEEKS (DAYS * 7) 281 282 /* 283 * Convert a time string into seconds; format is 284 * a sequence of: 285 * time[qualifier] 286 * 287 * Valid time qualifiers are: 288 * <none> seconds 289 * s|S seconds 290 * m|M minutes 291 * h|H hours 292 * d|D days 293 * w|W weeks 294 * 295 * Examples: 296 * 90m 90 minutes 297 * 1h30m 90 minutes 298 * 2d 2 days 299 * 1w 1 week 300 * 301 * Return -1 if time string is invalid. 302 */ 303 long 304 convtime(const char *s) 305 { 306 long total, secs; 307 const char *p; 308 char *endp; 309 310 errno = 0; 311 total = 0; 312 p = s; 313 314 if (p == NULL || *p == '\0') 315 return -1; 316 317 while (*p) { 318 secs = strtol(p, &endp, 10); 319 if (p == endp || 320 (errno == ERANGE && (secs == LONG_MIN || secs == LONG_MAX)) || 321 secs < 0) 322 return -1; 323 324 switch (*endp++) { 325 case '\0': 326 endp--; 327 break; 328 case 's': 329 case 'S': 330 break; 331 case 'm': 332 case 'M': 333 secs *= MINUTES; 334 break; 335 case 'h': 336 case 'H': 337 secs *= HOURS; 338 break; 339 case 'd': 340 case 'D': 341 secs *= DAYS; 342 break; 343 case 'w': 344 case 'W': 345 secs *= WEEKS; 346 break; 347 default: 348 return -1; 349 } 350 total += secs; 351 if (total < 0) 352 return -1; 353 p = endp; 354 } 355 356 return total; 357 } 358 359 /* 360 * Returns a standardized host+port identifier string. 361 * Caller must free returned string. 362 */ 363 char * 364 put_host_port(const char *host, u_short port) 365 { 366 char *hoststr; 367 368 if (port == 0 || port == SSH_DEFAULT_PORT) 369 return(xstrdup(host)); 370 if (asprintf(&hoststr, "[%s]:%d", host, (int)port) < 0) 371 fatal("put_host_port: asprintf: %s", strerror(errno)); 372 debug3("put_host_port: %s", hoststr); 373 return hoststr; 374 } 375 376 /* 377 * Search for next delimiter between hostnames/addresses and ports. 378 * Argument may be modified (for termination). 379 * Returns *cp if parsing succeeds. 380 * *cp is set to the start of the next delimiter, if one was found. 381 * If this is the last field, *cp is set to NULL. 382 */ 383 char * 384 hpdelim(char **cp) 385 { 386 char *s, *old; 387 388 if (cp == NULL || *cp == NULL) 389 return NULL; 390 391 old = s = *cp; 392 if (*s == '[') { 393 if ((s = strchr(s, ']')) == NULL) 394 return NULL; 395 else 396 s++; 397 } else if ((s = strpbrk(s, ":/")) == NULL) 398 s = *cp + strlen(*cp); /* skip to end (see first case below) */ 399 400 switch (*s) { 401 case '\0': 402 *cp = NULL; /* no more fields*/ 403 break; 404 405 case ':': 406 case '/': 407 *s = '\0'; /* terminate */ 408 *cp = s + 1; 409 break; 410 411 default: 412 return NULL; 413 } 414 415 return old; 416 } 417 418 char * 419 cleanhostname(char *host) 420 { 421 if (*host == '[' && host[strlen(host) - 1] == ']') { 422 host[strlen(host) - 1] = '\0'; 423 return (host + 1); 424 } else 425 return host; 426 } 427 428 char * 429 colon(char *cp) 430 { 431 int flag = 0; 432 433 if (*cp == ':') /* Leading colon is part of file name. */ 434 return NULL; 435 if (*cp == '[') 436 flag = 1; 437 438 for (; *cp; ++cp) { 439 if (*cp == '@' && *(cp+1) == '[') 440 flag = 1; 441 if (*cp == ']' && *(cp+1) == ':' && flag) 442 return (cp+1); 443 if (*cp == ':' && !flag) 444 return (cp); 445 if (*cp == '/') 446 return NULL; 447 } 448 return NULL; 449 } 450 451 /* function to assist building execv() arguments */ 452 void 453 addargs(arglist *args, char *fmt, ...) 454 { 455 va_list ap; 456 char *cp; 457 u_int nalloc; 458 int r; 459 460 va_start(ap, fmt); 461 r = vasprintf(&cp, fmt, ap); 462 va_end(ap); 463 if (r == -1) 464 fatal("addargs: argument too long"); 465 466 nalloc = args->nalloc; 467 if (args->list == NULL) { 468 nalloc = 32; 469 args->num = 0; 470 } else if (args->num+2 >= nalloc) 471 nalloc *= 2; 472 473 args->list = xrealloc(args->list, nalloc, sizeof(char *)); 474 args->nalloc = nalloc; 475 args->list[args->num++] = cp; 476 args->list[args->num] = NULL; 477 } 478 479 void 480 replacearg(arglist *args, u_int which, char *fmt, ...) 481 { 482 va_list ap; 483 char *cp; 484 int r; 485 486 va_start(ap, fmt); 487 r = vasprintf(&cp, fmt, ap); 488 va_end(ap); 489 if (r == -1) 490 fatal("replacearg: argument too long"); 491 492 if (which >= args->num) 493 fatal("replacearg: tried to replace invalid arg %d >= %d", 494 which, args->num); 495 xfree(args->list[which]); 496 args->list[which] = cp; 497 } 498 499 void 500 freeargs(arglist *args) 501 { 502 u_int i; 503 504 if (args->list != NULL) { 505 for (i = 0; i < args->num; i++) 506 xfree(args->list[i]); 507 xfree(args->list); 508 args->nalloc = args->num = 0; 509 args->list = NULL; 510 } 511 } 512 513 /* 514 * Expands tildes in the file name. Returns data allocated by xmalloc. 515 * Warning: this calls getpw*. 516 */ 517 char * 518 tilde_expand_filename(const char *filename, uid_t uid) 519 { 520 const char *path; 521 char user[128], ret[MAXPATHLEN]; 522 struct passwd *pw; 523 u_int len, slash; 524 525 if (*filename != '~') 526 return (xstrdup(filename)); 527 filename++; 528 529 path = strchr(filename, '/'); 530 if (path != NULL && path > filename) { /* ~user/path */ 531 slash = path - filename; 532 if (slash > sizeof(user) - 1) 533 fatal("tilde_expand_filename: ~username too long"); 534 memcpy(user, filename, slash); 535 user[slash] = '\0'; 536 if ((pw = getpwnam(user)) == NULL) 537 fatal("tilde_expand_filename: No such user %s", user); 538 } else if ((pw = getpwuid(uid)) == NULL) /* ~/path */ 539 fatal("tilde_expand_filename: No such uid %ld", (long)uid); 540 541 if (strlcpy(ret, pw->pw_dir, sizeof(ret)) >= sizeof(ret)) 542 fatal("tilde_expand_filename: Path too long"); 543 544 /* Make sure directory has a trailing '/' */ 545 len = strlen(pw->pw_dir); 546 if ((len == 0 || pw->pw_dir[len - 1] != '/') && 547 strlcat(ret, "/", sizeof(ret)) >= sizeof(ret)) 548 fatal("tilde_expand_filename: Path too long"); 549 550 /* Skip leading '/' from specified path */ 551 if (path != NULL) 552 filename = path + 1; 553 if (strlcat(ret, filename, sizeof(ret)) >= sizeof(ret)) 554 fatal("tilde_expand_filename: Path too long"); 555 556 return (xstrdup(ret)); 557 } 558 559 /* 560 * Expand a string with a set of %[char] escapes. A number of escapes may be 561 * specified as (char *escape_chars, char *replacement) pairs. The list must 562 * be terminated by a NULL escape_char. Returns replaced string in memory 563 * allocated by xmalloc. 564 */ 565 char * 566 percent_expand(const char *string, ...) 567 { 568 #define EXPAND_MAX_KEYS 16 569 u_int num_keys, i, j; 570 struct { 571 const char *key; 572 const char *repl; 573 } keys[EXPAND_MAX_KEYS]; 574 char buf[4096]; 575 va_list ap; 576 577 /* Gather keys */ 578 va_start(ap, string); 579 for (num_keys = 0; num_keys < EXPAND_MAX_KEYS; num_keys++) { 580 keys[num_keys].key = va_arg(ap, char *); 581 if (keys[num_keys].key == NULL) 582 break; 583 keys[num_keys].repl = va_arg(ap, char *); 584 if (keys[num_keys].repl == NULL) 585 fatal("%s: NULL replacement", __func__); 586 } 587 if (num_keys == EXPAND_MAX_KEYS && va_arg(ap, char *) != NULL) 588 fatal("%s: too many keys", __func__); 589 va_end(ap); 590 591 /* Expand string */ 592 *buf = '\0'; 593 for (i = 0; *string != '\0'; string++) { 594 if (*string != '%') { 595 append: 596 buf[i++] = *string; 597 if (i >= sizeof(buf)) 598 fatal("%s: string too long", __func__); 599 buf[i] = '\0'; 600 continue; 601 } 602 string++; 603 /* %% case */ 604 if (*string == '%') 605 goto append; 606 for (j = 0; j < num_keys; j++) { 607 if (strchr(keys[j].key, *string) != NULL) { 608 i = strlcat(buf, keys[j].repl, sizeof(buf)); 609 if (i >= sizeof(buf)) 610 fatal("%s: string too long", __func__); 611 break; 612 } 613 } 614 if (j >= num_keys) 615 fatal("%s: unknown key %%%c", __func__, *string); 616 } 617 return (xstrdup(buf)); 618 #undef EXPAND_MAX_KEYS 619 } 620 621 /* 622 * Read an entire line from a public key file into a static buffer, discarding 623 * lines that exceed the buffer size. Returns 0 on success, -1 on failure. 624 */ 625 int 626 read_keyfile_line(FILE *f, const char *filename, char *buf, size_t bufsz, 627 u_long *lineno) 628 { 629 while (fgets(buf, bufsz, f) != NULL) { 630 if (buf[0] == '\0') 631 continue; 632 (*lineno)++; 633 if (buf[strlen(buf) - 1] == '\n' || feof(f)) { 634 return 0; 635 } else { 636 debug("%s: %s line %lu exceeds size limit", __func__, 637 filename, *lineno); 638 /* discard remainder of line */ 639 while (fgetc(f) != '\n' && !feof(f)) 640 ; /* nothing */ 641 } 642 } 643 return -1; 644 } 645 646 int 647 tun_open(int tun, int mode) 648 { 649 #if defined(CUSTOM_SYS_TUN_OPEN) 650 return (sys_tun_open(tun, mode)); 651 #elif defined(SSH_TUN_OPENBSD) 652 struct ifreq ifr; 653 char name[100]; 654 int fd = -1, sock; 655 656 /* Open the tunnel device */ 657 if (tun <= SSH_TUNID_MAX) { 658 snprintf(name, sizeof(name), "/dev/tun%d", tun); 659 fd = open(name, O_RDWR); 660 } else if (tun == SSH_TUNID_ANY) { 661 for (tun = 100; tun >= 0; tun--) { 662 snprintf(name, sizeof(name), "/dev/tun%d", tun); 663 if ((fd = open(name, O_RDWR)) >= 0) 664 break; 665 } 666 } else { 667 debug("%s: invalid tunnel %u", __func__, tun); 668 return (-1); 669 } 670 671 if (fd < 0) { 672 debug("%s: %s open failed: %s", __func__, name, strerror(errno)); 673 return (-1); 674 } 675 676 debug("%s: %s mode %d fd %d", __func__, name, mode, fd); 677 678 /* Set the tunnel device operation mode */ 679 snprintf(ifr.ifr_name, sizeof(ifr.ifr_name), "tun%d", tun); 680 if ((sock = socket(PF_UNIX, SOCK_STREAM, 0)) == -1) 681 goto failed; 682 683 if (ioctl(sock, SIOCGIFFLAGS, &ifr) == -1) 684 goto failed; 685 686 /* Set interface mode */ 687 ifr.ifr_flags &= ~IFF_UP; 688 if (mode == SSH_TUNMODE_ETHERNET) 689 ifr.ifr_flags |= IFF_LINK0; 690 else 691 ifr.ifr_flags &= ~IFF_LINK0; 692 if (ioctl(sock, SIOCSIFFLAGS, &ifr) == -1) 693 goto failed; 694 695 /* Bring interface up */ 696 ifr.ifr_flags |= IFF_UP; 697 if (ioctl(sock, SIOCSIFFLAGS, &ifr) == -1) 698 goto failed; 699 700 close(sock); 701 return (fd); 702 703 failed: 704 if (fd >= 0) 705 close(fd); 706 if (sock >= 0) 707 close(sock); 708 debug("%s: failed to set %s mode %d: %s", __func__, name, 709 mode, strerror(errno)); 710 return (-1); 711 #else 712 error("Tunnel interfaces are not supported on this platform"); 713 return (-1); 714 #endif 715 } 716 717 void 718 sanitise_stdfd(void) 719 { 720 int nullfd, dupfd; 721 722 if ((nullfd = dupfd = open(_PATH_DEVNULL, O_RDWR)) == -1) { 723 fprintf(stderr, "Couldn't open /dev/null: %s\n", 724 strerror(errno)); 725 exit(1); 726 } 727 while (++dupfd <= 2) { 728 /* Only clobber closed fds */ 729 if (fcntl(dupfd, F_GETFL, 0) >= 0) 730 continue; 731 if (dup2(nullfd, dupfd) == -1) { 732 fprintf(stderr, "dup2: %s\n", strerror(errno)); 733 exit(1); 734 } 735 } 736 if (nullfd > 2) 737 close(nullfd); 738 } 739 740 char * 741 tohex(const void *vp, size_t l) 742 { 743 const u_char *p = (const u_char *)vp; 744 char b[3], *r; 745 size_t i, hl; 746 747 if (l > 65536) 748 return xstrdup("tohex: length > 65536"); 749 750 hl = l * 2 + 1; 751 r = xcalloc(1, hl); 752 for (i = 0; i < l; i++) { 753 snprintf(b, sizeof(b), "%02x", p[i]); 754 strlcat(r, b, hl); 755 } 756 return (r); 757 } 758 759 u_int64_t 760 get_u64(const void *vp) 761 { 762 const u_char *p = (const u_char *)vp; 763 u_int64_t v; 764 765 v = (u_int64_t)p[0] << 56; 766 v |= (u_int64_t)p[1] << 48; 767 v |= (u_int64_t)p[2] << 40; 768 v |= (u_int64_t)p[3] << 32; 769 v |= (u_int64_t)p[4] << 24; 770 v |= (u_int64_t)p[5] << 16; 771 v |= (u_int64_t)p[6] << 8; 772 v |= (u_int64_t)p[7]; 773 774 return (v); 775 } 776 777 u_int32_t 778 get_u32(const void *vp) 779 { 780 const u_char *p = (const u_char *)vp; 781 u_int32_t v; 782 783 v = (u_int32_t)p[0] << 24; 784 v |= (u_int32_t)p[1] << 16; 785 v |= (u_int32_t)p[2] << 8; 786 v |= (u_int32_t)p[3]; 787 788 return (v); 789 } 790 791 u_int16_t 792 get_u16(const void *vp) 793 { 794 const u_char *p = (const u_char *)vp; 795 u_int16_t v; 796 797 v = (u_int16_t)p[0] << 8; 798 v |= (u_int16_t)p[1]; 799 800 return (v); 801 } 802 803 void 804 put_u64(void *vp, u_int64_t v) 805 { 806 u_char *p = (u_char *)vp; 807 808 p[0] = (u_char)(v >> 56) & 0xff; 809 p[1] = (u_char)(v >> 48) & 0xff; 810 p[2] = (u_char)(v >> 40) & 0xff; 811 p[3] = (u_char)(v >> 32) & 0xff; 812 p[4] = (u_char)(v >> 24) & 0xff; 813 p[5] = (u_char)(v >> 16) & 0xff; 814 p[6] = (u_char)(v >> 8) & 0xff; 815 p[7] = (u_char)v & 0xff; 816 } 817 818 void 819 put_u32(void *vp, u_int32_t v) 820 { 821 u_char *p = (u_char *)vp; 822 823 p[0] = (u_char)(v >> 24) & 0xff; 824 p[1] = (u_char)(v >> 16) & 0xff; 825 p[2] = (u_char)(v >> 8) & 0xff; 826 p[3] = (u_char)v & 0xff; 827 } 828 829 830 void 831 put_u16(void *vp, u_int16_t v) 832 { 833 u_char *p = (u_char *)vp; 834 835 p[0] = (u_char)(v >> 8) & 0xff; 836 p[1] = (u_char)v & 0xff; 837 } 838 839 void 840 ms_subtract_diff(struct timeval *start, int *ms) 841 { 842 struct timeval diff, finish; 843 844 gettimeofday(&finish, NULL); 845 timersub(&finish, start, &diff); 846 *ms -= (diff.tv_sec * 1000) + (diff.tv_usec / 1000); 847 } 848 849 void 850 ms_to_timeval(struct timeval *tv, int ms) 851 { 852 if (ms < 0) 853 ms = 0; 854 tv->tv_sec = ms / 1000; 855 tv->tv_usec = (ms % 1000) * 1000; 856 } 857 858 void 859 bandwidth_limit_init(struct bwlimit *bw, u_int64_t kbps, size_t buflen) 860 { 861 bw->buflen = buflen; 862 bw->rate = kbps; 863 bw->thresh = bw->rate; 864 bw->lamt = 0; 865 timerclear(&bw->bwstart); 866 timerclear(&bw->bwend); 867 } 868 869 /* Callback from read/write loop to insert bandwidth-limiting delays */ 870 void 871 bandwidth_limit(struct bwlimit *bw, size_t read_len) 872 { 873 u_int64_t waitlen; 874 struct timespec ts, rm; 875 876 if (!timerisset(&bw->bwstart)) { 877 gettimeofday(&bw->bwstart, NULL); 878 return; 879 } 880 881 bw->lamt += read_len; 882 if (bw->lamt < bw->thresh) 883 return; 884 885 gettimeofday(&bw->bwend, NULL); 886 timersub(&bw->bwend, &bw->bwstart, &bw->bwend); 887 if (!timerisset(&bw->bwend)) 888 return; 889 890 bw->lamt *= 8; 891 waitlen = (double)1000000L * bw->lamt / bw->rate; 892 893 bw->bwstart.tv_sec = waitlen / 1000000L; 894 bw->bwstart.tv_usec = waitlen % 1000000L; 895 896 if (timercmp(&bw->bwstart, &bw->bwend, >)) { 897 timersub(&bw->bwstart, &bw->bwend, &bw->bwend); 898 899 /* Adjust the wait time */ 900 if (bw->bwend.tv_sec) { 901 bw->thresh /= 2; 902 if (bw->thresh < bw->buflen / 4) 903 bw->thresh = bw->buflen / 4; 904 } else if (bw->bwend.tv_usec < 10000) { 905 bw->thresh *= 2; 906 if (bw->thresh > bw->buflen * 8) 907 bw->thresh = bw->buflen * 8; 908 } 909 910 TIMEVAL_TO_TIMESPEC(&bw->bwend, &ts); 911 while (nanosleep(&ts, &rm) == -1) { 912 if (errno != EINTR) 913 break; 914 ts = rm; 915 } 916 } 917 918 bw->lamt = 0; 919 gettimeofday(&bw->bwstart, NULL); 920 } 921 922 /* Make a template filename for mk[sd]temp() */ 923 void 924 mktemp_proto(char *s, size_t len) 925 { 926 const char *tmpdir; 927 int r; 928 929 if ((tmpdir = getenv("TMPDIR")) != NULL) { 930 r = snprintf(s, len, "%s/ssh-XXXXXXXXXXXX", tmpdir); 931 if (r > 0 && (size_t)r < len) 932 return; 933 } 934 r = snprintf(s, len, "/tmp/ssh-XXXXXXXXXXXX"); 935 if (r < 0 || (size_t)r >= len) 936 fatal("%s: template string too short", __func__); 937 } 938 939 static const struct { 940 const char *name; 941 int value; 942 } ipqos[] = { 943 { "af11", IPTOS_DSCP_AF11 }, 944 { "af12", IPTOS_DSCP_AF12 }, 945 { "af13", IPTOS_DSCP_AF13 }, 946 { "af14", IPTOS_DSCP_AF21 }, 947 { "af22", IPTOS_DSCP_AF22 }, 948 { "af23", IPTOS_DSCP_AF23 }, 949 { "af31", IPTOS_DSCP_AF31 }, 950 { "af32", IPTOS_DSCP_AF32 }, 951 { "af33", IPTOS_DSCP_AF33 }, 952 { "af41", IPTOS_DSCP_AF41 }, 953 { "af42", IPTOS_DSCP_AF42 }, 954 { "af43", IPTOS_DSCP_AF43 }, 955 { "cs0", IPTOS_DSCP_CS0 }, 956 { "cs1", IPTOS_DSCP_CS1 }, 957 { "cs2", IPTOS_DSCP_CS2 }, 958 { "cs3", IPTOS_DSCP_CS3 }, 959 { "cs4", IPTOS_DSCP_CS4 }, 960 { "cs5", IPTOS_DSCP_CS5 }, 961 { "cs6", IPTOS_DSCP_CS6 }, 962 { "cs7", IPTOS_DSCP_CS7 }, 963 { "ef", IPTOS_DSCP_EF }, 964 { "lowdelay", IPTOS_LOWDELAY }, 965 { "throughput", IPTOS_THROUGHPUT }, 966 { "reliability", IPTOS_RELIABILITY }, 967 { NULL, -1 } 968 }; 969 970 int 971 parse_ipqos(const char *cp) 972 { 973 u_int i; 974 char *ep; 975 long val; 976 977 if (cp == NULL) 978 return -1; 979 for (i = 0; ipqos[i].name != NULL; i++) { 980 if (strcasecmp(cp, ipqos[i].name) == 0) 981 return ipqos[i].value; 982 } 983 /* Try parsing as an integer */ 984 val = strtol(cp, &ep, 0); 985 if (*cp == '\0' || *ep != '\0' || val < 0 || val > 255) 986 return -1; 987 return val; 988 } 989 990 const char * 991 iptos2str(int iptos) 992 { 993 int i; 994 static char iptos_str[sizeof "0xff"]; 995 996 for (i = 0; ipqos[i].name != NULL; i++) { 997 if (ipqos[i].value == iptos) 998 return ipqos[i].name; 999 } 1000 snprintf(iptos_str, sizeof iptos_str, "0x%02x", iptos); 1001 return iptos_str; 1002 } 1003 void 1004 sock_set_v6only(int s) 1005 { 1006 #ifdef IPV6_V6ONLY 1007 int on = 1; 1008 1009 debug3("%s: set socket %d IPV6_V6ONLY", __func__, s); 1010 if (setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY, &on, sizeof(on)) == -1) 1011 error("setsockopt IPV6_V6ONLY: %s", strerror(errno)); 1012 #endif 1013 } 1014
