159 lines
3.9 KiB
C++
159 lines
3.9 KiB
C++
/**
|
|
* Copyright (C) ARM Limited 2014. All rights reserved.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License version 2 as
|
|
* published by the Free Software Foundation.
|
|
*/
|
|
|
|
#include "FSDriver.h"
|
|
|
|
#include <fcntl.h>
|
|
#include <regex.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include "DriverSource.h"
|
|
#include "Logging.h"
|
|
|
|
class FSCounter : public DriverCounter {
|
|
public:
|
|
FSCounter(DriverCounter *next, char *name, char *path, const char *regex);
|
|
~FSCounter();
|
|
|
|
const char *getPath() const { return mPath; }
|
|
|
|
int64_t read();
|
|
|
|
private:
|
|
char *const mPath;
|
|
regex_t mReg;
|
|
bool mUseRegex;
|
|
|
|
// Intentionally unimplemented
|
|
FSCounter(const FSCounter &);
|
|
FSCounter &operator=(const FSCounter &);
|
|
};
|
|
|
|
FSCounter::FSCounter(DriverCounter *next, char *name, char *path, const char *regex) : DriverCounter(next, name), mPath(path), mUseRegex(regex != NULL) {
|
|
if (mUseRegex) {
|
|
int result = regcomp(&mReg, regex, REG_EXTENDED);
|
|
if (result != 0) {
|
|
char buf[128];
|
|
regerror(result, &mReg, buf, sizeof(buf));
|
|
logg->logError(__FILE__, __LINE__, "Invalid regex '%s': %s", regex, buf);
|
|
handleException();
|
|
}
|
|
}
|
|
}
|
|
|
|
FSCounter::~FSCounter() {
|
|
free(mPath);
|
|
if (mUseRegex) {
|
|
regfree(&mReg);
|
|
}
|
|
}
|
|
|
|
int64_t FSCounter::read() {
|
|
int64_t value;
|
|
if (mUseRegex) {
|
|
char buf[4096];
|
|
size_t pos = 0;
|
|
const int fd = open(mPath, O_RDONLY | O_CLOEXEC);
|
|
if (fd < 0) {
|
|
goto fail;
|
|
}
|
|
while (pos < sizeof(buf) - 1) {
|
|
const ssize_t bytes = ::read(fd, buf + pos, sizeof(buf) - pos - 1);
|
|
if (bytes < 0) {
|
|
goto fail;
|
|
} else if (bytes == 0) {
|
|
break;
|
|
}
|
|
pos += bytes;
|
|
}
|
|
close(fd);
|
|
buf[pos] = '\0';
|
|
|
|
regmatch_t match[2];
|
|
int result = regexec(&mReg, buf, 2, match, 0);
|
|
if (result != 0) {
|
|
regerror(result, &mReg, buf, sizeof(buf));
|
|
logg->logError(__FILE__, __LINE__, "Parsing %s failed: %s", mPath, buf);
|
|
handleException();
|
|
}
|
|
|
|
if (match[1].rm_so < 0) {
|
|
logg->logError(__FILE__, __LINE__, "Parsing %s failed", mPath);
|
|
handleException();
|
|
}
|
|
|
|
errno = 0;
|
|
value = strtoll(buf + match[1].rm_so, NULL, 0);
|
|
if (errno != 0) {
|
|
logg->logError(__FILE__, __LINE__, "Parsing %s failed: %s", mPath, strerror(errno));
|
|
handleException();
|
|
}
|
|
} else {
|
|
if (DriverSource::readInt64Driver(mPath, &value) != 0) {
|
|
goto fail;
|
|
}
|
|
}
|
|
return value;
|
|
|
|
fail:
|
|
logg->logError(__FILE__, __LINE__, "Unable to read %s", mPath);
|
|
handleException();
|
|
}
|
|
|
|
FSDriver::FSDriver() {
|
|
}
|
|
|
|
FSDriver::~FSDriver() {
|
|
}
|
|
|
|
void FSDriver::readEvents(mxml_node_t *const xml) {
|
|
mxml_node_t *node = xml;
|
|
while (true) {
|
|
node = mxmlFindElement(node, xml, "event", NULL, NULL, MXML_DESCEND);
|
|
if (node == NULL) {
|
|
break;
|
|
}
|
|
const char *counter = mxmlElementGetAttr(node, "counter");
|
|
if (counter == NULL) {
|
|
continue;
|
|
}
|
|
|
|
if (counter[0] == '/') {
|
|
logg->logError(__FILE__, __LINE__, "Old style filesystem counter (%s) detected, please create a new unique counter value and move the filename into the path attribute, see events-Filesystem.xml for examples", counter);
|
|
handleException();
|
|
}
|
|
|
|
if (strncmp(counter, "filesystem_", 11) != 0) {
|
|
continue;
|
|
}
|
|
|
|
const char *path = mxmlElementGetAttr(node, "path");
|
|
if (path == NULL) {
|
|
logg->logError(__FILE__, __LINE__, "The filesystem counter %s is missing the required path attribute", counter);
|
|
handleException();
|
|
}
|
|
const char *regex = mxmlElementGetAttr(node, "regex");
|
|
setCounters(new FSCounter(getCounters(), strdup(counter), strdup(path), regex));
|
|
}
|
|
}
|
|
|
|
int FSDriver::writeCounters(mxml_node_t *root) const {
|
|
int count = 0;
|
|
for (FSCounter *counter = static_cast<FSCounter *>(getCounters()); counter != NULL; counter = static_cast<FSCounter *>(counter->getNext())) {
|
|
if (access(counter->getPath(), R_OK) == 0) {
|
|
mxml_node_t *node = mxmlNewElement(root, "counter");
|
|
mxmlElementSetAttr(node, "name", counter->getName());
|
|
++count;
|
|
}
|
|
}
|
|
|
|
return count;
|
|
}
|