/*
|
* Copyright (C) 2008 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.
|
*/
|
|
#define LOG_TAG "FrameworkListener"
|
|
#include <errno.h>
|
#include <stdlib.h>
|
#include <string.h>
|
#include <unistd.h>
|
|
#include <log/log.h>
|
#include <sysutils/FrameworkCommand.h>
|
#include <sysutils/FrameworkListener.h>
|
#include <sysutils/SocketClient.h>
|
|
static const int CMD_BUF_SIZE = 4096;
|
|
FrameworkListener::FrameworkListener(const char *socketName, bool withSeq) :
|
SocketListener(socketName, true, withSeq) {
|
init(socketName, withSeq);
|
}
|
|
FrameworkListener::FrameworkListener(const char *socketName) :
|
SocketListener(socketName, true, false) {
|
init(socketName, false);
|
}
|
|
FrameworkListener::FrameworkListener(int sock) :
|
SocketListener(sock, true) {
|
init(nullptr, false);
|
}
|
|
void FrameworkListener::init(const char* /*socketName*/, bool withSeq) {
|
errorRate = 0;
|
mCommandCount = 0;
|
mWithSeq = withSeq;
|
mSkipToNextNullByte = false;
|
}
|
|
bool FrameworkListener::onDataAvailable(SocketClient *c) {
|
char buffer[CMD_BUF_SIZE];
|
int len;
|
|
len = TEMP_FAILURE_RETRY(read(c->getSocket(), buffer, sizeof(buffer)));
|
if (len < 0) {
|
SLOGE("read() failed (%s)", strerror(errno));
|
return false;
|
} else if (!len) {
|
return false;
|
} else if (buffer[len-1] != '\0') {
|
SLOGW("String is not zero-terminated");
|
android_errorWriteLog(0x534e4554, "29831647");
|
c->sendMsg(500, "Command too large for buffer", false);
|
mSkipToNextNullByte = true;
|
return true;
|
}
|
|
int offset = 0;
|
int i;
|
|
for (i = 0; i < len; i++) {
|
if (buffer[i] == '\0') {
|
/* IMPORTANT: dispatchCommand() expects a zero-terminated string */
|
if (mSkipToNextNullByte) {
|
mSkipToNextNullByte = false;
|
} else {
|
dispatchCommand(c, buffer + offset);
|
}
|
offset = i + 1;
|
}
|
}
|
|
mSkipToNextNullByte = false;
|
return true;
|
}
|
|
void FrameworkListener::registerCmd(FrameworkCommand *cmd) {
|
mCommands.push_back(cmd);
|
}
|
|
void FrameworkListener::dispatchCommand(SocketClient *cli, char *data) {
|
int argc = 0;
|
char *argv[FrameworkListener::CMD_ARGS_MAX];
|
char tmp[CMD_BUF_SIZE];
|
char *p = data;
|
char *q = tmp;
|
char *qlimit = tmp + sizeof(tmp) - 1;
|
bool esc = false;
|
bool quote = false;
|
bool haveCmdNum = !mWithSeq;
|
|
memset(argv, 0, sizeof(argv));
|
memset(tmp, 0, sizeof(tmp));
|
while(*p) {
|
if (*p == '\\') {
|
if (esc) {
|
if (q >= qlimit)
|
goto overflow;
|
*q++ = '\\';
|
esc = false;
|
} else
|
esc = true;
|
p++;
|
continue;
|
} else if (esc) {
|
if (*p == '"') {
|
if (q >= qlimit)
|
goto overflow;
|
*q++ = '"';
|
} else if (*p == '\\') {
|
if (q >= qlimit)
|
goto overflow;
|
*q++ = '\\';
|
} else {
|
cli->sendMsg(500, "Unsupported escape sequence", false);
|
goto out;
|
}
|
p++;
|
esc = false;
|
continue;
|
}
|
|
if (*p == '"') {
|
if (quote)
|
quote = false;
|
else
|
quote = true;
|
p++;
|
continue;
|
}
|
|
if (q >= qlimit)
|
goto overflow;
|
*q = *p++;
|
if (!quote && *q == ' ') {
|
*q = '\0';
|
if (!haveCmdNum) {
|
char *endptr;
|
int cmdNum = (int)strtol(tmp, &endptr, 0);
|
if (endptr == nullptr || *endptr != '\0') {
|
cli->sendMsg(500, "Invalid sequence number", false);
|
goto out;
|
}
|
cli->setCmdNum(cmdNum);
|
haveCmdNum = true;
|
} else {
|
if (argc >= CMD_ARGS_MAX)
|
goto overflow;
|
argv[argc++] = strdup(tmp);
|
}
|
memset(tmp, 0, sizeof(tmp));
|
q = tmp;
|
continue;
|
}
|
q++;
|
}
|
|
*q = '\0';
|
if (argc >= CMD_ARGS_MAX)
|
goto overflow;
|
argv[argc++] = strdup(tmp);
|
#if 0
|
for (int k = 0; k < argc; k++) {
|
SLOGD("arg[%d] = '%s'", k, argv[k]);
|
}
|
#endif
|
|
if (quote) {
|
cli->sendMsg(500, "Unclosed quotes error", false);
|
goto out;
|
}
|
|
if (errorRate && (++mCommandCount % errorRate == 0)) {
|
/* ignore this command - let the timeout handler handle it */
|
SLOGE("Faking a timeout");
|
goto out;
|
}
|
|
for (auto* c : mCommands) {
|
if (!strcmp(argv[0], c->getCommand())) {
|
if (c->runCommand(cli, argc, argv)) {
|
SLOGW("Handler '%s' error (%s)", c->getCommand(), strerror(errno));
|
}
|
goto out;
|
}
|
}
|
cli->sendMsg(500, "Command not recognized", false);
|
out:
|
int j;
|
for (j = 0; j < argc; j++)
|
free(argv[j]);
|
return;
|
|
overflow:
|
cli->sendMsg(500, "Command too long", false);
|
goto out;
|
}
|
