/*-------------------------------------------------------------------------
* drawElements Utility Library
* ----------------------------
*
* Copyright 2014 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*//*!
* \file
* \brief Command line parser.
*//*--------------------------------------------------------------------*/
#include "deCommandLine.h"
#include "deMemPool.h"
#include "dePoolArray.h"
#include "deMemory.h"
#include "deString.h"
#include <string.h>
DE_DECLARE_POOL_ARRAY(CharPtrArray, char*);
enum
{
MAX_ARGS = 64
};
deCommandLine* deCommandLine_parse (const char* commandLine)
{
deMemPool* tmpPool = deMemPool_createRoot(DE_NULL, 0);
CharPtrArray* args = tmpPool ? CharPtrArray_create(tmpPool) : DE_NULL;
char* buf = DE_NULL;
char* outPtr;
int pos;
int argNdx;
char strChr;
if (!args)
{
if (tmpPool)
deMemPool_destroy(tmpPool);
return DE_NULL;
}
DE_ASSERT(commandLine);
/* Create buffer for args (no expansion can happen). */
buf = (char*)deCalloc((int)strlen(commandLine)+1);
pos = 0;
argNdx = 0;
outPtr = buf;
strChr = 0;
if (!buf || !CharPtrArray_pushBack(args, buf))
{
deMemPool_destroy(tmpPool);
return DE_NULL;
}
while (commandLine[pos] != 0)
{
char c = commandLine[pos++];
if (strChr != 0 && c == '\\')
{
/* Escape. */
c = commandLine[pos++];
switch (c)
{
case 'n': *outPtr++ = '\n'; break;
case 't': *outPtr++ = '\t'; break;
default: *outPtr++ = c; break;
}
}
else if (strChr != 0 && c == strChr)
{
/* String end. */
strChr = 0;
}
else if (strChr == 0 && (c == '"' || c == '\''))
{
/* String start. */
strChr = c;
}
else if (c == ' ' && strChr == 0)
{
/* Arg end. */
*outPtr++ = 0;
argNdx += 1;
if (!CharPtrArray_pushBack(args, outPtr))
{
deFree(buf);
deMemPool_destroy(tmpPool);
return DE_NULL;
}
}
else
*outPtr++ = c;
}
DE_ASSERT(commandLine[pos] == 0);
/* Terminate last arg. */
*outPtr = 0;
/* Create deCommandLine. */
{
deCommandLine* cmdLine = (deCommandLine*)deCalloc(sizeof(deCommandLine));
if (!cmdLine || !(cmdLine->args = (char**)deCalloc(sizeof(char*)*CharPtrArray_getNumElements(args))))
{
deFree(cmdLine);
deFree(buf);
deMemPool_destroy(tmpPool);
return DE_NULL;
}
cmdLine->numArgs = CharPtrArray_getNumElements(args);
cmdLine->argBuf = buf;
for (argNdx = 0; argNdx < cmdLine->numArgs; argNdx++)
cmdLine->args[argNdx] = CharPtrArray_get(args, argNdx);
deMemPool_destroy(tmpPool);
return cmdLine;
}
}
void deCommandLine_destroy (deCommandLine* cmdLine)
{
deFree(cmdLine->argBuf);
deFree(cmdLine);
}
#if defined(DE_DEBUG)
static void testParse (const char* cmdLine, const char* const* refArgs, int numArgs)
{
deCommandLine* parsedCmdLine = deCommandLine_parse(cmdLine);
int argNdx;
DE_TEST_ASSERT(parsedCmdLine);
DE_TEST_ASSERT(parsedCmdLine->numArgs == numArgs);
for (argNdx = 0; argNdx < numArgs; argNdx++)
DE_TEST_ASSERT(deStringEqual(parsedCmdLine->args[argNdx], refArgs[argNdx]));
deCommandLine_destroy(parsedCmdLine);
}
void deCommandLine_selfTest (void)
{
{
const char* cmdLine = "hello";
const char* ref[] = { "hello" };
testParse(cmdLine, ref, DE_LENGTH_OF_ARRAY(ref));
}
{
const char* cmdLine = "hello world";
const char* ref[] = { "hello", "world" };
testParse(cmdLine, ref, DE_LENGTH_OF_ARRAY(ref));
}
{
const char* cmdLine = "hello/world";
const char* ref[] = { "hello/world" };
testParse(cmdLine, ref, DE_LENGTH_OF_ARRAY(ref));
}
{
const char* cmdLine = "hello/world --help";
const char* ref[] = { "hello/world", "--help" };
testParse(cmdLine, ref, DE_LENGTH_OF_ARRAY(ref));
}
{
const char* cmdLine = "hello/world --help foo";
const char* ref[] = { "hello/world", "--help", "foo" };
testParse(cmdLine, ref, DE_LENGTH_OF_ARRAY(ref));
}
{
const char* cmdLine = "hello\\world --help foo";
const char* ref[] = { "hello\\world", "--help", "foo" };
testParse(cmdLine, ref, DE_LENGTH_OF_ARRAY(ref));
}
{
const char* cmdLine = "\"hello/worl d\" --help --foo=\"bar\" \"ba z\\\"\"";
const char* ref[] = { "hello/worl d", "--help", "--foo=bar", "ba z\"" };
testParse(cmdLine, ref, DE_LENGTH_OF_ARRAY(ref));
}
{
const char* cmdLine = "'hello/worl d' --help --foo='bar' 'ba z\\\''";
const char* ref[] = { "hello/worl d", "--help", "--foo=bar", "ba z'" };
testParse(cmdLine, ref, DE_LENGTH_OF_ARRAY(ref));
}
{
const char* cmdLine = "hello \"'world'\"";
const char* ref[] = { "hello", "'world'" };
testParse(cmdLine, ref, DE_LENGTH_OF_ARRAY(ref));
}
{
const char* cmdLine = "hello '\"world\"'";
const char* ref[] = { "hello", "\"world\"" };
testParse(cmdLine, ref, DE_LENGTH_OF_ARRAY(ref));
}
{
const char* cmdLine = "hello \"world\\n\"";
const char* ref[] = { "hello", "world\n" };
testParse(cmdLine, ref, DE_LENGTH_OF_ARRAY(ref));
}
}
#endif