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 *str, long long *val)
  28 {
  29         char *end = NULL;
  30         errno = 0;
  31         *val = strtoll(str, &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 **args, int argc, int *val)
  40 {
  41         char *end = NULL;
  42         if (argc != 2)
  43                 return -EINVAL;
  44         errno = 0;
  45         *val = strtol(args[1], &end, 0);
  46         if (end && *end)
  47                 return -EINVAL;
  48         if (errno == ERANGE)
  49                 return -ERANGE;
  50         return 0;
  51 }
  52
  53 int set_string(char **args, int argc, char **str)
  54 {
  55         if (argc != 2)
  56                 return -EINVAL;
  57         *str = strdup(args[1]);
  58         if (!*str)
  59                 return -errno;
  60         return 0;
  61 }
  62
  63 int dumpfile(int fd, char *filename)
  64 {
  65         int inp;
  66         char buffer[256];
  67         int r;
  68         inp = open(filename, O_RDONLY);
  69         if (inp < 0)
  70                 return -errno;
  71         while ((r = read(inp, buffer, sizeof(buffer))) > 0) {
  72                 write(fd, buffer, r);
  73         }
  74         close(inp);
  75         return r;
  76 }