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   clib_error_t *error = 0;
 107
 108   fd = open (file_name, O_WRONLY);
 109   if (fd < 0)
 110     return clib_error_return_unix (0, "open `%s'", file_name);
 111
 112   va_list va;
 113   va_start (va, fmt);
 114   s = va_format (0, fmt, &va);
 115   va_end (va);
 116
 117   if (write (fd, s, vec_len (s)) < 0)
 118     error = clib_error_return_unix (0, "write `%s'", file_name);
 119
 120   vec_free (s);
 121   close (fd);
 122   return error;
 123 }
 124
 125 clib_error_t *
 126 vlib_sysfs_read (char *file_name, char *fmt, ...)
 127 {
 128   unformat_input_t input;
 129   u8 *s = 0;
 130   int fd;
 131   ssize_t sz;
 132   uword result;
 133
 134   fd = open (file_name, O_RDONLY);
 135   if (fd < 0)
 136     return clib_error_return_unix (0, "open `%s'", file_name);
 137
 138   vec_validate (s, 4095);
 139
 140   sz = read (fd, s, vec_len (s));
 141   if (sz < 0)
 142     {
 143       close (fd);
 144       vec_free (s);
 145       return clib_error_return_unix (0, "read `%s'", file_name);
 146     }
 147
 148   _vec_len (s) = sz;
 149   unformat_init_vector (&input, s);
 150
 151   va_list va;
 152   va_start (va, fmt);
 153   result = va_unformat (&input, fmt, &va);
 154   va_end (va);
 155
 156   vec_free (s);
 157   close (fd);
 158
 159   if (result == 0)
 160     return clib_error_return (0, "unformat error");
 161
 162   return 0;
 163 }
 164
 165 u8 *
 166 vlib_sysfs_link_to_name (char *link)
 167 {
 168   char *p, buffer[64];
 169   unformat_input_t in;
 170   u8 *s = 0;
 171   int r;
 172
 173   r = readlink (link, buffer, sizeof (buffer) - 1);
 174
 175   if (r < 0)
 176     return 0;
 177
 178   buffer[r] = 0;
 179   p = strrchr (buffer, '/');
 180
 181   if (!p)
 182     return 0;
 183
 184   unformat_init_string (&in, p + 1, strlen (p + 1));
 185   if (unformat (&in, "%s", &s) != 1)
 186     clib_unix_warning ("no string?");
 187   unformat_free (&in);
 188
 189   return s;
 190 }
 191
 192 int
 193 vlib_sysfs_get_free_hugepages (unsigned int numa_node, int page_size)
 194 {
 195   struct stat sb;
 196   u8 *p = 0;
 197   int r = -1;
 198
 199   p = format (p, "/sys/devices/system/node/node%u%c", numa_node, 0);
 200
 201   if (stat ((char *) p, &sb) == 0)
 202     {
 203       if (S_ISDIR (sb.st_mode) == 0)
 204         goto done;
 205     }
 206   else if (numa_node == 0)
 207     {
 208       vec_reset_length (p);
 209       p = format (p, "/sys/kernel/mm%c", 0);
 210       if (stat ((char *) p, &sb) < 0 || S_ISDIR (sb.st_mode) == 0)
 211         goto done;
 212     }
 213   else
 214     goto done;
 215
 216   _vec_len (p) -= 1;
 217   p = format (p, "/hugepages/hugepages-%ukB/free_hugepages%c", page_size, 0);
 218   vlib_sysfs_read ((char *) p, "%d", &r);
 219
 220 done:
 221   vec_free (p);
 222   return r;
 223 }
 224
 225 /*
 226  * fd.io coding-style-patch-verification: ON
 227  *
 228  * Local Variables:
 229  * eval: (c-set-style "gnu")
 230  * End:
 231  */