util.c

   1 /*      $OpenBSD: util.c,v 1.5 2000/09/07 20:27:55 deraadt Exp $        */
   2
   3 /*
   4  * Copyright (c) 2000 Markus Friedl.  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 RCSID("$OpenBSD: util.c,v 1.5 2000/09/07 20:27:55 deraadt Exp $");
  29
  30 #include "ssh.h"
  31
  32 char *
  33 chop(char *s)
  34 {
  35         char *t = s;
  36         while (*t) {
  37                 if(*t == '\n' || *t == '\r') {
  38                         *t = '\0';
  39                         return s;
  40                 }
  41                 t++;
  42         }
  43         return s;
  44
  45 }
  46
  47 void
  48 set_nonblock(int fd)
  49 {
  50         int val;
  51         if (isatty(fd)) {
  52                 /* do not mess with tty's */
  53                 debug("no set_nonblock for tty fd %d", fd);
  54                 return;
  55         }
  56         val = fcntl(fd, F_GETFL, 0);
  57         if (val < 0) {
  58                 error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
  59                 return;
  60         }
  61         if (val & O_NONBLOCK)
  62                 return;
  63         debug("fd %d setting O_NONBLOCK", fd);
  64         val |= O_NONBLOCK;
  65         if (fcntl(fd, F_SETFL, val) == -1)
  66                 if (errno != ENODEV)
  67                         error("fcntl(%d, F_SETFL, O_NONBLOCK): %s",
  68                             fd, strerror(errno));
  69 }
  70
  71 /* Characters considered whitespace in strsep calls. */
  72 #define WHITESPACE " \t\r\n"
  73
  74 char *
  75 strdelim(char **s)
  76 {
  77         char *old;
  78         int wspace = 0;
  79
  80         if (*s == NULL)
  81                 return NULL;
  82
  83         old = *s;
  84
  85         *s = strpbrk(*s, WHITESPACE "=");
  86         if (*s == NULL)
  87                 return (old);
  88
  89         /* Allow only one '=' to be skipped */
  90         if (*s[0] == '=')
  91                 wspace = 1;
  92         *s[0] = '\0';
  93
  94         *s += strspn(*s + 1, WHITESPACE) + 1;
  95         if (*s[0] == '=' && !wspace)
  96                 *s += strspn(*s + 1, WHITESPACE) + 1;
  97
  98         return (old);
  99 }