misc.c

   1 /*      $OpenBSD: misc.c,v 1.3 2001/02/28 05:34:28 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: misc.c,v 1.3 2001/02/28 05:34:28 deraadt Exp $");
  29
  30 #include "misc.h"
  31 #include "log.h"
  32 #include "xmalloc.h"
  33
  34 char *
  35 chop(char *s)
  36 {
  37         char *t = s;
  38         while (*t) {
  39                 if(*t == '\n' || *t == '\r') {
  40                         *t = '\0';
  41                         return s;
  42                 }
  43                 t++;
  44         }
  45         return s;
  46
  47 }
  48
  49 void
  50 set_nonblock(int fd)
  51 {
  52         int val;
  53         val = fcntl(fd, F_GETFL, 0);
  54         if (val < 0) {
  55                 error("fcntl(%d, F_GETFL, 0): %s", fd, strerror(errno));
  56                 return;
  57         }
  58         if (val & O_NONBLOCK) {
  59                 debug("fd %d IS O_NONBLOCK", fd);
  60                 return;
  61         }
  62         debug("fd %d setting O_NONBLOCK", fd);
  63         val |= O_NONBLOCK;
  64         if (fcntl(fd, F_SETFL, val) == -1)
  65                 if (errno != ENODEV)
  66                         error("fcntl(%d, F_SETFL, O_NONBLOCK): %s",
  67                             fd, strerror(errno));
  68 }
  69
  70 /* Characters considered whitespace in strsep calls. */
  71 #define WHITESPACE " \t\r\n"
  72
  73 char *
  74 strdelim(char **s)
  75 {
  76         char *old;
  77         int wspace = 0;
  78
  79         if (*s == NULL)
  80                 return NULL;
  81
  82         old = *s;
  83
  84         *s = strpbrk(*s, WHITESPACE "=");
  85         if (*s == NULL)
  86                 return (old);
  87
  88         /* Allow only one '=' to be skipped */
  89         if (*s[0] == '=')
  90                 wspace = 1;
  91         *s[0] = '\0';
  92
  93         *s += strspn(*s + 1, WHITESPACE) + 1;
  94         if (*s[0] == '=' && !wspace)
  95                 *s += strspn(*s + 1, WHITESPACE) + 1;
  96
  97         return (old);
  98 }
  99
 100 struct passwd *
 101 pwcopy(struct passwd *pw)
 102 {
 103         struct passwd *copy = xmalloc(sizeof(*copy));
 104         memset(copy, 0, sizeof(*copy));
 105         copy->pw_name = xstrdup(pw->pw_name);
 106         copy->pw_passwd = xstrdup(pw->pw_passwd);
 107         copy->pw_uid = pw->pw_uid;
 108         copy->pw_gid = pw->pw_gid;
 109         copy->pw_class = xstrdup(pw->pw_class);
 110         copy->pw_dir = xstrdup(pw->pw_dir);
 111         copy->pw_shell = xstrdup(pw->pw_shell);
 112         return copy;
 113 }
 114
 115 mysig_t
 116 mysignal(int sig, mysig_t act)
 117 {
 118 #ifdef HAVE_SIGACTION
 119         struct sigaction sa, osa;
 120
 121         if (sigaction(sig, NULL, &osa) == -1)
 122                 return (mysig_t) -1;
 123         if (osa.sa_handler != act) {
 124                 memset(&sa, 0, sizeof(sa));
 125                 sigemptyset(&sa.sa_mask);
 126                 sa.sa_flags = 0;
 127 #if defined(SA_RESTART)
 128                 if (sig == SIGCHLD)
 129                         sa.sa_flags |= SA_RESTART;
 130 #endif
 131 #if defined(SA_INTERRUPT)
 132                 if (sig == SIGALRM)
 133                         sa.sa_flags |= SA_INTERRUPT;
 134 #endif
 135                 sa.sa_handler = act;
 136                 if (sigaction(sig, &sa, NULL) == -1)
 137                         return (mysig_t) -1;
 138         }
 139         return (osa.sa_handler);
 140 #else
 141         return (signal(sig, act));
 142 #endif
 143 }