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 (*access)(const char *, int); 54 typedef int (*open)(const char *, int, ...); 55 typedef int (*close)(int); 56 typedef int (*poll)(struct pollfd *, nfds_t, int); 57 typedef ssize_t (*pread)(int, void *, size_t, off_t); 58 typedef ssize_t (*pwrite)(int, const void *, size_t, off_t); 59 typedef int (*pthread_cancel)(pthread_t thread); 60 typedef int (*dup)(int fd); 61 typedef ssize_t (*read)(int, void *, size_t); 62 typedef ssize_t (*write)(int, const void *, size_t); 63 typedef int (*eventfd)(unsigned int, int); 64 65 static bool getline_(fstream &fs, std::string &line); // NOLINT 66 67 static ioctl ioctl_; 68 static access access_; 69 static open open_; 70 static close close_; 71 static poll poll_; 72 static pread pread_; 73 static pwrite pwrite_; 74 static pthread_cancel pthread_cancel_; 75 static dup dup_; 76 static read read_; 77 static write write_; 78 static eventfd eventfd_; 79 }; 80 81 class DynLib { 82 public: 83 ~DynLib(); 84 bool Open(const char *lib_name); 85 bool Sym(const char *func_name, void **func_ptr); Error()86 const char * Error() { return ::dlerror(); } 87 operator bool() const { return lib_ != NULL; } 88 89 private: 90 void Close(); 91 92 void *lib_ = NULL; 93 }; 94 95 } // namespace sdm 96 97 #endif // __SYS_H__ 98