vlib/vlib/unix/util.c

   1 /*
   2  * Copyright (c) 2016 Cisco and/or its affiliates.
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15 /*
  16  * pci.c: Linux user space PCI bus management.
  17  *
  18  * Copyright (c) 2008 Eliot Dresselhaus
  19  *
  20  * Permission is hereby granted, free of charge, to any person obtaining
  21  * a copy of this software and associated documentation files (the
  22  * "Software"), to deal in the Software without restriction, including
  23  * without limitation the rights to use, copy, modify, merge, publish,
  24  * distribute, sublicense, and/or sell copies of the Software, and to
  25  * permit persons to whom the Software is furnished to do so, subject to
  26  * the following conditions:
  27  *
  28  * The above copyright notice and this permission notice shall be
  29  * included in all copies or substantial portions of the Software.
  30  *
  31  *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  32  *  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  33  *  MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  34  *  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  35  *  LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  36  *  OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  37  *  WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  38  */
  39
  40 #include <vlib/vlib.h>
  41 #include <vlib/unix/unix.h>
  42
  43 #include <sys/types.h>
  44 #include <sys/stat.h>
  45 #include <fcntl.h>
  46 #include <dirent.h>
  47
  48 clib_error_t *
  49 foreach_directory_file (char *dir_name,
  50                         clib_error_t * (*f) (void *arg, u8 * path_name,
  51                                              u8 * file_name), void *arg,
  52                         int scan_dirs)
  53 {
  54   DIR *d;
  55   struct dirent *e;
  56   clib_error_t *error = 0;
  57   u8 *s, *t;
  58
  59   d = opendir (dir_name);
  60   if (!d)
  61     {
  62       if (errno == ENOENT)
  63         return 0;
  64       return clib_error_return_unix (0, "open `%s'", dir_name);
  65     }
  66
  67   s = t = 0;
  68   while (1)
  69     {
  70       e = readdir (d);
  71       if (!e)
  72         break;
  73       if (scan_dirs)
  74         {
  75           if (e->d_type == DT_DIR
  76               && (!strcmp (e->d_name, ".") || !strcmp (e->d_name, "..")))
  77             continue;
  78         }
  79       else
  80         {
  81           if (e->d_type == DT_DIR)
  82             continue;
  83         }
  84
  85       s = format (s, "%s/%s", dir_name, e->d_name);
  86       t = format (t, "%s", e->d_name);
  87       error = f (arg, s, t);
  88       _vec_len (s) = 0;
  89       _vec_len (t) = 0;
  90
  91       if (error)
  92         break;
  93     }
  94
  95   vec_free (s);
  96   closedir (d);
  97
  98   return error;
  99 }
 100
 101 clib_error_t *
 102 vlib_sysfs_write (char *file_name, char *fmt, ...)
 103 {
 104   u8 *s;
 105   int fd;
 106
 107   fd = open (file_name, O_WRONLY);
 108   if (fd < 0)
 109     return clib_error_return_unix (0, "open `%s'", file_name);
 110
 111   va_list va;
 112   va_start (va, fmt);
 113   s = va_format (0, fmt, &va);
 114   va_end (va);
 115
 116   if (write (fd, s, vec_len (s)) < 0)
 117     return clib_error_return_unix (0, "write `%s'", file_name);
 118
 119   vec_free (s);
 120   close (fd);
 121   return 0;
 122 }
 123
 124 clib_error_t *
 125 vlib_sysfs_read (char *file_name, char *fmt, ...)
 126 {
 127   unformat_input_t input;
 128   u8 *s = 0;
 129   int fd;
 130   ssize_t sz;
 131   uword result;
 132
 133   fd = open (file_name, O_RDONLY);
 134   if (fd < 0)
 135     return clib_error_return_unix (0, "open `%s'", file_name);
 136
 137   vec_validate (s, 4095);
 138
 139   sz = read (fd, s, vec_len (s));
 140   if (sz < 0)
 141     {
 142       close (fd);
 143       vec_free (s);
 144       return clib_error_return_unix (0, "read `%s'", file_name);
 145     }
 146
 147   _vec_len (s) = sz;
 148   unformat_init_vector (&input, s);
 149
 150   va_list va;
 151   va_start (va, fmt);
 152   result = va_unformat (&input, fmt, &va);
 153   va_end (va);
 154
 155   vec_free (s);
 156   close (fd);
 157
 158   if (result == 0)
 159     return clib_error_return (0, "unformat error");
 160
 161   return 0;
 162 }
 163
 164 u8 *
 165 vlib_sysfs_link_to_name (char *link)
 166 {
 167   char *p, buffer[64];
 168   unformat_input_t in;
 169   u8 *s = 0;
 170   int r;
 171
 172   r = readlink (link, buffer, sizeof (buffer) - 1);
 173
 174   if (r < 0)
 175     return 0;
 176
 177   buffer[r] = 0;
 178   p = strrchr (buffer, '/');
 179
 180   if (!p)
 181     return 0;
 182
 183   unformat_init_string (&in, p + 1, strlen (p + 1));
 184   unformat (&in, "%s", &s);
 185   unformat_free (&in);
 186
 187   return s;
 188 }
 189
 190 /*
 191  * fd.io coding-style-patch-verification: ON
 192  *
 193  * Local Variables:
 194  * eval: (c-set-style "gnu")
 195  * End:
 196  */