2016-01-27 15:43:34 +00:00
|
|
|
|
|
|
|
#include <vector>
|
|
|
|
#include <string>
|
|
|
|
#include <unordered_map>
|
|
|
|
|
|
|
|
#include <unistd.h>
|
|
|
|
#include <fcntl.h>
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
2016-01-30 17:45:50 +00:00
|
|
|
#include <cerrno>
|
2016-01-27 15:43:34 +00:00
|
|
|
|
|
|
|
#include "mpw-shell.h"
|
2016-02-02 01:38:29 +00:00
|
|
|
#include "fdset.h"
|
2016-01-27 15:43:34 +00:00
|
|
|
|
2016-01-30 17:45:50 +00:00
|
|
|
#include "phase1.h"
|
|
|
|
#include "phase2.h"
|
|
|
|
#include "command.h"
|
2016-01-27 15:43:34 +00:00
|
|
|
|
2016-02-02 03:32:21 +00:00
|
|
|
#include "mapped_file.h"
|
2016-02-02 21:20:32 +00:00
|
|
|
#include "error.h"
|
2016-02-02 03:32:21 +00:00
|
|
|
|
2016-02-02 02:17:46 +00:00
|
|
|
//#include <histedit.h>
|
|
|
|
#include <editline/readline.h>
|
2016-01-27 15:43:34 +00:00
|
|
|
|
2016-02-03 02:57:53 +00:00
|
|
|
#include <sys/types.h>
|
|
|
|
#include <pwd.h>
|
|
|
|
#include <sysexits.h>
|
|
|
|
|
|
|
|
//#include <uuid/uuid.h>
|
|
|
|
|
|
|
|
std::string root() {
|
|
|
|
|
|
|
|
static std::string root;
|
|
|
|
|
|
|
|
if (root.empty()) {
|
|
|
|
const char *cp = getenv("HOME");
|
|
|
|
if (!cp || !*cp) {
|
|
|
|
auto pw = getpwuid(getuid());
|
|
|
|
if (!pw) {
|
|
|
|
fprintf(stderr,"Unable to determine home directory\n.");
|
|
|
|
exit(EX_NOUSER);
|
|
|
|
}
|
|
|
|
cp = pw->pw_dir;
|
|
|
|
}
|
|
|
|
root = cp;
|
|
|
|
if (root.back() != '/') root.push_back('/');
|
|
|
|
root += "mpw/";
|
|
|
|
}
|
|
|
|
return root;
|
|
|
|
}
|
2016-01-27 15:43:34 +00:00
|
|
|
// should set {MPW}, {MPWVersion}, then execute {MPW}StartUp
|
2016-02-03 02:57:53 +00:00
|
|
|
void init(Environment &env) {
|
|
|
|
|
|
|
|
env.set("mpw", root());
|
|
|
|
env.set("status", std::string("0"));
|
|
|
|
env.set("exit", std::string("1")); // terminate script on error.
|
|
|
|
env.set("echo", std::string("1"));
|
2016-01-27 15:43:34 +00:00
|
|
|
}
|
|
|
|
|
2016-02-02 03:32:21 +00:00
|
|
|
int read_file(phase1 &p, const std::string &file) {
|
2016-02-02 21:20:32 +00:00
|
|
|
const mapped_file mf(file, mapped_file::readonly);
|
2016-02-02 03:32:21 +00:00
|
|
|
|
2016-02-02 21:20:32 +00:00
|
|
|
p.process(mf.begin(), mf.end(), true);
|
2016-02-02 03:32:21 +00:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
2016-02-02 21:20:32 +00:00
|
|
|
|
|
|
|
int read_fd(phase1 &p, int fd) {
|
2016-01-30 17:45:50 +00:00
|
|
|
|
|
|
|
unsigned char buffer[2048];
|
|
|
|
ssize_t size;
|
|
|
|
|
|
|
|
for (;;) {
|
2016-02-02 21:20:32 +00:00
|
|
|
size = read(fd, buffer, sizeof(buffer));
|
2016-01-30 17:45:50 +00:00
|
|
|
if (size == 0) break;
|
|
|
|
if (size < 0) {
|
|
|
|
if (errno == EINTR) continue;
|
|
|
|
perror("read: ");
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
try {
|
|
|
|
p.process(buffer, buffer + size);
|
|
|
|
} catch(std::exception &ex) {
|
|
|
|
fprintf(stderr, "%s\n", ex.what());
|
|
|
|
p.reset();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
try {
|
|
|
|
p.finish();
|
|
|
|
} catch(std::exception &ex) {
|
|
|
|
fprintf(stderr, "%s\n", ex.what());
|
|
|
|
p.reset();
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2016-02-02 02:17:46 +00:00
|
|
|
int interactive(phase1 &p) {
|
|
|
|
|
|
|
|
for(;;) {
|
|
|
|
char *cp = readline("# ");
|
|
|
|
if (!cp) break;
|
|
|
|
|
|
|
|
std::string s(cp);
|
|
|
|
free(cp);
|
|
|
|
|
|
|
|
if (s.empty()) continue;
|
2016-02-02 21:20:32 +00:00
|
|
|
|
|
|
|
// don't add if same as previous entry.
|
|
|
|
HIST_ENTRY *he = current_history();
|
|
|
|
if (he == nullptr || s != he->line)
|
|
|
|
add_history(s.c_str());
|
|
|
|
|
2016-02-02 02:17:46 +00:00
|
|
|
s.push_back('\n');
|
|
|
|
try {
|
|
|
|
p.process(s);
|
|
|
|
|
|
|
|
} catch(std::exception &ex) {
|
|
|
|
fprintf(stderr, "%s\n", ex.what());
|
|
|
|
p.reset();
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
try {
|
|
|
|
p.finish();
|
|
|
|
} catch(std::exception &ex) {
|
|
|
|
fprintf(stderr, "%s\n", ex.what());
|
|
|
|
p.reset();
|
|
|
|
}
|
|
|
|
|
|
|
|
fprintf(stdout, "\n");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2016-01-27 15:43:34 +00:00
|
|
|
int main(int argc, char **argv) {
|
|
|
|
|
2016-02-02 01:38:29 +00:00
|
|
|
Environment e;
|
|
|
|
init(e);
|
2016-01-27 15:43:34 +00:00
|
|
|
|
2016-01-30 17:45:50 +00:00
|
|
|
phase1 p1;
|
|
|
|
phase2 p2;
|
2016-01-31 05:41:30 +00:00
|
|
|
|
2016-01-30 17:45:50 +00:00
|
|
|
p1 >>= [&p2](std::string &&s) {
|
2016-02-02 21:20:32 +00:00
|
|
|
if (s.empty()) p2.finish();
|
|
|
|
else p2(std::move(s));
|
|
|
|
};
|
2016-01-30 17:45:50 +00:00
|
|
|
|
2016-02-02 21:20:32 +00:00
|
|
|
p2 >>= [&e](command_ptr_vector &&v) {
|
|
|
|
|
|
|
|
for (auto iter = v.begin(); iter != v.end(); ++iter) {
|
|
|
|
auto &ptr = *iter;
|
|
|
|
fdmask fds;
|
|
|
|
try {
|
|
|
|
ptr->execute(e, fds);
|
|
|
|
} catch (execution_of_input_terminated &ex) {
|
|
|
|
if (!(ptr->terminal() && ++iter == v.end())) {
|
|
|
|
fprintf(stderr, "%s\n", ex.what());
|
|
|
|
}
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
2016-01-30 17:45:50 +00:00
|
|
|
};
|
2016-02-02 21:20:32 +00:00
|
|
|
|
|
|
|
fprintf(stdout, "MPW Shell 0.0\n");
|
|
|
|
e.startup(true);
|
2016-02-02 03:32:21 +00:00
|
|
|
read_file(p1, "/Users/kelvin/mpw/Startup");
|
2016-02-02 21:20:32 +00:00
|
|
|
//p2.finish();
|
|
|
|
e.startup(false);
|
2016-02-02 03:32:21 +00:00
|
|
|
|
2016-02-02 21:20:32 +00:00
|
|
|
if (isatty(STDIN_FILENO))
|
|
|
|
interactive(p1);
|
|
|
|
else
|
|
|
|
read_fd(p1, STDIN_FILENO);
|
2016-01-31 05:41:30 +00:00
|
|
|
p2.finish();
|
2016-01-30 17:45:50 +00:00
|
|
|
|
2016-02-02 02:17:46 +00:00
|
|
|
return 0;
|
2016-01-27 15:43:34 +00:00
|
|
|
}
|