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;
}
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;
}
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;
}
