123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280 |
- #include <vector>
- #include <utility>
- #include <algorithm>
- #include <memory>
- #include <map>
- #include <cstdlib>
- #include <cstring>
- #include <cerrno>
- #include "baseproc-sys.h"
- namespace {
- // which simulated file descriptors are currently "open"
- std::vector<bool> usedfds = {true, true, true};
- struct read_result
- {
- read_result(int errcode_p) : errcode(errcode_p) {}
- read_result(std::vector<char> &data_p) : errcode(0), data(data_p) {}
- read_result(std::vector<char> &&data_p) : errcode(0), data(std::move(data_p)) {}
- int errcode; // errno return
- std::vector<char> data; // data (if errcode == 0)
- };
- class read_cond : public std::vector<read_result>
- {
- public:
- using vector<read_result>::vector;
- // if blocking, return EAGAIN rather than end-of-file:
- bool is_blocking = false;
- };
- // map of fd to read results to supply for reads of that fd
- std::map<int,read_cond> read_data;
- // map of fd to the handler for writes to that fd
- std::map<int, std::unique_ptr<bp_sys::write_handler>> write_hndlr_map;
- // map of path to file content
- std::map<std::string, std::vector<char>> file_content_map;
- // environment variables, in "NAME=VALUE" form
- std::vector<char *> env_vars;
- } // anon namespace
- namespace bp_sys {
- char **environ = nullptr;
- int last_sig_sent = -1; // last signal number sent, accessible for tests.
- pid_t last_forked_pid = 1; // last forked process id (incremented each 'fork')
- // Test helper methods:
- void init_bpsys()
- {
- write_hndlr_map[0] = std::unique_ptr<bp_sys::write_handler> { new default_write_handler() };
- write_hndlr_map[1] = std::unique_ptr<bp_sys::write_handler> { new default_write_handler() };
- write_hndlr_map[2] = std::unique_ptr<bp_sys::write_handler> { new default_write_handler() };
- }
- // Allocate a file descriptor
- int allocfd()
- {
- return allocfd(new default_write_handler());
- }
- int allocfd(write_handler *whndlr)
- {
- auto f = std::find(usedfds.begin(), usedfds.end(), false);
- if (f == usedfds.end()) {
- int r = usedfds.size();
- usedfds.push_back(true);
- write_hndlr_map[r] = std::unique_ptr<bp_sys::write_handler>(whndlr);
- return r;
- }
- *f = true;
- auto r = f - usedfds.begin();
- write_hndlr_map[r] = std::unique_ptr<bp_sys::write_handler>(whndlr);
- return r;
- }
- // Supply data to be returned by read()
- void supply_read_data(int fd, std::vector<char> &data)
- {
- read_data[fd].emplace_back(data);
- }
- void supply_read_data(int fd, std::vector<char> &&data)
- {
- read_data[fd].emplace_back(std::move(data));
- }
- void set_blocking(int fd)
- {
- read_data[fd].is_blocking = true;
- }
- // retrieve data written via write()
- void extract_written_data(int fd, std::vector<char> &data)
- {
- auto &whandler = write_hndlr_map[fd];
- if (whandler == nullptr) abort();
- default_write_handler *dwhndlr = static_cast<default_write_handler *>(whandler.get());
- data = std::move(dwhndlr->data);
- }
- // Supply a file content
- void supply_file_content(const std::string &path, std::vector<char> &data)
- {
- file_content_map[path] = data;
- }
- void supply_file_content(const std::string &path, std::vector<char> &&data)
- {
- file_content_map[path] = std::move(data);
- }
- // Mock implementations of system calls:
- int open(const char *pathname, int flags)
- {
- auto i = file_content_map.find(pathname);
- if (i == file_content_map.end()) {
- errno = ENOENT;
- return -1;
- }
- int nfd = allocfd();
- supply_read_data(nfd, i->second);
- return nfd;
- }
- int pipe2(int fds[2], int flags)
- {
- fds[0] = allocfd();
- fds[1] = allocfd();
- return 0;
- }
- int close(int fd)
- {
- if (size_t(fd) >= usedfds.size()) abort();
- if (! usedfds[fd]) abort();
- usedfds[fd] = false;
- write_hndlr_map.erase(fd);
- return 0;
- }
- int kill(pid_t pid, int sig)
- {
- last_sig_sent = sig;
- return 0;
- }
- ssize_t read(int fd, void *buf, size_t count)
- {
- read_cond & rrs = read_data[fd];
- if (rrs.empty()) {
- if (rrs.is_blocking) {
- errno = EAGAIN;
- return -1;
- }
- return 0;
- }
- read_result &rr = rrs.front();
- if (rr.errcode != 0) {
- errno = rr.errcode;
- // Remove the result record:
- auto i = rrs.begin();
- i++;
- rrs.erase(rrs.begin(), i);
- return -1;
- }
- auto dsize = rr.data.size();
- if (dsize <= count) {
- // Consume entire result:
- std::copy_n(rr.data.begin(), dsize, (char *)buf);
- // Remove the result record:
- rrs.erase(rrs.begin());
- return dsize;
- }
- // Consume partial result:
- std::copy_n(rr.data.begin(), count, (char *)buf);
- rr.data.erase(rr.data.begin(), rr.data.begin() + count);
- return count;
- }
- ssize_t write(int fd, const void *buf, size_t count)
- {
- return write_hndlr_map[fd]->write(fd, buf, count);
- }
- ssize_t writev(int fd, const struct iovec *iov, int iovcnt)
- {
- ssize_t r = 0;
- for (int i = 0; i < iovcnt; i++) {
- ssize_t wr = write(fd, iov[i].iov_base, iov[i].iov_len);
- if (wr < 0) {
- if (r > 0) {
- return r;
- }
- return wr;
- }
- r += wr;
- if (size_t(wr) < iov[i].iov_len) {
- return r;
- }
- }
- return r;
- }
- char *getenv(const char *name)
- {
- size_t name_len = strlen(name);
- for (auto *var : env_vars) {
- if (strncmp(name, var, name_len) == 0) {
- if (var[name_len] == '=') {
- return &(var[name_len + 1]);
- }
- }
- }
- return nullptr;
- }
- int setenv(const char *name, const char *value, int overwrite)
- {
- size_t name_len = strlen(name);
- if (env_vars.empty()) {
- env_vars.push_back(nullptr);
- }
- else for (unsigned i = 0; i < (env_vars.size() - 1); ++i) {
- if (strncmp(name, env_vars[i], name_len) == 0) {
- if (env_vars[i][name_len] == '=') {
- // name matches, replace the value
- if (overwrite) {
- delete[](env_vars[i]);
- env_vars[i] = new char[name_len + 1 + strlen(value) + 1];
- strcpy(env_vars[i], name);
- env_vars[i][name_len] = '=';
- strcpy(env_vars[i] + name_len + 1, value);
- }
- return 0;
- }
- }
- }
- // not found, add
- char *new_var = new char[name_len + 1 + strlen(value) + 1];
- strcpy(new_var, name);
- new_var[name_len] = '=';
- strcpy(new_var + name_len + 1, value);
- env_vars[env_vars.size() - 1] = new_var;
- env_vars.push_back(nullptr);
- environ = env_vars.data();
- return 0;
- }
- int clearenv()
- {
- for (char *env_var : env_vars) {
- delete[](env_var);
- }
- env_vars.clear();
- environ = nullptr;
- return 0;
- }
- }
|