sdm/include/utils/sys.h

/*
* Copyright (c) 2015-2020, The Linux Foundation. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without modification, are permitted
* provided that the following conditions are met:
*    * Redistributions of source code must retain the above copyright notice, this list of
*      conditions and the following disclaimer.
*    * Redistributions in binary form must reproduce the above copyright notice, this list of
*      conditions and the following disclaimer in the documentation and/or other materials provided
*      with the distribution.
*    * Neither the name of The Linux Foundation nor the names of its contributors may be used to
*      endorse or promote products derived from this software without specific prior written
*      permission.
*
* THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NON-INFRINGEMENT ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
* OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
* STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/

#ifndef __SYS_H__
#define __SYS_H__

#include <sys/eventfd.h>
#include <dlfcn.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <poll.h>
#include <pthread.h>
#include <fstream>

#ifdef SDM_VIRTUAL_DRIVER
#include <virtual_driver.h>
#endif

namespace sdm {

class Sys {
 public:
#ifndef SDM_VIRTUAL_DRIVER
  typedef std::fstream fstream;
#else
  typedef VirtualFStream fstream;
#endif

  // Pointers to system calls which are either mapped to actual system call or virtual driver.
#ifdef TRUSTED_VM
  typedef int (*ioctl)(int, unsigned long int, ...);  // NOLINT
#else
  typedef int (*ioctl)(int, int, ...);
#endif
  typedef int (*access)(const char *, int);
  typedef int (*open)(const char *, int, ...);
  typedef int (*close)(int);
  typedef int (*poll)(struct pollfd *, nfds_t, int);
  typedef ssize_t (*pread)(int, void *, size_t, off_t);
  typedef ssize_t (*pwrite)(int, const void *, size_t, off_t);
  typedef int (*pthread_cancel)(pthread_t thread);
  typedef int (*dup)(int fd);
  typedef ssize_t (*read)(int, void *, size_t);
  typedef ssize_t (*write)(int, const void *, size_t);
  typedef int (*eventfd)(unsigned int, int);
  typedef int (*inotify_init)(void);
  typedef int (*inotify_add_watch)(int, const char *, uint32_t);
#ifdef TRUSTED_VM
  typedef int (*inotify_rm_watch)(int, int);
#else
  typedef int (*inotify_rm_watch)(int, uint32_t);
#endif

  static bool getline_(fstream &fs, std::string &line);  // NOLINT

  static ioctl ioctl_;
  static access access_;
  static open open_;
  static close close_;
  static poll poll_;
  static pread pread_;
  static pwrite pwrite_;
  static pthread_cancel pthread_cancel_;
  static dup dup_;
  static read read_;
  static write write_;
  static eventfd eventfd_;
  static inotify_init inotify_init_;
  static inotify_add_watch inotify_add_watch_;
  static inotify_rm_watch inotify_rm_watch_;
};

class DynLib {
 public:
  ~DynLib();
  bool Open(const char *lib_name);
  bool Sym(const char *func_name, void **func_ptr);
  const char * Error() { return ::dlerror(); }
  operator bool() const { return lib_ != NULL; }

 private:
  void Close();

  void *lib_ = NULL;
};

}  // namespace sdm

#endif  // __SYS_H__