misc.c

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