Home | History | Annotate | Download | only in utils
      1 /*
      2 * Copyright (c) 2015, The Linux Foundation. All rights reserved.
      3 *
      4 * Redistribution and use in source and binary forms, with or without modification, are permitted
      5 * provided that the following conditions are met:
      6 *    * Redistributions of source code must retain the above copyright notice, this list of
      7 *      conditions and the following disclaimer.
      8 *    * Redistributions in binary form must reproduce the above copyright notice, this list of
      9 *      conditions and the following disclaimer in the documentation and/or other materials provided
     10 *      with the distribution.
     11 *    * Neither the name of The Linux Foundation nor the names of its contributors may be used to
     12 *      endorse or promote products derived from this software without specific prior written
     13 *      permission.
     14 *
     15 * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     16 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
     17 * NON-INFRINGEMENT ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
     18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
     19 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
     20 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
     21 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
     22 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     23 */
     24 
     25 #ifndef __SYS_H__
     26 #define __SYS_H__
     27 
     28 #include <sys/eventfd.h>
     29 #include <dlfcn.h>
     30 #include <unistd.h>
     31 #include <stdio.h>
     32 #include <stdlib.h>
     33 #include <poll.h>
     34 #include <pthread.h>
     35 #include <fstream>
     36 
     37 #ifdef SDM_VIRTUAL_DRIVER
     38 #include <virtual_driver.h>
     39 #endif
     40 
     41 namespace sdm {
     42 
     43 class Sys {
     44  public:
     45 #ifndef SDM_VIRTUAL_DRIVER
     46   typedef std::fstream fstream;
     47 #else
     48   typedef VirtualFStream fstream;
     49 #endif
     50 
     51   // Pointers to system calls which are either mapped to actual system call or virtual driver.
     52   typedef int (*ioctl)(int, int, ...);
     53   typedef int (*open)(const char *, int, ...);
     54   typedef int (*close)(int);
     55   typedef int (*poll)(struct pollfd *, nfds_t, int);
     56   typedef ssize_t (*pread)(int, void *, size_t, off_t);
     57   typedef ssize_t (*pwrite)(int, const void *, size_t, off_t);
     58   typedef int (*pthread_cancel)(pthread_t thread);
     59   typedef int (*dup)(int fd);
     60   typedef ssize_t (*read)(int, void *, size_t);
     61   typedef ssize_t (*write)(int, const void *, size_t);
     62   typedef int (*eventfd)(unsigned int, int);
     63 
     64   static bool getline_(fstream &fs, std::string &line);  // NOLINT
     65 
     66   static ioctl ioctl_;
     67   static open open_;
     68   static close close_;
     69   static poll poll_;
     70   static pread pread_;
     71   static pwrite pwrite_;
     72   static pthread_cancel pthread_cancel_;
     73   static dup dup_;
     74   static read read_;
     75   static write write_;
     76   static eventfd eventfd_;
     77 };
     78 
     79 class DynLib {
     80  public:
     81   ~DynLib();
     82   bool Open(const char *lib_name);
     83   bool Sym(const char *func_name, void **func_ptr);
     84   const char * Error() { return ::dlerror(); }
     85   operator bool() const { return lib_ != NULL; }
     86 
     87  private:
     88   void Close();
     89 
     90   void *lib_ = NULL;
     91 };
     92 
     93 }  // namespace sdm
     94 
     95 #endif  // __SYS_H__
     96