MAIN(appweb, int argc, char **argv, char **envp)
{
Mpr *mpr;
cchar *argp, *jail;
char *logSpec, *traceSpec;
int argind;
jail = 0;
logSpec = 0;
traceSpec = 0;
if ((mpr = mprCreate(argc, argv, MPR_USER_EVENTS_THREAD)) == NULL) {
exit(1);
}
if ((app = mprAllocObj(AppwebApp, manageApp)) == NULL) {
exit(2);
}
mprAddRoot(app);
mprAddStandardSignals();
#if ME_ROM
extern MprRomInode romFiles[];
mprSetRomFileSystem(romFiles);
#endif
if (httpCreate(HTTP_CLIENT_SIDE | HTTP_SERVER_SIDE) == 0) {
exit(3);
}
app->mpr = mpr;
app->workers = -1;
app->home = sclone(ME_SERVER_ROOT);
app->documents = app->home;
argc = mpr->argc;
argv = (char**) mpr->argv;
for (argind = 1; argind < argc; argind++) {
argp = argv[argind];
if (*argp != '-') {
break;
}
if (smatch(argp, "--config") || smatch(argp, "--conf")) {
if (argind >= argc) {
usageError();
}
app->configFile = sclone(argv[++argind]);
#if ME_UNIX_LIKE
} else if (smatch(argp, "--chroot")) {
if (argind >= argc) {
usageError();
}
jail = mprGetAbsPath(argv[++argind]);
#endif
} else if (smatch(argp, "--debugger") || smatch(argp, "-D")) {
mprSetDebugMode(1);
} else if (smatch(argp, "--exe")) {
if (argind >= argc) {
usageError();
}
mpr->argv[0] = mprGetAbsPath(argv[++argind]);
mprSetAppPath(mpr->argv[0]);
mprSetModuleSearchPath(NULL);
} else if (smatch(argp, "--home")) {
if (argind >= argc) {
usageError();
}
app->home = sclone(argv[++argind]);
if (chdir(app->home) < 0) {
mprLog("error appweb", 0, "Cannot change directory to %s", app->home);
exit(4);
}
} else if (smatch(argp, "--log") || smatch(argp, "-l")) {
if (argind >= argc) {
usageError();
}
logSpec = argv[++argind];
} else if (smatch(argp, "--name") || smatch(argp, "-n")) {
if (argind >= argc) {
usageError();
}
mprSetAppName(argv[++argind], 0, 0);
} else if (smatch(argp, "--threads")) {
if (argind >= argc) {
usageError();
}
app->workers = atoi(argv[++argind]);
} else if (smatch(argp, "--show") || smatch(argp, "-s")) {
app->show = 1;
} else if (smatch(argp, "--trace") || smatch(argp, "-t")) {
if (argind >= argc) {
usageError();
}
traceSpec = argv[++argind];
} else if (smatch(argp, "--verbose") || smatch(argp, "-v")) {
if (!logSpec) {
logSpec = sfmt("stderr:2");
}
if (!traceSpec) {
traceSpec = sfmt("stderr:2");
}
} else if (smatch(argp, "--version") || smatch(argp, "-V")) {
mprPrintf("%s\n", ME_VERSION);
exit(0);
} else if (*argp == '-' && isdigit((uchar) argp[1])) {
if (!logSpec) {
logSpec = sfmt("stderr:%d", (int) stoi(&argp[1]));
}
if (!traceSpec) {
traceSpec = sfmt("stderr:%d", (int) stoi(&argp[1]));
}
} else if (smatch(argp, "-?") || scontains(argp, "-help")) {
usageError();
exit(5);
} else if (*argp == '-') {
mprLog("error appweb", 0, "Unknown switch \"%s\"", argp);
usageError();
exit(5);
}
}
app->home = mprGetAbsPath(app->home);
if (logSpec) {
mprStartLogging(logSpec, MPR_LOG_DETAILED | MPR_LOG_CONFIG | MPR_LOG_CMDLINE);
}
if (traceSpec) {
httpStartTracing(traceSpec);
}
if (mprStart() < 0) {
mprLog("error appweb", 0, "Cannot start MPR");
mprDestroy();
return MPR_ERR_CANT_INITIALIZE;
}
if (checkEnvironment(argv[0]) < 0) {
exit(6);
}
if (argc == argind && !app->configFile) {
if (findAppwebConf() < 0) {
exit(7);
}
}
if (jail && changeRoot(jail) < 0) {
exit(8);
}
if (createEndpoints(argc - argind, &argv[argind]) < 0) {
return MPR_ERR_CANT_INITIALIZE;
}
if (maStartAppweb(app->appweb) < 0) {
mprLog("error appweb", 0, "Cannot start HTTP service, exiting.");
exit(9);
}
if (app->show) {
httpLogRoutes(0, 0);
}
mprServiceEvents(-1, 0);
mprLog("info appweb", 1, "Stopping Appweb ...");
mprDestroy();
return mprGetExitStatus();
}
MAIN(ejsMain, int argc, char **argv, char **envp)
{
Mpr *mpr;
EcCompiler *cp;
Ejs *ejs;
cchar *cmd, *className, *method, *homeDir;
char *argp, *searchPath, *modules, *name, *tok, *extraFiles;
int nextArg, err, ecFlags, stats, merge, bind, noout, debug, optimizeLevel, warnLevel, strict, i, next;
/*
Initialize Multithreaded Portable Runtime (MPR)
*/
mpr = mprCreate(argc, argv, 0);
app = mprAllocObj(App, manageApp);
mprAddRoot(app);
mprAddStandardSignals();
if (mprStart(mpr) < 0) {
mprError("Cannot start mpr services");
return EJS_ERR;
}
err = 0;
className = 0;
cmd = 0;
method = 0;
searchPath = 0;
stats = 0;
merge = 0;
bind = 1;
noout = 1;
debug = 1;
warnLevel = 1;
optimizeLevel = 9;
strict = 0;
app->files = mprCreateList(-1, 0);
app->iterations = 1;
argc = mpr->argc;
argv = (char**) mpr->argv;
for (nextArg = 1; nextArg < argc; nextArg++) {
argp = argv[nextArg];
if (*argp != '-') {
break;
}
if (smatch(argp, "--bind")) {
bind = 1;
} else if (smatch(argp, "--class")) {
if (nextArg >= argc) {
err++;
} else {
className = argv[++nextArg];
}
} else if (smatch(argp, "--chdir") || smatch(argp, "--home") || smatch(argp, "-C")) {
if (nextArg >= argc) {
err++;
} else {
homeDir = argv[++nextArg];
if (chdir((char*) homeDir) < 0) {
mprError("Cannot change directory to %s", homeDir);
}
}
#if BIT_UNIX_LIKE
} else if (smatch(argp, "--chroot")) {
/* Not documented or supported */
if (nextArg >= argc) {
err++;
} else {
homeDir = mprGetAbsPath(argv[++nextArg]);
if (chroot(homeDir) < 0) {
if (errno == EPERM) {
mprPrintfError("%s: Must be super user to use the --chroot option", mprGetAppName(mpr));
} else {
mprPrintfError("%s: Cannot change change root directory to %s, errno %d",
mprGetAppName(), homeDir, errno);
}
return 4;
}
}
#endif
} else if (smatch(argp, "--cmd") || smatch(argp, "-c")) {
if (nextArg >= argc) {
err++;
} else {
cmd = argv[++nextArg];
}
#if BIT_WIN_LIKE
} else if (smatch(argp, "--cygroot")) {
if (nextArg >= argc) {
err++;
} else {
app->cygroot = sclone(argv[++nextArg]);
}
#endif
} else if (smatch(argp, "--debug")) {
debug = 1;
} else if (smatch(argp, "--debugger") || smatch(argp, "-D")) {
mprSetDebugMode(1);
} else if (smatch(argp, "--files") || smatch(argp, "-f")) {
/* Compatibility with mozilla shell */
if (nextArg >= argc) {
err++;
} else {
extraFiles = sclone(argv[++nextArg]);
name = stok(extraFiles, " \t", &tok);
while (name != NULL) {
mprAddItem(app->files, sclone(name));
name = stok(NULL, " \t", &tok);
}
}
} else if (smatch(argp, "--iterations") || smatch(argp, "-i")) {
if (nextArg >= argc) {
err++;
} else {
app->iterations = atoi(argv[++nextArg]);
}
} else if (smatch(argp, "--log")) {
if (nextArg >= argc) {
err++;
} else {
mprStartLogging(argv[++nextArg], 0);
mprSetCmdlineLogging(1);
}
} else if (smatch(argp, "--method")) {
if (nextArg >= argc) {
err++;
} else {
method = argv[++nextArg];
}
} else if (smatch(argp, "--name")) {
/* Just ignore. Used to tag commands with a unique command line */
nextArg++;
} else if (smatch(argp, "--nobind")) {
bind = 0;
} else if (smatch(argp, "--nodebug")) {
debug = 0;
} else if (smatch(argp, "--optimize")) {
if (nextArg >= argc) {
err++;
} else {
optimizeLevel = atoi(argv[++nextArg]);
}
} else if (smatch(argp, "-s")) {
/* Compatibility with mozilla shell. Just ignore */
} else if (smatch(argp, "--search") || smatch(argp, "--searchpath")) {
if (nextArg >= argc) {
err++;
} else {
searchPath = argv[++nextArg];
}
} else if (smatch(argp, "--standard")) {
strict = 0;
} else if (smatch(argp, "--stats")) {
stats = 1;
} else if (smatch(argp, "--strict")) {
strict = 1;
} else if (smatch(argp, "--require")) {
if (nextArg >= argc) {
err++;
} else {
if (app->modules == 0) {
app->modules = mprCreateList(-1, 0);
}
modules = sclone(argv[++nextArg]);
name = stok(modules, " \t", &tok);
while (name != NULL) {
require(name);
name = stok(NULL, " \t", &tok);
}
}
} else if (smatch(argp, "--verbose") || smatch(argp, "-v")) {
mprStartLogging("stderr:2", 0);
mprSetCmdlineLogging(1);
} else if (smatch(argp, "--version") || smatch(argp, "-V")) {
mprPrintf("%s-%s\n", BIT_VERSION, BIT_BUILD_NUMBER);
return 0;
} else if (smatch(argp, "--warn")) {
if (nextArg >= argc) {
err++;
} else {
warnLevel = atoi(argv[++nextArg]);
}
} else {
err++;
break;
}
}
if (err) {
/*
If --method or --class is specified, then the named class.method will be run (method defaults to "main", class
defaults to first class with a "main").
Examples:
ejs
ejs script.es arg1 arg2 arg3
ejs --class "Customer" --method "start" --files "script1.es script2.es" main.es
*/
mprPrintfError("Usage: %s [options] script.es [arguments] ...\n"
" Ejscript shell program options:\n"
" --class className # Name of class containing method to run\n"
" --cmd ejscriptCode # Literal ejscript statements to execute\n"
" --cygroot path # Set cygwin root for resolving script paths\n"
" --debug # Use symbolic debugging information (default)\n"
" --debugger # Disable timeouts to make using a debugger easier\n"
" --files \"files..\" # Extra source to compile\n"
" --log logSpec # Internal compiler diagnostics logging\n"
" --method methodName # Name of method to run. Defaults to main\n"
" --nodebug # Omit symbolic debugging information\n"
" --optimize level # Set the optimization level (0-9 default is 9)\n"
" --require 'module,...' # Required list of modules to pre-load\n"
" --search ejsPath # Module search path\n"
" --standard # Default compilation mode to standard (default)\n"
" --stats # Print memory stats on exit\n"
" --strict # Default compilation mode to strict\n"
" --verbose | -v # Same as --log stderr:2 \n"
" --version # Emit the compiler version information\n"
" --warn level # Set the warning message level (0-9 default is 0)\n\n",
mpr->name);
return -1;
}
if ((ejs = ejsCreateVM(argc - nextArg, (cchar **) &argv[nextArg], 0)) == 0) {
return MPR_ERR_MEMORY;
}
app->ejs = ejs;
if (ejsLoadModules(ejs, searchPath, app->modules) < 0) {
return MPR_ERR_CANT_READ;
}
ecFlags = 0;
ecFlags |= (merge) ? EC_FLAGS_MERGE: 0;
ecFlags |= (bind) ? EC_FLAGS_BIND: 0;
ecFlags |= (noout) ? EC_FLAGS_NO_OUT: 0;
ecFlags |= (debug) ? EC_FLAGS_DEBUG: 0;
cp = app->compiler = ecCreateCompiler(ejs, ecFlags);
if (cp == 0) {
return MPR_ERR_MEMORY;
}
ecSetRequire(cp, app->modules);
ecSetOptimizeLevel(cp, optimizeLevel);
ecSetWarnLevel(cp, warnLevel);
ecSetStrictMode(cp, strict);
if (nextArg < argc) {
mprAddItem(app->files, sclone(argv[nextArg]));
}
if (app->cygroot) {
/*
When cygwin invokes a script with shebang, it passes a cygwin path to the script
The ejs --cygroot option permits ejscript to conver this to a native path
*/
for (next = 0; (name = mprGetNextItem(app->files, &next)) != 0; ) {
if (*name == '/' || *name == '\\') {
mprSetItem(app->files, next - 1, sjoin(app->cygroot, name, NULL));
}
}
}
for (i = 0; !err && i < app->iterations; i++) {
if (cmd) {
if (interpretCommands(cp, cmd) < 0) {
err++;
}
} else if (mprGetListLength(app->files) > 0) {
if (interpretFiles(cp, app->files, argc - nextArg, &argv[nextArg], className, method) < 0) {
err++;
}
} else {
/*
No args - run as an interactive shell
*/
if (interpretCommands(cp, NULL) < 0) {
err++;
}
}
}
if (stats) {
#if BIT_DEBUG
mprSetLogLevel(1);
mprPrintMem("Memory Usage", 1);
#endif
}
if (!err) {
err = mpr->exitStatus;
}
app->ejs = 0;
mprTerminate(MPR_EXIT_DEFAULT, err);
ejsDestroyVM(ejs);
mprDestroy(MPR_EXIT_DEFAULT);
return err;
}
static bool parseArgs(int argc, char **argv)
{
char *argp, *key, *value;
int i, setWorkers, nextArg;
setWorkers = 0;
for (nextArg = 1; nextArg < argc; nextArg++) {
argp = argv[nextArg];
if (*argp != '-') {
break;
}
if (smatch(argp, "--benchmark") || smatch(argp, "-b")) {
app->benchmark++;
} else if (smatch(argp, "--cert")) {
if (nextArg >= argc) {
return 0;
} else {
app->cert = sclone(argv[++nextArg]);
}
} else if (smatch(argp, "--chunk")) {
if (nextArg >= argc) {
return 0;
} else {
value = argv[++nextArg];
app->chunkSize = atoi(value);
if (app->chunkSize < 0) {
mprError("Bad chunksize %d", app->chunkSize);
return 0;
}
}
} else if (smatch(argp, "--continue")) {
app->continueOnErrors++;
} else if (smatch(argp, "--cookie")) {
if (nextArg >= argc) {
return 0;
} else {
mprAddItem(app->headers, mprCreateKeyPair("Cookie", argv[++nextArg]));
}
} else if (smatch(argp, "--data")) {
if (nextArg >= argc) {
return 0;
} else {
if (app->bodyData == 0) {
app->bodyData = mprCreateBuf(-1, -1);
}
mprPutStringToBuf(app->bodyData, argv[++nextArg]);
}
} else if (smatch(argp, "--debugger") || smatch(argp, "-D")) {
mprSetDebugMode(1);
app->retries = 0;
app->timeout = MAXINT;
} else if (smatch(argp, "--delete")) {
app->method = "DELETE";
} else if (smatch(argp, "--form") || smatch(argp, "-f")) {
if (nextArg >= argc) {
return 0;
} else {
if (app->formData == 0) {
app->formData = mprCreateList(-1, 0);
}
addFormVars(argv[++nextArg]);
}
} else if (smatch(argp, "--header")) {
if (nextArg >= argc) {
return 0;
} else {
key = argv[++nextArg];
if ((value = strchr(key, ':')) == 0) {
mprError("Bad header format. Must be \"key: value\"");
return 0;
}
*value++ = '\0';
while (isspace((uchar) *value)) {
value++;
}
mprAddItem(app->headers, mprCreateKeyPair(key, value));
}
} else if (smatch(argp, "--host")) {
if (nextArg >= argc) {
return 0;
} else {
app->host = argv[++nextArg];
if (*app->host == ':') {
app->host = &app->host[1];
}
if (isPort(app->host)) {
app->host = sfmt("http://127.0.0.1:%s", app->host);
} else {
app->host = sclone(app->host);
}
}
} else if (smatch(argp, "--insecure")) {
app->insecure++;
} else if (smatch(argp, "--iterations") || smatch(argp, "-i")) {
if (nextArg >= argc) {
return 0;
} else {
app->iterations = atoi(argv[++nextArg]);
}
} else if (smatch(argp, "--log") || smatch(argp, "-l")) {
if (nextArg >= argc) {
return 0;
} else {
mprStartLogging(argv[++nextArg], 0);
}
} else if (smatch(argp, "--method") || smatch(argp, "-m")) {
if (nextArg >= argc) {
return 0;
} else {
app->method = argv[++nextArg];
}
} else if (smatch(argp, "--out") || smatch(argp, "-o")) {
if (nextArg >= argc) {
return 0;
} else {
app->outFilename = sclone(argv[++nextArg]);
}
} else if (smatch(argp, "--noout") || smatch(argp, "-n") ||
smatch(argp, "--quiet") || smatch(argp, "-q")) {
app->noout++;
} else if (smatch(argp, "--nofollow")) {
app->nofollow++;
} else if (smatch(argp, "--password") || smatch(argp, "-p")) {
if (nextArg >= argc) {
return 0;
} else {
app->password = sclone(argv[++nextArg]);
}
} else if (smatch(argp, "--post")) {
app->method = "POST";
} else if (smatch(argp, "--printable")) {
app->printable++;
} else if (smatch(argp, "--protocol")) {
if (nextArg >= argc) {
return 0;
} else {
app->protocol = supper(argv[++nextArg]);
}
} else if (smatch(argp, "--put")) {
app->method = "PUT";
} else if (smatch(argp, "--range")) {
if (nextArg >= argc) {
return 0;
} else {
// TODO - should allow multiple ranges
if (app->ranges == 0) {
app->ranges = sfmt("bytes=%s", argv[++nextArg]);
} else {
app->ranges = srejoin(app->ranges, ",", argv[++nextArg], NULL);
}
}
} else if (smatch(argp, "--retries") || smatch(argp, "-r")) {
if (nextArg >= argc) {
return 0;
} else {
app->retries = atoi(argv[++nextArg]);
}
} else if (smatch(argp, "--sequence")) {
app->sequence++;
} else if (smatch(argp, "--showHeaders") || smatch(argp, "--show") || smatch(argp, "-s")) {
app->showHeaders++;
} else if (smatch(argp, "--showStatus") || smatch(argp, "--showCode")) {
app->showStatus++;
} else if (smatch(argp, "--single") || smatch(argp, "-s")) {
app->singleStep++;
} else if (smatch(argp, "--text")) {
app->text++;
} else if (smatch(argp, "--threads") || smatch(argp, "-t")) {
if (nextArg >= argc) {
return 0;
} else {
app->loadThreads = atoi(argv[++nextArg]);
}
} else if (smatch(argp, "--timeout")) {
if (nextArg >= argc) {
return 0;
} else {
app->timeout = atoi(argv[++nextArg]) * MPR_TICKS_PER_SEC;
}
} else if (smatch(argp, "--upload") || smatch(argp, "-u")) {
app->upload++;
} else if (smatch(argp, "--user") || smatch(argp, "--username")) {
if (nextArg >= argc) {
return 0;
} else {
app->username = argv[++nextArg];
}
} else if (smatch(argp, "--verbose") || smatch(argp, "-v")) {
app->verbose++;
} else if (smatch(argp, "--version") || smatch(argp, "-V")) {
mprPrintfError("%s %s\n"
"Copyright (C) Embedthis Software 2003-2012\n"
"Copyright (C) Michael O'Brien 2003-2012\n",
BIT_NAME, BIT_VERSION);
exit(0);
} else if (smatch(argp, "--workerTheads") || smatch(argp, "-w")) {
if (nextArg >= argc) {
return 0;
} else {
app->workers = atoi(argv[++nextArg]);
}
setWorkers++;
} else if (smatch(argp, "--zero")) {
app->zeroOnErrors++;
} else if (smatch(argp, "--")) {
nextArg++;
break;
} else if (smatch(argp, "-")) {
break;
} else {
return 0;
break;
}
}
if (argc == nextArg) {
return 0;
}
app->nextArg = nextArg;
argc = argc - nextArg;
argv = &argv[nextArg];
app->target = argv[argc - 1];
app->iterations *= app->loadThreads;
if (--argc > 0) {
/*
Files present on command line
*/
app->files = mprCreateList(argc, MPR_LIST_STATIC_VALUES);
for (i = 0; i < argc; i++) {
mprAddItem(app->files, argv[i]);
}
}
if (!setWorkers) {
app->workers = app->loadThreads + 2;
}
if (app->method == 0) {
if (app->bodyData || app->formData || app->upload) {
app->method = "POST";
} else if (app->files) {
app->method = "PUT";
} else {
app->method = "GET";
}
}
return 1;
}
int main(int argc, char *argv[])
{
double elapsed;
char *programName, *argp, *logSpec;
int c, errflg, start;
#if BLD_FEATURE_LOG
MprLogToFile *logger;
#endif
#if BLD_FEATURE_MULTITHREAD
MprThread *threadp;
#endif
programName = mprGetBaseName(argv[0]);
method = "GET";
fileList = cmpDir = writeDir = 0;
verbose = continueOnErrors = outputHeader = errflg = 0;
poolThreads = 4; // Need at least one to run efficiently
httpVersion = 1; // HTTP/1.1
success = 1;
trace = 0;
host = "localhost";
logSpec = "stdout:1";
postData = 0;
postLen = 0;
retries = MPR_HTTP_CLIENT_RETRIES;
iterations = HTTP_DEFAULT_ITERATIONS;
loadThreads = HTTP_DEFAULT_LOAD_THREADS;
timeout = MPR_HTTP_CLIENT_TIMEOUT;
#if BLD_FEATURE_MULTITHREAD
mutex = new MprMutex();
#endif
//
// FUTURE: switch to GNU style --args with a better usage message
//
MprCmdLine cmdLine(argc, argv, "bC:cDd:f:Hh:i:l:M:mqo:r:st:T:vV:w:");
while ((c = cmdLine.next(&argp)) != EOF) {
switch(c) {
case 'b':
benchmark++;
break;
case 'c':
continueOnErrors++;
break;
case 'C':
cmpDir = argp;
break;
case 'd':
postData = argp;
postLen = strlen(postData);
break;
case 'D':
mprSetDebugMode(1);
break;
case 'f':
fileList = argp;
break;
case 'h':
host = argp;
break;
case 'H':
outputHeader++;
break;
case 'i':
iterations = atoi(argp);
break;
case 'l':
logSpec = argp;
break;
case 'm':
mprRequestMemStats(1);
break;
case 'M':
method = argp;
break;
case 'o':
timeout = atoi(argp);
break;
case 'q':
quietMode++;
break;
case 'r':
retries = atoi(argp);
break;
case 's':
singleStep++;
break;
case 't':
loadThreads = atoi(argp);
break;
case 'T':
poolThreads = atoi(argp);
break;
case 'v':
verbose++;
trace++;
break;
case 'V':
httpVersion = atoi(argp);
break;
case 'w':
writeDir = argp;
break;
default:
errflg++;
break;
}
}
if (writeDir && (loadThreads > 1)) {
errflg++;
}
if (errflg) {
mprFprintf(MPR_STDERR,
"usage: %s [-bcHMmqsTv] [-C cmpDir] [-d postData] [-f fileList]\n"
" [-i iterations] [-l logSpec] [-M method] [-o timeout]\n"
" [-h host] [-r retries] [-t threads] [-T poolThreads]\n"
" [-V httpVersion] [-w writeDir] [urls...]\n", programName);
exit(2);
}
saveArgc = argc - cmdLine.firstArg();
saveArgv = &argv[cmdLine.firstArg()];
mpr = new Mpr(programName);
#if BLD_FEATURE_LOG
tMod = new MprLogModule("httpClient");
logger = new MprLogToFile();
mpr->addListener(logger);
if (mpr->setLogSpec(logSpec) < 0) {
mprFprintf(MPR_STDERR, "Can't open log file %s\n", logSpec);
exit(2);
}
#endif
//
// Alternatively, set the configuration manually
//
mpr->setAppTitle("Embedthis HTTP Client");
#if BLD_FEATURE_MULTITHREAD
mpr->setMaxPoolThreads(poolThreads);
#endif
//
// Start the Timer, Socket and Pool services
//
if (mpr->start(MPR_SERVICE_THREAD) < 0) {
mprError(MPR_L, MPR_USER, "Can't start MPR for %s", mpr->getAppTitle());
delete mpr;
exit(2);
}
//
// Create extra test threads to run the tests as required. We use
// the main thread also (so start with j==1)
//
start = mprGetTime(0);
#if BLD_FEATURE_MULTITHREAD
activeLoadThreads = loadThreads;
for (int j = 1; j < loadThreads; j++) {
char name[64];
mprSprintf(name, sizeof(name), "t.%d", j - 1);
threadp = new MprThread(doTests, MPR_NORMAL_PRIORITY, (void*) j, name);
threadp->start();
}
#endif
doTests(0, 0);
//
// Wait for all the threads to complete (simple but effective). Keep
// servicing events as we wind down.
//
while (activeLoadThreads > 1) {
mprSleep(100);
}
if (benchmark && success) {
elapsed = (mprGetTime(0) - start);
if (fetchCount == 0) {
elapsed = 0;
fetchCount = 1;
}
mprPrintf("\tThreads %d, Pool Threads %d \t%13.2f\t%12.2f\t%6d\n",
loadThreads, poolThreads, elapsed * 1000.0 / fetchCount,
elapsed / 1000.0, fetchCount);
mprPrintf("\nTime elapsed: %13.4f sec\n", elapsed / 1000.0);
mprPrintf("Time per request: %13.4f sec\n", elapsed / 1000.0
/ fetchCount);
mprPrintf("Requests per second: %13.4f\n", fetchCount * 1.0 /
(elapsed / 1000.0));
}
if (! quietMode) {
mprPrintf("\n");
}
mpr->stop(0);
#if BLD_FEATURE_MULTITHREAD
delete mutex;
#endif
#if BLD_FEATURE_LOG
delete tMod;
#endif
delete mpr;
#if BLD_FEATURE_LOG
delete logger;
#endif
mprMemClose();
return (success) ? 0 : 255;
}
MAIN(appweb, int argc, char **argv)
{
Mpr *mpr;
MaHttp *http;
cchar *ipAddrPort, *documentRoot, *argp, *logSpec;
char *configFile, *ipAddr, *homeDir, *timeText, *ejsPrefix, *ejsPath, *changeRoot;
int workers, outputVersion, argind, port;
documentRoot = 0;
changeRoot = ejsPrefix = ejsPath = 0;
ipAddrPort = 0;
ipAddr = 0;
port = -1;
logSpec = 0;
server = 0;
outputVersion = 0;
workers = -1;
configFile = BLD_FEATURE_CONFIG_FILE;
homeDir = BLD_FEATURE_SERVER_ROOT;
mpr = mprCreate(argc, argv, memoryFailure);
argc = mpr->argc;
argv = mpr->argv;
#if BLD_FEATURE_ROMFS
extern MprRomInode romFiles[];
mprSetRomFileSystem(mpr, romFiles);
#endif
if (osInit(mpr) < 0) {
exit(2);
}
if (mprStart(mpr, 0) < 0) {
mprUserError(mpr, "Can't start MPR for %s", mprGetAppName(mpr));
mprFree(mpr);
return MPR_ERR_CANT_INITIALIZE;
}
for (argind = 1; argind < argc; argind++) {
argp = argv[argind];
if (*argp != '-') {
break;
}
if (strcmp(argp, "--config") == 0) {
if (argind >= argc) {
return printUsage(mpr);
}
configFile = argv[++argind];
#if BLD_UNIX_LIKE
} else if (strcmp(argp, "--chroot") == 0) {
if (argind >= argc) {
return printUsage(mpr);
}
changeRoot = mprGetAbsPath(mpr, argv[++argind]);
#endif
} else if (strcmp(argp, "--debug") == 0 || strcmp(argp, "-D") == 0 ||
strcmp(argp, "-d") == 0 || strcmp(argp, "--debugger") == 0) {
mprSetDebugMode(mpr, 1);
} else if (strcmp(argp, "--ejs") == 0) {
if (argind >= argc) {
return printUsage(mpr);
}
ejsPrefix = mprStrdup(mpr, argv[++argind]);
if ((ejsPath = strchr(ejsPrefix, ':')) != 0) {
*ejsPath++ = '\0';
}
ejsPath = mprGetAbsPath(mpr, ejsPath);
} else if (strcmp(argp, "--home") == 0) {
if (argind >= argc) {
return printUsage(mpr);
}
homeDir = mprGetAbsPath(mpr, argv[++argind]);
if (chdir((char*) homeDir) < 0) {
mprPrintfError(mpr, "%s: Can't change directory to %s\n", mprGetAppName(mpr), homeDir);
exit(2);
}
} else if (strcmp(argp, "--log") == 0 || strcmp(argp, "-l") == 0) {
if (argind >= argc) {
return printUsage(mpr);
}
logSpec = argv[++argind];
maStartLogging(mpr, logSpec);
} else if (strcmp(argp, "--name") == 0 || strcmp(argp, "-n") == 0) {
if (argind >= argc) {
return printUsage(mpr);
}
mprSetAppName(mpr, argv[++argind], 0, 0);
} else if (strcmp(argp, "--threads") == 0) {
if (argind >= argc) {
return printUsage(mpr);
}
workers = atoi(argv[++argind]);
} else if (strcmp(argp, "--verbose") == 0 || strcmp(argp, "-v") == 0) {
maStartLogging(mpr, "stdout:2");
} else if (strcmp(argp, "--version") == 0 || strcmp(argp, "-V") == 0) {
outputVersion++;
} else {
mprPrintfError(mpr, "Unknown switch \"%s\"\n", argp);
printUsage(mpr);
exit(2);
}
}
if (argc > argind) {
if (argc > (argind + 2)) {
return printUsage(mpr);
}
ipAddrPort = argv[argind++];
if (argc > argind) {
documentRoot = argv[argind++];
} else {
documentRoot = ".";
}
}
if (outputVersion) {
mprPrintf(mpr, "%s %s-%s\n", mprGetAppTitle(mpr), BLD_VERSION, BLD_NUMBER);
exit(0);
}
if (ipAddrPort) {
mprParseIp(mpr, ipAddrPort, &ipAddr, &port, MA_SERVER_DEFAULT_PORT_NUM);
} else {
#if BLD_FEATURE_CONFIG_PARSE
if (configFile == 0) {
configFile = mprStrcat(mpr, -1, mprGetAppName(mpr), ".conf", NULL);
}
if (!mprPathExists(mpr, configFile, R_OK)) {
mprPrintfError(mpr, "Can't open config file %s\n", configFile);
exit(2);
}
#else
return printUsage(mpr);
#endif
#if !BLD_FEATURE_ROMFS
if (homeDir == 0) {
homeDir = mprGetPathParent(mpr, configFile);
if (chdir((char*) homeDir) < 0) {
mprPrintfError(mpr, "%s: Can't change directory to %s\n", mprGetAppName(mpr), homeDir);
exit(2);
}
}
#endif
}
homeDir = mprGetCurrentPath(mpr);
if (checkEnvironment(mpr, argv[0], homeDir) < 0) {
exit(3);
}
#if BLD_UNIX_LIKE
if (changeRoot) {
homeDir = mprGetAbsPath(mpr, changeRoot);
if (chdir(homeDir) < 0) {
mprError(mpr, "%s: Can't change directory to %s", homeDir);
exit(4);
}
if (chroot(homeDir) < 0) {
if (errno == EPERM) {
mprError(mpr, "%s: Must be super user to use the --chroot option", mprGetAppName(mpr));
} else {
mprError(mpr, "%s: Can't change change root directory to %s, errno %d",
mprGetAppName(mpr), homeDir, errno);
}
exit(5);
}
}
#endif
/*
* Create the top level HTTP service and default HTTP server. Set the initial server root to "."
*/
http = maCreateHttp(mpr);
if (http == 0) {
mprUserError(mpr, "Can't create HTTP service for %s", mprGetAppName(mpr));
return MPR_ERR_CANT_INITIALIZE;
}
server = maCreateServer(http, "default", ".", 0, -1);
if (server == 0) {
mprUserError(mpr, "Can't create HTTP server for %s", mprGetAppName(mpr));
return MPR_ERR_CANT_INITIALIZE;
}
if (maConfigureServer(mpr, http, server, configFile, ipAddr, port, documentRoot) < 0) {
/* mprUserError(mpr, "Can't configure the server, exiting."); */
exit(6);
}
if (mpr->ipAddr) {
mprLog(mpr, 2, "Server IP address %s", mpr->ipAddr);
}
timeText = mprFormatLocalTime(mpr, mprGetTime(mpr));
mprLog(mpr, 1, "Started at %s", timeText);
mprFree(timeText);
#if BLD_FEATURE_EJS
if (ejsPrefix) {
createEjsAlias(mpr, http, server, ejsPrefix, ejsPath);
}
#endif
#if BLD_FEATURE_MULTITHREAD
if (workers >= 0) {
mprSetMaxWorkers(http, workers);
}
#endif
#if BLD_WIN_LIKE
if (!ejsPrefix) {
writePort(server->defaultHost);
}
#endif
if (maStartHttp(http) < 0) {
mprUserError(mpr, "Can't start HTTP service, exiting.");
exit(7);
}
#if BLD_FEATURE_MULTITHREAD
mprLog(mpr, 1, "HTTP services are ready with max %d worker threads", mprGetMaxWorkers(mpr));
#else
mprLog(mpr, 1, "HTTP services are ready (single-threaded)");
#endif
/*
* Service I/O events until instructed to exit
*/
while (!mprIsExiting(mpr)) {
mprServiceEvents(mpr->dispatcher, -1, MPR_SERVICE_EVENTS | MPR_SERVICE_IO);
}
/*
* Signal a graceful shutdown
*/
mprLog(http, 1, "Exiting ...");
maStopHttp(http);
mprLog(http, 1, "Exit complete");
#if VXWORKS
if (mprStop(mpr)) {
mprFree(mpr);
}
#endif
return 0;
}
static int realMain(MprCmdLine *cmdLine)
{
MaHttp *http;
char portNumBuf[MPR_MAX_IP_PORT];
char *argp, *logSpec;
int c, errflg, kill, poolThreads, outputVersion;
mprSetMemHandler(memoryFailure);
mprCreateMemHeap(0, 16 * 1024, MAXINT);
program = mprGetBaseName(cmdLine->getArgv()[0]);
poolThreads = -1;
kill = errflg = 0;
logSpec = 0;
outputVersion = 0;
autoScan = 1;
background = 0;
serverRoot = 0;
docRoot = "web";
#if !WIN && !WINCE && !VXWORKS
if (getuid()) {
ipAddr = mprStrdup("4000");
} else {
mprSprintf(portNumBuf, sizeof(portNumBuf), "%d",
MA_SERVER_DEFAULT_PORT_NUM);
ipAddr = mprStrdup(portNumBuf);
}
#else
mprSprintf(portNumBuf, sizeof(portNumBuf), "%d",
MA_SERVER_DEFAULT_PORT_NUM);
ipAddr = mprStrdup(portNumBuf);
#endif
while ((c = cmdLine->next(&argp)) != EOF) {
switch(c) {
case 'A':
autoScan = 0;
break;
case 'a':
mprFree(ipAddr);
ipAddr = mprStrdup(argp);
break;
case 'b':
background++;
break;
case 'c':
/* Ignored */
break;
case 'D':
mprSetDebugMode(1);
break;
case 'd':
docRoot = argp;
break;
case 'f':
#if BLD_FEATURE_CONFIG_PARSE
configFile = argp;
autoScan = 0;
#else
errflg++;
#endif
break;
case 'k':
kill++;
logSpec = 0;
break;
case 'l':
logSpec = (*argp) ? argp : 0;
break;
case 'm':
mprRequestMemStats(1);
break;
case 'r':
serverRoot = argp;
break;
case 't':
poolThreads = atoi(argp);
break;
//
// FUTURE -- just for test. Will be removed
//
case 'w':
writeFile = argp;
break;
case 'v':
outputVersion++;
break;
#if WIN && BLD_FEATURE_RUN_AS_SERVICE
case 'i':
serviceOp = MPR_INSTALL_SERVICE;
if (strcmp(argp, "none") == 0) {
serviceCmdLine = "";
} else if (strcmp(argp, "default") == 0) {
serviceCmdLine = "-b -c -f " BLD_PRODUCT ".conf";
} else {
serviceCmdLine = argp;
}
break;
case 'g':
serviceOp = MPR_GO_SERVICE;
break;
case 's':
serviceOp = MPR_STOP_SERVICE;
break;
case 'u':
serviceOp = MPR_UNINSTALL_SERVICE;
break;
#endif
default:
errflg++;
break;
}
}
if (errflg) {
printUsage(program);
return MPR_ERR_BAD_SYNTAX;
}
mp = new Mpr(program);
mp->setAppName(BLD_PRODUCT);
mp->setAppTitle(BLD_NAME);
mp->setHeadless(isService || background);
#if BLD_HOST_UNIX || VXWORKS
initSignals();
#endif
if (kill) {
mp->killMpr();
delete mp;
exit(0);
}
if (outputVersion) {
printVersion();
delete mp;
exit(0);
}
//
// Create the top level HTTP service and default HTTP server
//
http = new MaHttp();
server = new MaServer(http, "default");
setupFileSystem();
setLogging(logSpec);
//
// Confirm the location of the server root
//
if (locateServerRoot(serverRoot) < 0) {
mprError(MPR_L, MPR_USER, "Can't start server, exiting.");
exit(2);
}
if (securityChecks(cmdLine->getArgv()[0]) < 0) {
exit(3);
}
#if WIN
#if BLD_FEATURE_RUN_AS_SERVICE
if (serviceOp) {
windowsServiceOps();
delete mp;
return 0;
}
#endif
if (windowsInit() < 0) {
delete mp;
return MPR_ERR_CANT_INITIALIZE;
}
#endif
//
// Start the MPR. This starts Timer, Socket and Pool services
//
if (mp->start(MPR_KILLABLE) < 0) {
mprError(MPR_L, MPR_USER, "Can't start MPR for %s", mp->getAppTitle());
delete mp;
return MPR_ERR_CANT_INITIALIZE;
}
//
// Load the statically linked modules
//
maLoadStaticModules();
if (setupServer(http, poolThreads) < 0) {
mprError(MPR_L, MPR_USER, "Can't configure the server, exiting.");
exit(6);
}
#if BLD_FEATURE_CONFIG_SAVE
if (writeFile) {
server->saveConfig(writeFile);
mprLog(0, "Configuration saved to %s\nExiting ...\n", writeFile);
exit(0);
}
#endif
if (http->start() < 0) {
mprError(MPR_L, MPR_USER, "Can't start server, exiting.");
exit(7);
} else {
#if LINUX && BLD_FEATURE_RUN_AS_SERVICE
if (background && mp->makeDaemon(1) < 0) {
mprError(MPR_L, MPR_USER, "Could not run in the background");
} else
#endif
{
#if BLD_FEATURE_MULTITHREAD
mprLog(MPR_CONFIG,
"HTTP services are ready with %d pool threads\n",
http->getLimits()->maxThreads);
#else
mprLog(MPR_CONFIG,
"HTTP services are ready (single-threaded).\n");
#endif
}
mp->setHeadless(1);
eventLoop();
mprLog(MPR_WARN, "Stopping HTTP services.\n");
http->stop();
}
#if WIN
if (trayIcon > 0) {
closeTrayIcon();
}
if (trayTimer) {
trayTimer->stop(MPR_TIMEOUT_STOP);
trayTimer->dispose();
trayTimer = 0;
}
#endif
mprLog(MPR_WARN, "Stopping MPR services.\n");
mp->stop(0);
maUnloadStaticModules();
delete server;
delete http;
delete mp;
mprFree(ipAddr);
#if BLD_FEATURE_ROMFS
delete romFileSystem;
#endif
#if BLD_FEATURE_LOG
mprLog(MPR_WARN, "Closing log.\n");
if (logger) {
delete logger;
}
#if WIN
if (dialog) {
delete dialog;
}
#endif
#endif
return 0;
}
MAIN(ejscMain, int argc, char **argv, char **envp)
{
Mpr *mpr;
MprList *args;
Ejs *ejs;
EcCompiler *cp;
char *argp, *searchPath, *outputFile, *outputDir, *certFile, *name, *tok, *modules;
int nextArg, err, ejsFlags, ecFlags, bind, debug, doc, merge, modver;
int warnLevel, noout, parseOnly, tabWidth, optimizeLevel, strict;
/*
Initialize the Multithreaded Portable Runtime (MPR)
*/
mpr = mprCreate(argc, argv, 0);
app = mprAllocObj(App, manageApp);
mprAddRoot(app);
mprAddStandardSignals();
if (mprStart() < 0) {
mprLog("ejsc", 0, "Cannot start mpr services");
return EJS_ERR;
}
err = 0;
searchPath = 0;
strict = 0;
certFile = 0;
ecFlags = 0;
bind = 0;
debug = 0;
doc = 0;
merge = 0;
modver = 0;
noout = 0;
parseOnly = 0;
tabWidth = 4;
warnLevel = 1;
outputFile = 0;
outputDir = 0;
optimizeLevel = 9;
for (nextArg = 1; nextArg < argc; nextArg++) {
argp = argv[nextArg];
if (*argp != '-') {
break;
}
if (strcmp(argp, "--bind") == 0) {
bind = 1;
} else if (strcmp(argp, "--debug") == 0) {
debug = 1;
} else if (strcmp(argp, "--debugger") == 0 || strcmp(argp, "-D") == 0) {
mprSetDebugMode(1);
} else if (strcmp(argp, "--dir") == 0) {
/*
Set the output directory for modules
*/
if (nextArg >= argc) {
err++;
} else {
outputDir = argv[++nextArg];
}
} else if (strcmp(argp, "--doc") == 0) {
doc = 1;
} else if (strcmp(argp, "--log") == 0) {
/*
Undocumented logging switch
*/
if (nextArg >= argc) {
err++;
} else {
mprStartLogging(argv[++nextArg], 0);
mprSetCmdlineLogging(1);
}
} else if (strcmp(argp, "--merge") == 0) {
merge = 1;
} else if (strcmp(argp, "--modver") == 0) {
if (nextArg >= argc) {
err++;
} else {
modver = ejsParseModuleVersion(argv[++nextArg]);
}
} else if (strcmp(argp, "--nobind") == 0) {
bind = 0;
} else if (strcmp(argp, "--noout") == 0) {
noout = 1;
} else if (strcmp(argp, "--standard") == 0) {
strict = 0;
} else if (strcmp(argp, "--strict") == 0) {
strict = 1;
} else if (strcmp(argp, "--optimize") == 0) {
if (nextArg >= argc) {
err++;
} else {
optimizeLevel = atoi(argv[++nextArg]);
}
} else if (strcmp(argp, "--out") == 0) {
/*
Create a single output module file containing all modules
*/
if (nextArg >= argc) {
err++;
} else {
outputFile = argv[++nextArg];
}
} else if (strcmp(argp, "--parse") == 0) {
parseOnly = 1;
} else if (strcmp(argp, "--search") == 0 || strcmp(argp, "--searchpath") == 0) {
if (nextArg >= argc) {
err++;
} else {
searchPath = argv[++nextArg];
}
} else if (strcmp(argp, "--require") == 0) {
if (nextArg >= argc) {
err++;
} else {
if (app->modules == 0) {
app->modules = mprCreateList(-1, 0);
}
modules = sclone(argv[++nextArg]);
/* Fix for Xcode and Visual Studio */
if (modules[0] == ' ' || scmp(modules, "null") == 0) {
modules[0] = '\0';
}
name = stok(modules, " \t,", &tok);
while (name != NULL) {
require(name);
name = stok(NULL, " \t", &tok);
}
}
} else if (strcmp(argp, "--sign") == 0) {
if (nextArg >= argc) {
err++;
} else {
certFile = argv[++nextArg];
}
#if FUTURE
} else if (strcmp(argp, "--strip") == 0) {
strip = 1;
#endif
} else if (strcmp(argp, "--tabWidth") == 0) {
if (nextArg >= argc) {
err++;
} else {
tabWidth = atoi(argv[++nextArg]);
}
} else if (strcmp(argp, "--version") == 0 || strcmp(argp, "-V") == 0) {
mprPrintf("%s\n", ME_TITLE, EJS_VERSION);
return 0;
} else if (strcmp(argp, "--warn") == 0) {
if (nextArg >= argc) {
err++;
} else {
warnLevel = atoi(argv[++nextArg]);
}
} else if (strcmp(argp, "--web") == 0) {
if (app->modules == 0) {
app->modules = mprCreateList(-1, 0);
}
require("ejs");
require("ejs.unix");
require("ejs.db");
// TODO - decouple and remove this
require("ejs.db.mapper");
require("ejs.web");
} else {
err++;
break;
}
}
if (noout || merge) {
bind = 1;
}
if (outputFile && noout) {
mprEprintf("Cannot use --out and --noout\n");
err++;
}
if (argc == nextArg) {
err++;
}
if (err) {
/*
Usage Examples:
ejsc Person.es User.es Customer.es
ejsc --out group.mod Person.es User.es Customer.es
ejsc --out group.mod Person.es User.es Customer.es
NOTE: bind is deliberately not documented and is for internal use only.
*/
mprEprintf("Usage: %s [options] files...\n"
" Ejscript compiler options:\n"
" --debug # Include symbolic debugging information in output\n"
" --doc # Include documentation strings in output\n"
" --dir directory # Set the output directory for modules (default: \".\")\n"
" --merge # Merge dependent input modules into the output\n"
" --modver version # Set the default module version\n"
" --noout # Do not generate any output\n"
" --optimize level # Set optimization level (0-9)\n"
" --out filename # Name a single output module (default: \"default.mod\")\n"
" --parse # Just parse source. No output\n"
" --require 'module ...' # List of required modules to pre-load\n"
" --search ejsPath # Module search path\n"
" --standard # Default compilation mode to standard (default)\n"
" --strict # Default compilation mode to strict\n"
#if FUTURE
" --sign certFile # Sign the module file (not implemented) \n"
" --strip # Strip all symbolic names (Cannot import)\n"
" --tabwidth # Tab width for '^' error reporting\n"
#endif
" --version # Emit the compiler version information\n"
" --warn level # Set the warning message level (0-9)\n\n",
mpr->name);
return -1;
}
ejsFlags = EJS_FLAG_NO_INIT;
if (doc) {
ejsFlags |= EJS_FLAG_DOC;
}
if ((ejs = ejsCreateVM(0, 0, ejsFlags)) == 0) {
return MPR_ERR_MEMORY;
}
#if UNUSED
mprRunDispatcher(ejs->dispatcher);
#endif
if (ejsLoadModules(ejs, searchPath, app->modules) < 0) {
return MPR_ERR_CANT_READ;
}
app->ejs = ejs;
ecFlags |= (debug) ? EC_FLAGS_DEBUG: 0;
ecFlags |= (merge) ? EC_FLAGS_MERGE: 0;
ecFlags |= (bind) ? EC_FLAGS_BIND: 0;
ecFlags |= (noout) ? EC_FLAGS_NO_OUT: 0;
ecFlags |= (parseOnly) ? EC_FLAGS_PARSE_ONLY: 0;
ecFlags |= (doc) ? EC_FLAGS_DOC: 0;
cp = app->compiler = ecCreateCompiler(ejs, ecFlags);
if (cp == 0) {
return MPR_ERR_MEMORY;
}
cp->require = app->modules;
cp->modver = modver;
ecSetOptimizeLevel(cp, optimizeLevel);
ecSetWarnLevel(cp, warnLevel);
ecSetStrictMode(cp, strict);
ecSetTabWidth(cp, tabWidth);
ecSetOutputDir(cp, outputDir);
ecSetOutputFile(cp, outputFile);
ecSetCertFile(cp, certFile);
if (nextArg < argc) {
/*
Compile the source files supplied on the command line. This will compile in-memory and
optionally also save to module files.
*/
if ((args = expandWild(ejs, argc - nextArg, &argv[nextArg])) == 0) {
err++;
} else if (ecCompile(cp, args->length, (char**) args->items) < 0) {
err++;
}
if (cp->warningCount > 0 || cp->errorCount > 0) {
mprLog("ejs", 0, "%s\n", cp->errorMsg);
}
}
if (cp->errorCount > 0) {
err++;
}
app->ejs = 0;
app->compiler = 0;
ejsDestroy(ejs);
mprDestroy();
return err;
}
MAIN(appweb, int argc, char **argv, char **envp)
{
Mpr *mpr;
cchar *ipAddrPort, *argp, *jail;
char *ip, *logSpec;
int argind, port, status, verbose;
ipAddrPort = 0;
ip = 0;
jail = 0;
port = -1;
verbose = 0;
logSpec = 0;
argv[0] = BLD_APPWEB_PATH;
if ((mpr = mprCreate(argc, argv, MPR_USER_EVENTS_THREAD)) == NULL) {
exit(1);
}
mprSetAppName(BLD_PRODUCT, BLD_NAME, BLD_VERSION);
if ((app = mprAllocObj(App, manageApp)) == NULL) {
exit(2);
}
mprAddRoot(app);
mprAddStandardSignals();
#if BLD_FEATURE_ROMFS
extern MprRomInode romFiles[];
mprSetRomFileSystem(romFiles);
#endif
app->mpr = mpr;
app->workers = -1;
app->configFile = BLD_CONFIG_FILE;
app->home = BLD_SERVER_ROOT;
app->documents = app->home;
argc = mpr->argc;
argv = mpr->argv;
for (argind = 1; argind < argc; argind++) {
argp = argv[argind];
if (*argp != '-') {
break;
}
if (smatch(argp, "--config") || smatch(argp, "--conf")) {
if (argind >= argc) {
usageError();
}
app->configFile = sclone(argv[++argind]);
#if BLD_UNIX_LIKE
} else if (smatch(argp, "--chroot")) {
if (argind >= argc) {
usageError();
}
jail = mprGetAbsPath(argv[++argind]);
#endif
} else if (smatch(argp, "--debugger") || smatch(argp, "-D")) {
mprSetDebugMode(1);
} else if (smatch(argp, "--exe")) {
if (argind >= argc) {
usageError();
}
mpr->argv[0] = mprGetAbsPath(argv[++argind]);
mprSetAppPath(mpr->argv[0]);
mprSetModuleSearchPath(NULL);
} else if (smatch(argp, "--home")) {
if (argind >= argc) {
usageError();
}
app->home = mprGetAbsPath(argv[++argind]);
#if UNUSED && KEEP
if (chdir(app->home) < 0) {
mprError("%s: Can't change directory to %s", mprGetAppName(), app->home);
exit(4);
}
#endif
} else if (smatch(argp, "--log") || smatch(argp, "-l")) {
if (argind >= argc) {
usageError();
}
logSpec = argv[++argind];
} else if (smatch(argp, "--name") || smatch(argp, "-n")) {
if (argind >= argc) {
usageError();
}
mprSetAppName(argv[++argind], 0, 0);
} else if (smatch(argp, "--threads")) {
if (argind >= argc) {
usageError();
}
app->workers = atoi(argv[++argind]);
} else if (smatch(argp, "--verbose") || smatch(argp, "-v")) {
verbose++;
} else if (smatch(argp, "--version") || smatch(argp, "-V")) {
mprPrintf("%s %s-%s\n", mprGetAppTitle(), BLD_VERSION, BLD_NUMBER);
exit(0);
} else {
mprError("Unknown switch \"%s\"", argp);
usageError();
exit(5);
}
}
if (logSpec) {
mprStartLogging(logSpec, 1);
mprSetCmdlineLogging(1);
} else if (verbose) {
mprStartLogging(sfmt("stderr:%d", verbose + 1), 1);
mprSetCmdlineLogging(1);
}
if (mprStart() < 0) {
mprUserError("Can't start MPR for %s", mprGetAppName());
mprDestroy(MPR_EXIT_DEFAULT);
return MPR_ERR_CANT_INITIALIZE;
}
if (checkEnvironment(argv[0]) < 0) {
exit(6);
}
if (argc > argind) {
if (argc > (argind + 2)) {
usageError();
}
ipAddrPort = argv[argind++];
if (argc > argind) {
app->documents = sclone(argv[argind++]);
}
mprParseSocketAddress(ipAddrPort, &ip, &port, HTTP_DEFAULT_PORT);
} else if (findConfigFile() < 0) {
exit(7);
}
if (jail && changeRoot(jail) < 0) {
exit(8);
}
if (initialize(ip, port) < 0) {
return MPR_ERR_CANT_INITIALIZE;
}
if (maStartAppweb(app->appweb) < 0) {
mprUserError("Can't start HTTP service, exiting.");
exit(9);
}
/*
Service I/O events until instructed to exit
*/
while (!mprIsStopping()) {
mprServiceEvents(-1, 0);
}
status = mprGetExitStatus();
mprLog(1, "Stopping Appweb ...");
maStopAppweb(app->appweb);
mprDestroy(MPR_EXIT_DEFAULT);
return status;
}
static int parseArgs(int argc, char **argv)
{
char *argp, *key, *value;
int i, setWorkers, nextArg, ssl;
setWorkers = 0;
ssl = 0;
for (nextArg = 1; nextArg < argc; nextArg++) {
argp = argv[nextArg];
if (*argp != '-') {
break;
}
if (smatch(argp, "--auth")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->authType = slower(argv[++nextArg]);
}
} else if (smatch(argp, "--benchmark") || smatch(argp, "-b")) {
app->benchmark++;
} else if (smatch(argp, "--ca")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->ca = sclone(argv[++nextArg]);
if (!mprPathExists(app->ca, R_OK)) {
mprError("Cannot find ca file %s", app->ca);
return MPR_ERR_BAD_ARGS;
}
}
ssl = 1;
} else if (smatch(argp, "--cert")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->cert = sclone(argv[++nextArg]);
if (!mprPathExists(app->cert, R_OK)) {
mprError("Cannot find cert file %s", app->cert);
return MPR_ERR_BAD_ARGS;
}
}
ssl = 1;
} else if (smatch(argp, "--chunk")) {
if (nextArg >= argc) {
return showUsage();
} else {
value = argv[++nextArg];
app->chunkSize = atoi(value);
if (app->chunkSize < 0) {
mprError("Bad chunksize %d", app->chunkSize);
return MPR_ERR_BAD_ARGS;
}
}
} else if (smatch(argp, "--ciphers")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->ciphers = sclone(argv[++nextArg]);
}
ssl = 1;
} else if (smatch(argp, "--continue") || smatch(argp, "-c")) {
app->continueOnErrors++;
} else if (smatch(argp, "--cookie")) {
if (nextArg >= argc) {
return showUsage();
} else {
mprAddItem(app->headers, mprCreateKeyPair("Cookie", argv[++nextArg], 0));
}
} else if (smatch(argp, "--data")) {
if (nextArg >= argc) {
return showUsage();
} else {
if (app->bodyData == 0) {
app->bodyData = mprCreateBuf(-1, -1);
}
mprPutStringToBuf(app->bodyData, argv[++nextArg]);
}
} else if (smatch(argp, "--debugger") || smatch(argp, "-D")) {
mprSetDebugMode(1);
app->retries = 0;
app->timeout = MAXINT;
} else if (smatch(argp, "--delete")) {
app->method = "DELETE";
} else if (smatch(argp, "--form") || smatch(argp, "-f")) {
if (nextArg >= argc) {
return showUsage();
} else {
if (app->formData == 0) {
app->formData = mprCreateList(-1, 0);
}
addFormVars(argv[++nextArg]);
}
} else if (smatch(argp, "--header")) {
if (nextArg >= argc) {
return showUsage();
} else {
key = argv[++nextArg];
if ((value = strchr(key, ':')) == 0) {
mprError("Bad header format. Must be \"key: value\"");
return MPR_ERR_BAD_ARGS;
}
*value++ = '\0';
while (isspace((uchar) *value)) {
value++;
}
mprAddItem(app->headers, mprCreateKeyPair(key, value, 0));
}
} else if (smatch(argp, "--host")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->host = argv[++nextArg];
if (*app->host == ':') {
app->host = &app->host[1];
}
if (isPort(app->host)) {
app->host = sfmt("http://127.0.0.1:%s", app->host);
} else {
app->host = sclone(app->host);
}
}
} else if (smatch(argp, "--iterations") || smatch(argp, "-i")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->iterations = atoi(argv[++nextArg]);
}
} else if (smatch(argp, "--key")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->key = sclone(argv[++nextArg]);
if (!mprPathExists(app->key, R_OK)) {
mprError("Cannot find key file %s", app->key);
return MPR_ERR_BAD_ARGS;
}
}
ssl = 1;
} else if (smatch(argp, "--log") || smatch(argp, "-l")) {
if (nextArg >= argc) {
return showUsage();
} else {
mprStartLogging(argv[++nextArg], 0);
}
} else if (smatch(argp, "--method") || smatch(argp, "-m")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->method = argv[++nextArg];
}
} else if (smatch(argp, "--out") || smatch(argp, "-o")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->outFilename = sclone(argv[++nextArg]);
}
} else if (smatch(argp, "--noout") || smatch(argp, "-n") ||
smatch(argp, "--quiet") || smatch(argp, "-q")) {
app->noout++;
} else if (smatch(argp, "--nofollow")) {
app->nofollow++;
} else if (smatch(argp, "--password") || smatch(argp, "-p")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->password = sclone(argv[++nextArg]);
}
} else if (smatch(argp, "--post")) {
app->method = "POST";
} else if (smatch(argp, "--printable")) {
app->printable++;
} else if (smatch(argp, "--protocol")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->protocol = supper(argv[++nextArg]);
}
} else if (smatch(argp, "--provider")) {
/* Undocumented SSL provider selection */
if (nextArg >= argc) {
return showUsage();
} else {
app->provider = sclone(argv[++nextArg]);
}
ssl = 1;
} else if (smatch(argp, "--put")) {
app->method = "PUT";
} else if (smatch(argp, "--range")) {
if (nextArg >= argc) {
return showUsage();
} else {
if (app->ranges == 0) {
app->ranges = sfmt("bytes=%s", argv[++nextArg]);
} else {
app->ranges = srejoin(app->ranges, ",", argv[++nextArg], NULL);
}
}
} else if (smatch(argp, "--retries") || smatch(argp, "-r")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->retries = atoi(argv[++nextArg]);
}
} else if (smatch(argp, "--self")) {
/* Undocumented. Allow self-signed certs. Users should just not set --verify */
app->verifyIssuer = 0;
ssl = 1;
} else if (smatch(argp, "--sequence")) {
app->sequence++;
} else if (smatch(argp, "--showHeaders") || smatch(argp, "--show") || smatch(argp, "-s")) {
app->showHeaders++;
} else if (smatch(argp, "--showStatus") || smatch(argp, "--showCode")) {
app->showStatus++;
} else if (smatch(argp, "--single") || smatch(argp, "-s")) {
app->singleStep++;
} else if (smatch(argp, "--text")) {
app->text++;
} else if (smatch(argp, "--threads") || smatch(argp, "-t")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->loadThreads = atoi(argv[++nextArg]);
}
} else if (smatch(argp, "--timeout")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->timeout = atoi(argv[++nextArg]) * MPR_TICKS_PER_SEC;
}
} else if (smatch(argp, "--upload") || smatch(argp, "-u")) {
app->upload++;
} else if (smatch(argp, "--user") || smatch(argp, "--username")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->username = argv[++nextArg];
}
// DEPRECATE validate. Preserve verify.
} else if (smatch(argp, "--validate") || smatch(argp, "--verify")) {
app->verifyPeer = 1;
ssl = 1;
} else if (smatch(argp, "--verbose") || smatch(argp, "-v")) {
app->verbose++;
} else if (smatch(argp, "--version") || smatch(argp, "-V")) {
mprEprintf("%s %s\n"
"Copyright (C) Embedthis Software 2003-2013\n"
"Copyright (C) Michael O'Brien 2003-2013\n",
BIT_TITLE, BIT_VERSION);
exit(0);
} else if (smatch(argp, "--workerTheads") || smatch(argp, "-w")) {
if (nextArg >= argc) {
return showUsage();
} else {
app->workers = atoi(argv[++nextArg]);
}
setWorkers++;
} else if (smatch(argp, "--zero")) {
app->zeroOnErrors++;
} else if (smatch(argp, "--")) {
nextArg++;
break;
} else if (smatch(argp, "-")) {
break;
} else {
return showUsage();
}
}
if (argc == nextArg) {
return showUsage();
}
app->nextArg = nextArg;
argc = argc - nextArg;
argv = &argv[nextArg];
app->target = argv[argc - 1];
if (--argc > 0) {
/*
Files present on command line
*/
app->files = mprCreateList(argc, MPR_LIST_STATIC_VALUES);
for (i = 0; i < argc; i++) {
mprAddItem(app->files, argv[i]);
}
}
if (!setWorkers) {
app->workers = app->loadThreads + 2;
}
if (app->method == 0) {
if (app->bodyData || app->formData || app->upload) {
app->method = "POST";
} else if (app->files) {
app->method = "PUT";
} else {
app->method = "GET";
}
}
#if BIT_PACK_SSL
{
HttpUri *uri = httpCreateUri(app->target, 0);
if (uri->secure || ssl) {
app->ssl = mprCreateSsl(0);
if (app->provider) {
mprSetSslProvider(app->ssl, app->provider);
}
if (app->cert) {
if (!app->key) {
mprError("Must specify key file");
return 0;
}
mprSetSslCertFile(app->ssl, app->cert);
mprSetSslKeyFile(app->ssl, app->key);
}
if (app->ca) {
mprLog(4, "Using CA: \"%s\"", app->ca);
mprSetSslCaFile(app->ssl, app->ca);
}
if (app->verifyIssuer == -1) {
app->verifyIssuer = app->verifyPeer ? 1 : 0;
}
mprVerifySslPeer(app->ssl, app->verifyPeer);
mprVerifySslIssuer(app->ssl, app->verifyIssuer);
if (app->ciphers) {
mprSetSslCiphers(app->ssl, app->ciphers);
}
} else {
mprVerifySslPeer(NULL, 0);
}
}
#else
/* Suppress comp warning */
mprNop(&ssl);
#endif
return 0;
}
MAIN(ejsmodMain, int argc, char **argv, char **envp)
{
Mpr *mpr;
EjsMod *mp;
Ejs *ejs;
MprList *requiredModules;
char *argp, *searchPath, *output, *modules, *name, *tok;
int nextArg, err, flags;
err = 0;
output = searchPath = 0;
requiredModules = 0;
/*
Initialze the Multithreaded Portable Runtime (MPR)
*/
mpr = mprCreate(argc, argv, 0);
mprSetAppName(argv[0], 0, 0);
/*
Allocate the primary control structure
*/
if ((mp = mprAllocObj(EjsMod, manageMod)) == NULL) {
return MPR_ERR_MEMORY;
}
mprAddRoot(mp);
mp->lstRecords = mprCreateList(0, 0);
mp->blocks = mprCreateList(0, 0);
mp->docDir = sclone(".");
mp->outputDir = sclone(".");
for (nextArg = 1; nextArg < argc; nextArg++) {
argp = argv[nextArg];
if (*argp != '-') {
break;
}
if (strcmp(argp, "--cslots") == 0) {
mp->cslots = 1;
mp->genSlots = 1;
} else if (strcmp(argp, "--debugger") == 0 || strcmp(argp, "-D") == 0) {
mprSetDebugMode(1);
} else if (strcmp(argp, "--depends") == 0) {
mp->depends = 1;
} else if (strcmp(argp, "--dir") == 0) {
if (nextArg >= argc) {
err++;
} else {
mp->outputDir = sclone(argv[++nextArg]);
}
} else if (strcmp(argp, "--error") == 0) {
/*
Undocumented switch
*/
mp->exitOnError++;
mp->warnOnError++;
} else if (strcmp(argp, "--html") == 0) {
if (nextArg >= argc) {
err++;
} else {
mp->docDir = sclone(argv[++nextArg]);
mp->html = 1;
}
} else if (strcmp(argp, "--listing") == 0) {
mp->listing = 1;
mp->showAsm = 1;
} else if (strcmp(argp, "--log") == 0) {
/*
Undocumented switch
*/
if (nextArg >= argc) {
err++;
} else {
mprStartLogging(argv[++nextArg], 0);
mprSetCmdlineLogging(1);
}
} else if (strcmp(argp, "--out") == 0) {
if (nextArg >= argc) {
err++;
} else {
output = argv[++nextArg];
mp->cslots = 1;
mp->genSlots = 1;
}
} else if (strcmp(argp, "--search") == 0 || strcmp(argp, "--searchpath") == 0) {
if (nextArg >= argc) {
err++;
} else {
searchPath = argv[++nextArg];
}
} else if (strcmp(argp, "--version") == 0 || strcmp(argp, "-V") == 0) {
mprPrintfError("%s %s-%s\n", BIT_NAME, BIT_VERSION, BIT_NUMBER);
return 0;
} else if (strcmp(argp, "--require") == 0) {
if (nextArg >= argc) {
err++;
} else {
if (requiredModules == 0) {
requiredModules = mprCreateList(-1, 0);
}
modules = sclone(argv[++nextArg]);
/* Fix for Xcode and Visual Studio */
if (modules[0] == ' ' || scmp(modules, "null") == 0) {
modules[0] = '\0';
}
name = stok(modules, " \t", &tok);
while (name != NULL) {
require(requiredModules, name);
name = stok(NULL, " \t", &tok);
}
}
} else if (strcmp(argp, "--warn") == 0) {
mp->warnOnError++;
} else if (strcmp(argp, "--xml") == 0) {
mp->xml = 1;
} else {
err++;
break;
}
}
if (argc == nextArg) {
err++;
}
if (mp->genSlots == 0 && mp->listing == 0 && mp->html == 0 && mp->xml == 0 && mp->depends == 0) {
mp->listing = 1;
}
if (mp->depends && requiredModules == 0) {
requiredModules = mprCreateList(-1, 0);
}
if (err) {
/*
Examples:
ejsmod file.mod # Defaults to --listing
ejsmod --listing embedthis.mod
ejsmod --out slots.h embedthis.mod
*/
mprPrintfError("Usage: %s [options] modules ...\n"
" Ejscript module manager options:\n"
" --cslots # Generate a C slot definitions file\n"
" --html dir # Generate HTML documentation to the specified directory\n"
" --listing # Create assembler listing files (default)\n"
" --out file # Output file for all C slots (implies --cslots)\n"
" --require \"modules\" # List of modules to preload\n"
" --search ejsPath # Module file search path\n"
" --version # Emit the program version information\n"
" --warn # Warn about undocumented methods or parameters in doc\n\n", mpr->name);
return -1;
}
/*
Need an interpreter to load modules
*/
flags = EJS_FLAG_NO_INIT;
if (mp->html || mp->xml) {
flags |= EJS_FLAG_DOC;
}
if ((ejs = ejsCreateVM(0, 0, flags)) == 0) {
return MPR_ERR_MEMORY;
}
if (ejsLoadModules(ejs, searchPath, requiredModules) < 0) {
return MPR_ERR_CANT_READ;
}
mp->ejs = ejs;
if (nextArg < argc) {
if (process(mp, output, argc - nextArg, &argv[nextArg]) < 0) {
err++;
}
}
if (mp->errorCount > 0) {
err = -1;
}
mprRemoveRoot(mp);
ejsDestroyVM(ejs);
mprDestroy(MPR_EXIT_DEFAULT);
return err;
}
MAIN(appweb, int argc, char **argv, char **envp)
{
Mpr *mpr;
cchar *argp, *jail;
char *logSpec;
int argind, status, verbose;
jail = 0;
verbose = 0;
logSpec = 0;
if ((mpr = mprCreate(argc, argv, MPR_USER_EVENTS_THREAD)) == NULL) {
exit(1);
}
mprSetAppName(BIT_PRODUCT, BIT_TITLE, BIT_VERSION);
/*
Allocate the top level application object. ManageApp is the GC manager function and is called
by the GC to mark references in the app object.
*/
if ((app = mprAllocObj(AppwebApp, manageApp)) == NULL) {
exit(2);
}
mprAddRoot(app);
mprAddStandardSignals();
app->mpr = mpr;
app->configFile = sclone("appweb.conf");
app->home = mprGetCurrentPath();
app->documents = app->home;
argc = mpr->argc;
argv = (char**) mpr->argv;
for (argind = 1; argind < argc; argind++) {
argp = argv[argind];
if (*argp != '-') {
break;
}
if (smatch(argp, "--config") || smatch(argp, "--conf")) {
if (argind >= argc) {
usageError();
}
app->configFile = sclone(argv[++argind]);
#if BIT_UNIX_LIKE
} else if (smatch(argp, "--chroot")) {
if (argind >= argc) {
usageError();
}
jail = mprGetAbsPath(argv[++argind]);
#endif
} else if (smatch(argp, "--debugger") || smatch(argp, "-D")) {
mprSetDebugMode(1);
} else if (smatch(argp, "--exe")) {
if (argind >= argc) {
usageError();
}
mpr->argv[0] = mprGetAbsPath(argv[++argind]);
mprSetAppPath(mpr->argv[0]);
mprSetModuleSearchPath(NULL);
} else if (smatch(argp, "--home")) {
if (argind >= argc) {
usageError();
}
app->home = mprGetAbsPath(argv[++argind]);
if (chdir(app->home) < 0) {
mprError("%s: Cannot change directory to %s", mprGetAppName(), app->home);
exit(4);
}
} else if (smatch(argp, "--log") || smatch(argp, "-l")) {
if (argind >= argc) {
usageError();
}
logSpec = argv[++argind];
} else if (smatch(argp, "--name") || smatch(argp, "-n")) {
if (argind >= argc) {
usageError();
}
mprSetAppName(argv[++argind], 0, 0);
} else if (smatch(argp, "--verbose") || smatch(argp, "-v")) {
verbose++;
} else if (smatch(argp, "--version") || smatch(argp, "-V")) {
mprPrintf("%s-%s\n", BIT_VERSION, BIT_BUILD_NUMBER);
exit(0);
} else {
if (!smatch(argp, "?")) {
mprError("Unknown switch \"%s\"", argp);
}
usageError();
exit(5);
}
}
if (logSpec) {
mprStartLogging(logSpec, 1);
mprSetCmdlineLogging(1);
} else if (verbose) {
mprStartLogging(sfmt("stderr:%d", verbose + 1), 1);
mprSetCmdlineLogging(1);
}
/*
Start the multithreaded portable runtime (MPR)
*/
if (mprStart() < 0) {
mprError("Cannot start MPR for %s", mprGetAppName());
mprDestroy(MPR_EXIT_DEFAULT);
return MPR_ERR_CANT_INITIALIZE;
}
if (checkEnvironment(argv[0]) < 0) {
exit(6);
}
if (findAppwebConf() < 0) {
exit(7);
}
if (jail && changeRoot(jail) < 0) {
exit(8);
}
/*
Open the sockets to listen on
*/
if (createEndpoints(argc - argind, &argv[argind]) < 0) {
return MPR_ERR_CANT_INITIALIZE;
}
/*
Start HTTP services
*/
if (maStartAppweb(app->appweb) < 0) {
mprError("Cannot start HTTP service, exiting.");
exit(9);
}
/*
Service I/O events until instructed to exit
*/
while (!mprIsStopping()) {
mprServiceEvents(-1, 0);
}
status = mprGetExitStatus();
mprLog(1, "Stopping Appweb ...");
maStopAppweb(app->appweb);
mprDestroy(MPR_EXIT_DEFAULT);
return status;
}
MAIN(ejsMain, int argc, char **argv, char **envp)
{
Mpr *mpr;
Ejs *ejs;
EcCompiler *ec;
char *argp, *searchPath, *path, *homeDir;
int nextArg, err, flags;
/*
Initialize Multithreaded Portable Runtime (MPR)
*/
mpr = mprCreate(argc, argv, 0);
app = mprAllocObj(App, manageApp);
mprAddRoot(app);
mprAddStandardSignals();
if (mprStart(mpr) < 0) {
mprError("Cannot start mpr services");
return EJS_ERR;
}
err = 0;
searchPath = 0;
argc = mpr->argc;
argv = (char**) mpr->argv;
for (nextArg = 1; nextArg < argc; nextArg++) {
argp = argv[nextArg];
if (*argp != '-') {
break;
}
if (smatch(argp, "--chdir") || smatch(argp, "--home") || smatch(argp, "-C")) {
if (nextArg >= argc) {
err++;
} else {
homeDir = argv[++nextArg];
if (chdir((char*) homeDir) < 0) {
mprError("Cannot change directory to %s", homeDir);
}
}
} else if (smatch(argp, "--debugger") || smatch(argp, "-D")) {
mprSetDebugMode(1);
} else if (smatch(argp, "--log")) {
if (nextArg >= argc) {
err++;
} else {
mprStartLogging(argv[++nextArg], 0);
mprSetCmdlineLogging(1);
}
} else if (smatch(argp, "--name")) {
/* Just ignore. Used to tag commands with a unique command line */
nextArg++;
} else if (smatch(argp, "--search") || smatch(argp, "--searchpath")) {
if (nextArg >= argc) {
err++;
} else {
searchPath = argv[++nextArg];
}
} else if (smatch(argp, "--verbose") || smatch(argp, "-v")) {
mprStartLogging("stderr:1", 0);
mprSetCmdlineLogging(1);
} else if (smatch(argp, "--version") || smatch(argp, "-V")) {
mprPrintf("%s-%s\n", BIT_VERSION, BIT_BUILD_NUMBER);
return 0;
} else {
/* Ignore */
}
}
path = mprJoinPath(mprGetAppDir(), mprGetPathBase(argv[0]));
path = mprReplacePathExt(path, ".es");
mprAddRoot(path);
argv[0] = path;
if ((ejs = ejsCreateVM(argc, (cchar **) &argv[0], 0)) == 0) {
return MPR_ERR_MEMORY;
}
app->ejs = ejs;
if (ejsLoadModules(ejs, searchPath, NULL) < 0) {
return MPR_ERR_CANT_READ;
}
mprLog(2, "Load script \"%s\"", path);
flags = EC_FLAGS_BIND | EC_FLAGS_DEBUG | EC_FLAGS_NO_OUT | EC_FLAGS_THROW;
if ((ec = ecCreateCompiler(ejs, flags)) == 0) {
return MPR_ERR_MEMORY;
}
mprAddRoot(ec);
ecSetOptimizeLevel(ec, 9);
ecSetWarnLevel(ec, 1);
if (ecCompile(ec, 1, (char**) &path) < 0) {
if (flags & EC_FLAGS_THROW) {
ejsThrowSyntaxError(ejs, "%s", ec->errorMsg ? ec->errorMsg : "Cannot parse script");
ejsReportError(ejs, "Error in script");
}
err = MPR_ERR;
} else {
mprRemoveRoot(ec);
if (ejsRunProgram(ejs, NULL, NULL) < 0) {
ejsReportError(ejs, "Error in script");
err = MPR_ERR;
}
}
if (!err) {
err = mpr->exitStatus;
}
app->ejs = 0;
mprTerminate(MPR_EXIT_DEFAULT, err);
ejsDestroyVM(ejs);
mprDestroy(MPR_EXIT_DEFAULT);
return err;
}
MAIN(http, int argc, char **argv, char **envp)
{
Mpr *mpr;
cchar *argp;
char *logSpec, *traceSpec;
int argind;
logSpec = 0;
traceSpec = 0;
if ((mpr = mprCreate(argc, argv, MPR_USER_EVENTS_THREAD)) == NULL) {
exit(1);
}
if ((app = mprAllocObj(App, manageApp)) == NULL) {
exit(2);
}
mprAddRoot(app);
mprAddStandardSignals();
if (httpCreate(HTTP_CLIENT_SIDE | HTTP_SERVER_SIDE) < 0) {
exit(2);
}
app->mpr = mpr;
app->home = mprGetCurrentPath();
argc = mpr->argc;
argv = (char**) mpr->argv;
for (argind = 1; argind < argc; argind++) {
argp = argv[argind];
if (*argp != '-') {
break;
}
if (smatch(argp, "--config") || smatch(argp, "--conf")) {
if (argind >= argc) {
usageError();
}
app->configFile = sclone(argv[++argind]);
} else if (smatch(argp, "--debugger") || smatch(argp, "-D")) {
mprSetDebugMode(1);
} else if (smatch(argp, "--log") || smatch(argp, "-l")) {
if (argind >= argc) {
usageError();
}
logSpec = argv[++argind];
} else if (smatch(argp, "--name") || smatch(argp, "-n")) {
if (argind >= argc) {
usageError();
}
mprSetAppName(argv[++argind], 0, 0);
} else if (smatch(argp, "--show") || smatch(argp, "-s")) {
app->show = 1;
} else if (smatch(argp, "--trace") || smatch(argp, "-t")) {
if (argind >= argc) {
usageError();
}
traceSpec = argv[++argind];
} else if (smatch(argp, "--verbose") || smatch(argp, "-v")) {
if (!logSpec) {
logSpec = sfmt("stderr:2");
}
if (!traceSpec) {
traceSpec = sfmt("stderr:2");
}
} else if (smatch(argp, "--version") || smatch(argp, "-V")) {
mprPrintf("%s\n", ME_VERSION);
exit(0);
} else if (*argp == '-' && isdigit((uchar) argp[1])) {
if (!logSpec) {
logSpec = sfmt("stderr:%d", (int) stoi(&argp[1]));
}
if (!traceSpec) {
traceSpec = sfmt("stderr:%d", (int) stoi(&argp[1]));
}
} else if (smatch(argp, "-?") || scontains(argp, "-help")) {
usageError();
exit(3);
} else if (*argp == '-') {
mprLog("error http", 0, "Unknown switch \"%s\"", argp);
usageError();
exit(4);
}
}
if (logSpec) {
mprStartLogging(logSpec, MPR_LOG_CMDLINE);
}
if (traceSpec) {
httpStartTracing(traceSpec);
}
if (mprStart() < 0) {
mprLog("error http", 0, "Cannot start MPR");
mprDestroy();
return MPR_ERR_CANT_INITIALIZE;
}
if (argc == argind && !app->configFile) {
if (findConfig() < 0) {
exit(5);
}
}
if (createEndpoints(argc - argind, &argv[argind]) < 0) {
return MPR_ERR_CANT_INITIALIZE;
}
if (httpStartEndpoints() < 0) {
mprLog("error http", 0, "Cannot start HTTP service, exiting.");
exit(6);
}
if (app->show) {
httpLogRoutes(0, 0);
}
mprServiceEvents(-1, 0);
mprLog("info http", 1, "Stopping http-server ...");
mprDestroy();
return mprGetExitStatus();
}
static bool parseArgs(Mpr *mpr, int argc, char **argv)
{
char *argp, *key, *value;
int i, setWorkers;
setWorkers = 0;
for (nextArg = 1; nextArg < argc; nextArg++) {
argp = argv[nextArg];
if (*argp != '-') {
break;
}
if (strcmp(argp, "--benchmark") == 0 || strcmp(argp, "-b") == 0) {
benchmark++;
} else if (strcmp(argp, "--chunk") == 0) {
if (nextArg >= argc) {
return 0;
} else {
chunkSize = argv[++nextArg];
i = atoi(chunkSize);
if (i < 0) {
mprError(mpr, "Bad chunksize %d", i);
return 0;
}
}
} else if (strcmp(argp, "--continue") == 0) {
continueOnErrors++;
} else if (strcmp(argp, "--cookie") == 0) {
if (nextArg >= argc) {
return 0;
} else {
mprAddItem(headers, mprCreateKeyPair(headers, "Cookie", argv[++nextArg]));
}
} else if (strcmp(argp, "--debug") == 0 || strcmp(argp, "-D") == 0 || strcmp(argp, "--debugger") == 0) {
mprSetDebugMode(mpr, 1);
retries = 0;
} else if (strcmp(argp, "--delete") == 0) {
method = "DELETE";
} else if (strcmp(argp, "--form") == 0 || strcmp(argp, "-f") == 0) {
if (nextArg >= argc) {
return 0;
} else {
if (formData == 0) {
formData = mprCreateList(mpr);
}
addFormVars(mpr, argv[++nextArg]);
}
} else if (strcmp(argp, "--header") == 0) {
if (nextArg >= argc) {
return 0;
} else {
key = argv[++nextArg];
if ((value = strchr(key, ':')) == 0) {
mprError(mpr, "Bad header format. Must be \"key: value\"");
return 0;
}
*value++ = '\0';
while (isspace((int) *value)) {
value++;
}
mprAddItem(headers, mprCreateKeyPair(headers, key, value));
}
} else if (strcmp(argp, "--host") == 0) {
if (nextArg >= argc) {
return 0;
} else {
host = argv[++nextArg];
}
} else if (strcmp(argp, "--http") == 0) {
if (nextArg >= argc) {
return 0;
} else {
httpVersion = atoi(argv[++nextArg]);
}
} else if (strcmp(argp, "--iterations") == 0 || strcmp(argp, "-i") == 0) {
if (nextArg >= argc) {
return 0;
} else {
iterations = atoi(argv[++nextArg]);
}
} else if (strcmp(argp, "--log") == 0 || strcmp(argp, "-l") == 0) {
if (nextArg >= argc) {
return 0;
} else {
startLogging(mpr, argv[++nextArg]);
}
} else if (strcmp(argp, "--method") == 0) {
if (nextArg >= argc) {
return 0;
} else {
method = argv[++nextArg];
}
} else if (strcmp(argp, "--noout") == 0 || strcmp(argp, "-n") == 0 ||
strcmp(argp, "--quiet") == 0 || strcmp(argp, "-q") == 0) {
noout++;
} else if (strcmp(argp, "--nofollow") == 0) {
nofollow++;
} else if (strcmp(argp, "--password") == 0 || strcmp(argp, "-p") == 0) {
if (nextArg >= argc) {
return 0;
} else {
password = argv[++nextArg];
}
} else if (strcmp(argp, "--post") == 0) {
method = "POST";
} else if (strcmp(argp, "--printable") == 0) {
printable++;
} else if (strcmp(argp, "--put") == 0) {
method = "PUT";
} else if (strcmp(argp, "--range") == 0) {
if (nextArg >= argc) {
return 0;
} else {
if (ranges == 0) {
ranges = mprAsprintf(mpr, -1, "bytes=%s", argv[++nextArg]);
} else {
ranges = mprStrcat(mpr, -1, ranges, ",", argv[++nextArg], NULL);
}
}
} else if (strcmp(argp, "--retries") == 0 || strcmp(argp, "-r") == 0) {
if (nextArg >= argc) {
return 0;
} else {
retries = atoi(argv[++nextArg]);
}
} else if (strcmp(argp, "--sequence") == 0) {
sequence++;
} else if (strcmp(argp, "--showCode") == 0 || strcmp(argp, "--showStatus") == 0) {
showCode++;
} else if (strcmp(argp, "--show") == 0 || strcmp(argp, "--showHeaders") == 0) {
showHeaders++;
} else if (strcmp(argp, "--single") == 0 || strcmp(argp, "-s") == 0) {
singleStep++;
} else if (strcmp(argp, "--threads") == 0 || strcmp(argp, "-t") == 0) {
if (nextArg >= argc) {
return 0;
} else {
loadThreads = atoi(argv[++nextArg]);
}
} else if (strcmp(argp, "--timeout") == 0) {
if (nextArg >= argc) {
return 0;
} else {
timeout = atoi(argv[++nextArg]) * 1000;
}
} else if (strcmp(argp, "--upload") == 0 || strcmp(argp, "-u") == 0) {
upload++;
} else if (strcmp(argp, "--user") == 0 || strcmp(argp, "--username") == 0) {
if (nextArg >= argc) {
return 0;
} else {
username = argv[++nextArg];
}
} else if (strcmp(argp, "--verbose") == 0 || strcmp(argp, "-v") == 0) {
verbose++;
} else if (strcmp(argp, "--workerTheads") == 0 || strcmp(argp, "-w") == 0) {
if (nextArg >= argc) {
return 0;
} else {
workers = atoi(argv[++nextArg]);
}
setWorkers++;
} else {
return 0;
break;
}
}
if (argc == nextArg) {
return 0;
}
argc = argc - nextArg;
argv = &argv[nextArg];
target = argv[argc - 1];
argc--;
if (argc > 0) {
/*
* Files present on command line
*/
fileData = mprCreateList(mpr);
for (i = 0; i < argc; i++) {
mprAddItem(fileData, argv[i]);
}
}
if (!setWorkers) {
workers = loadThreads + 2;
}
if (method == 0) {
if (upload || formData) {
method = "POST";
} else if (fileData) {
method = "PUT";
} else {
method = "GET";
}
}
return 1;
}
int MprTestSession::setupTests(MprTestResult *result, Mpr *mpr, int argc,
char *argv[], char *switches)
{
char switchBuf[80];
char *programName, *argp;
int errflg, c, i, l;
#if BLD_FEATURE_LOG
char *logSpec;
#endif
this->mpr = mpr;
programName = mprGetBaseName(argv[0]);
errflg = 0;
#if BLD_FEATURE_LOG
logSpec = "stdout:1";
logger = new MprLogToFile();
#endif
switchBuf[0] = '\0';
mprStrcat(switchBuf, sizeof(switchBuf), 0, "cDeg:i:l:n:msT:t:v?",
switches, (void*) 0);
MprCmdLine cmdLine(argc, argv, switchBuf);
while ((c = cmdLine.next(&argp)) != EOF) {
switch(c) {
case 'c':
result->setContinueOnFailures(1);
break;
case 'D':
mprSetDebugMode(1);
result->setDebugOnFailures(1);
break;
case 'e':
needEventsThread = 1;
break;
case 'g':
testGroups->parse(argp);
break;
case 'i':
iterations = atoi(argp);
break;
case 'l':
#if BLD_FEATURE_LOG
logSpec = argp;
#endif
break;
case 'n':
l = atoi(argp);
if (l == 0) {
sessionLevel = MPR_BASIC;
} else if (l == 1) {
sessionLevel = MPR_THOROUGH;
} else {
sessionLevel = MPR_DEDICATED;
}
break;
case 'm':
mprRequestMemStats(1);
break;
case 's':
result->setSingleStep(1);
break;
case 't':
i = atoi(argp);
if (i <= 0 || i > 100) {
mprFprintf(MPR_STDERR, "%s: Bad number of threads (0-100)\n",
programName);
exit(2);
}
#if BLD_FEATURE_MULTITHREAD
numThreads = i;
#endif
break;
case 'T':
#if BLD_FEATURE_MULTITHREAD
poolThreads = atoi(argp);
#endif
break;
case 'v':
verbose++;
break;
default:
//
// Ignore args we don't understand
//
break;
case '?':
errflg++;
break;
}
}
if (errflg) {
mprFprintf(MPR_STDERR,
"usage: %s [-cDemsv] [-g groups] [-i iterations] "
"[-l logSpec] [-n testLevel] [-T poolThreads] [-t threads]\n",
programName);
exit(2);
}
#if !BLD_FEATURE_MULTITHREAD
needEventsThread = 0;
#endif
#if BLD_FEATURE_LOG
mpr->addListener(logger);
mpr->setLogSpec(logSpec);
#endif
initSignals();
this->argc = argc;
this->argv = argv;
this->firstArg = cmdLine.firstArg();
#if BLD_FEATURE_MULTITHREAD
mpr->setMaxPoolThreads(poolThreads);
#endif
if (mpr->start(needEventsThread ? MPR_SERVICE_THREAD : 0) < 0) {
return MPR_ERR_CANT_INITIALIZE;
}
result->adjustThreadCount(numThreads);
result->setVerbosity(verbose);
if (result->getListenerCount() == 0) {
result->addListener(new MprTestListener("__default__"));
}
if (verbose) {
mprFprintf(MPR_STDOUT,
"Testing: iterations %d, threads %d, pool %d, service thread %d\n",
iterations, numThreads, poolThreads, needEventsThread);
}
//
// Use current session object for the main thread
//
sessions = (MprTestSession**) mprMalloc(sizeof(MprTestSession*) *
numThreads);
sessions[0] = this;
if (sessions[0]->initializeClasses(result) < 0) {
exit(3);
}
#if BLD_FEATURE_MULTITHREAD
//
// Now clone this session object for all other threads
//
for (i = 1; i < numThreads; i++) {
char tName[64];
sessions[i] = this->newSession();
sessions[i]->setResult(result);
sessions[i]->cloneSettings(this);
mprSprintf(tName, sizeof(tName), "test.%d", i);
}
#endif
return 0;
}
