lib/util.c

   1 /*
   2  *  Copyright (C) 2015-2016  Thadeu Lima de Souza Cascardo <cascardo@minaslivre.org>
   3  *
   4  *  This program is free software; you can redistribute it and/or modify
   5  *  it under the terms of the GNU General Public License as published by
   6  *  the Free Software Foundation; either version 3 of the License, or
   7  *  (at your option) any later version.
   8  *
   9  *  This program is distributed in the hope that it will be useful,
  10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  *  GNU General Public License for more details.
  13  *
  14  *  You should have received a copy of the GNU General Public License along
  15  *  with this program; if not, write to the Free Software Foundation, Inc.,
  16  *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 #include "util.h"
  20 #include <stdlib.h>
  21 #include <errno.h>
  22 #include <string.h>
  23
  24 #include <fcntl.h>
  25 #include <unistd.h>
  26
  27 int set_llong(char *arg, long long *val)
  28 {
  29         char *end = NULL;
  30         errno = 0;
  31         *val = strtoll(arg, &end, 0);
  32         if (end && *end)
  33                 return -EINVAL;
  34         if (errno == ERANGE)
  35                 return -ERANGE;
  36         return 0;
  37 }
  38
  39 int set_int(char *arg, int *val)
  40 {
  41         char *end = NULL;
  42         errno = 0;
  43         *val = strtol(arg, &end, 0);
  44         if (end && *end)
  45                 return -EINVAL;
  46         if (errno == ERANGE)
  47                 return -ERANGE;
  48         return 0;
  49 }
  50
  51 int set_string(char *arg, char **str)
  52 {
  53         *str = strdup(arg);
  54         if (!*str)
  55                 return -errno;
  56         return 0;
  57 }
  58
  59 int dumpfile(int fd, char *filename)
  60 {
  61         int inp;
  62         char buffer[256];
  63         int r;
  64         inp = open(filename, O_RDONLY);
  65         if (inp < 0)
  66                 return -errno;
  67         while ((r = read(inp, buffer, sizeof(buffer))) > 0) {
  68                 write(fd, buffer, r);
  69         }
  70         close(inp);
  71         return r;
  72 }