kopia lustrzana https://github.com/weetmuts/wmbusmeters
349 wiersze
12 KiB
C++
349 wiersze
12 KiB
C++
/*
|
|
Copyright (C) 2017-2019 Fredrik Öhrström
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include"cmdline.h"
|
|
#include"config.h"
|
|
#include"meters.h"
|
|
#include"util.h"
|
|
|
|
#include<string>
|
|
|
|
using namespace std;
|
|
|
|
unique_ptr<Configuration> parseCommandLine(int argc, char **argv) {
|
|
|
|
Configuration * c = new Configuration;
|
|
|
|
int i=1;
|
|
const char *filename = strrchr(argv[0], '/');
|
|
if (filename) {
|
|
filename++;
|
|
} else {
|
|
filename = argv[0];
|
|
}
|
|
if (!strcmp(filename, "wmbusmetersd")) {
|
|
c->daemon = true;
|
|
if (argc != 2) {
|
|
error("Usage error: wmbusmetersd must have a single argument to the pid file.\n");
|
|
}
|
|
c->pid_file = argv[1];
|
|
return unique_ptr<Configuration>(c);
|
|
}
|
|
if (argc < 2) {
|
|
c->need_help = true;
|
|
return unique_ptr<Configuration>(c);
|
|
}
|
|
while (argv[i] && argv[i][0] == '-') {
|
|
if (!strcmp(argv[i], "-h") || !strcmp(argv[i], "-help") || !strcmp(argv[i], "--help")) {
|
|
c->need_help = true;
|
|
return unique_ptr<Configuration>(c);
|
|
}
|
|
if (!strcmp(argv[i], "--silence")) {
|
|
c->silence = true;
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strcmp(argv[i], "--verbose")) {
|
|
c->verbose = true;
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strcmp(argv[i], "--version")) {
|
|
c->version = true;
|
|
return unique_ptr<Configuration>(c);
|
|
}
|
|
if (!strcmp(argv[i], "--license")) {
|
|
c->license = true;
|
|
return unique_ptr<Configuration>(c);
|
|
}
|
|
if (!strcmp(argv[i], "--debug")) {
|
|
c->debug = true;
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strncmp(argv[i], "--listento=", 11)) {
|
|
LinkModeSet lms = parseLinkModes(argv[i]+11);
|
|
if (lms.bits() == 0) {
|
|
error("Unknown link mode \"%s\"!\n", argv[i]+11);
|
|
}
|
|
if (c->link_mode_configured) {
|
|
error("You have already specified a link mode!\n");
|
|
}
|
|
c->listen_to_link_modes = lms;
|
|
c->link_mode_configured = true;
|
|
i++;
|
|
continue;
|
|
}
|
|
|
|
LinkMode lm = isLinkModeOption(argv[i]);
|
|
if (lm != LinkMode::UNKNOWN) {
|
|
if (c->link_mode_configured) {
|
|
error("You have already specified a link mode!\n");
|
|
}
|
|
c->listen_to_link_modes.addLinkMode(lm);
|
|
c->link_mode_configured = true;
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strcmp(argv[i], "--logtelegrams")) {
|
|
c->logtelegrams = true;
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strncmp(argv[i], "--useconfig", 11)) {
|
|
if (strlen(argv[i]) == 11)
|
|
{
|
|
c->useconfig = true;
|
|
c->config_root = "";
|
|
return unique_ptr<Configuration>(c);
|
|
}
|
|
else if (strlen(argv[i]) > 12 && argv[i][11] == '=')
|
|
{
|
|
size_t len = strlen(argv[i]) - 12;
|
|
c->useconfig = true;
|
|
c->config_root = string(argv[i]+12, len);
|
|
if (c->config_root == "/") {
|
|
c->config_root = "";
|
|
}
|
|
return unique_ptr<Configuration>(c);
|
|
}
|
|
else
|
|
{
|
|
error("You must supply a directory to --useconfig=dir\n");
|
|
}
|
|
i++;
|
|
if (i > 1 || argc > 2) {
|
|
error("Usage error: --useconfig implies no other arguments on the command line.\n");
|
|
}
|
|
continue;
|
|
}
|
|
if (!strcmp(argv[i], "--reload")) {
|
|
c->reload = true;
|
|
if (i > 1 || argc > 2) {
|
|
error("Usage error: --reload implies no other arguments on the command line.\n");
|
|
}
|
|
return unique_ptr<Configuration>(c);
|
|
}
|
|
if (!strncmp(argv[i], "--format", 8)) {
|
|
if (strlen(argv[i]) == 8 ||
|
|
(strlen(argv[i]) == 13 &&
|
|
!strncmp(argv[i]+8, "=json", 5)))
|
|
{
|
|
c->json = true;
|
|
c->fields = false;
|
|
}
|
|
else if (strlen(argv[i]) == 15 &&
|
|
!strncmp(argv[i]+7, "=fields", 7))
|
|
{
|
|
c->json = false;
|
|
c->fields = true;
|
|
c->separator = ';';
|
|
} else {
|
|
error("Unknown output format: \"%s\"\n", argv[i]+8);
|
|
}
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strncmp(argv[i], "--addconversions=", 17)) {
|
|
if (strlen(argv[i]) > 16)
|
|
{
|
|
string s = string(argv[i]+17);
|
|
handleConversions(c, s);
|
|
} else {
|
|
error("You must supply conversion units.\n");
|
|
}
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strncmp(argv[i], "--separator=", 12)) {
|
|
if (!c->fields) {
|
|
error("You must specify --format=fields before --separator=X\n");
|
|
}
|
|
if (strlen(argv[i]) != 13) {
|
|
error("You must supply a single character as the field separator.\n");
|
|
}
|
|
c->separator = argv[i][12];
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strncmp(argv[i], "--meterfilesaction", 18)) {
|
|
if (strlen(argv[i]) > 18 && argv[i][18] == '=') {
|
|
if (!strncmp(argv[i]+19, "overwrite", 9)) {
|
|
c->meterfiles_action = MeterFileType::Overwrite;
|
|
} else if (!strncmp(argv[i]+19, "append", 6)) {
|
|
c->meterfiles_action = MeterFileType::Append;
|
|
} else {
|
|
error("No such meter file action %s\n", argv[i]+19);
|
|
}
|
|
} else {
|
|
error("Incorrect option %s\n", argv[i]);
|
|
}
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strncmp(argv[i], "--meterfilesnaming", 18)) {
|
|
if (strlen(argv[i]) > 18 && argv[i][18] == '=') {
|
|
if (!strncmp(argv[i]+19, "name-id", 7))
|
|
{
|
|
c->meterfiles_naming = MeterFileNaming::NameId;
|
|
}
|
|
else if (!strncmp(argv[i]+19, "name", 4))
|
|
{
|
|
c->meterfiles_naming = MeterFileNaming::Name;
|
|
}
|
|
else if (!strncmp(argv[i]+19, "id", 2))
|
|
{
|
|
c->meterfiles_naming = MeterFileNaming::Id;
|
|
} else
|
|
{
|
|
error("No such meter file naming %s\n", argv[i]+19);
|
|
}
|
|
} else {
|
|
error("Incorrect option %s\n", argv[i]);
|
|
}
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strcmp(argv[i], "--meterfiles") ||
|
|
(!strncmp(argv[i], "--meterfiles", 12) &&
|
|
strlen(argv[i]) > 12 &&
|
|
argv[i][12] == '='))
|
|
{
|
|
c->meterfiles = true;
|
|
size_t len = strlen(argv[i]);
|
|
if (len > 13) {
|
|
c->meterfiles_dir = string(argv[i]+13, len-13);
|
|
} else {
|
|
c->meterfiles_dir = "/tmp";
|
|
}
|
|
if (!checkIfDirExists(c->meterfiles_dir.c_str())) {
|
|
error("Cannot write meter files into dir \"%s\"\n", c->meterfiles_dir.c_str());
|
|
}
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strncmp(argv[i], "--logfile=", 10)) {
|
|
c->use_logfile = true;
|
|
if (strlen(argv[i]) > 10) {
|
|
size_t len = strlen(argv[i])-10;
|
|
if (len > 0) {
|
|
c->logfile = string(argv[i]+10, len);
|
|
} else {
|
|
error("Not a valid log file name.");
|
|
}
|
|
}
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strncmp(argv[i], "--shell=", 8)) {
|
|
string cmd = string(argv[i]+8);
|
|
if (cmd == "") {
|
|
error("The shell command cannot be empty.\n");
|
|
}
|
|
c->shells.push_back(cmd);
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strncmp(argv[i], "--shellenvs", 11)) {
|
|
c->list_shell_envs = true;
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strcmp(argv[i], "--oneshot")) {
|
|
c->oneshot = true;
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strncmp(argv[i], "--exitafter=", 12) && strlen(argv[i]) > 12) {
|
|
c->exitafter = parseTime(argv[i]+12);
|
|
if (c->exitafter <= 0) {
|
|
error("Not a valid time to exit after. \"%s\"\n", argv[i]+12);
|
|
}
|
|
i++;
|
|
continue;
|
|
}
|
|
if (!strcmp(argv[i], "--")) {
|
|
i++;
|
|
break;
|
|
}
|
|
error("Unknown option \"%s\"\n", argv[i]);
|
|
}
|
|
|
|
char *extra = argv[i] ? strchr(argv[i], ':') : NULL ;
|
|
if (extra) {
|
|
*extra = 0;
|
|
extra++;
|
|
c->device_extra = extra;
|
|
}
|
|
if (argv[i]) {
|
|
c->device = argv[i];
|
|
}
|
|
i++;
|
|
if (c->device.length() == 0) {
|
|
error("You must supply the usb device to which the wmbus dongle is connected.\n");
|
|
}
|
|
|
|
if ((argc-i) % 4 != 0) {
|
|
error("For each meter you must supply a: name,type,id and key.\n");
|
|
}
|
|
int num_meters = (argc-i)/4;
|
|
|
|
for (int m=0; m<num_meters; ++m) {
|
|
string name = argv[m*4+i+0];
|
|
string type = argv[m*4+i+1];
|
|
string id = argv[m*4+i+2];
|
|
string key = argv[m*4+i+3];
|
|
LinkModeSet modes;
|
|
size_t colon = type.find(':');
|
|
MeterType mt = toMeterType(type);
|
|
if (colon != string::npos)
|
|
{
|
|
// The config can be supplied after the type, like this:
|
|
// apator162:c1
|
|
string modess = type.substr(colon+1);
|
|
type = type.substr(0, colon);
|
|
mt = toMeterType(type);
|
|
if (mt == MeterType::UNKNOWN) error("Not a valid meter type \"%s\"\n", type.c_str());
|
|
modes = parseLinkModes(modess);
|
|
LinkModeSet default_modes = toMeterLinkModeSet(type);
|
|
if (!default_modes.hasAll(modes))
|
|
{
|
|
string want = modes.hr();
|
|
string has = default_modes.hr();
|
|
error("(cmdline) cannot set link modes to: %s because meter %s only transmits on: %s\n",
|
|
want.c_str(), type.c_str(), has.c_str());
|
|
}
|
|
string modeshr = modes.hr();
|
|
debug("(cmdline) setting link modes to %s for meter %s\n",
|
|
modeshr.c_str(), name.c_str());
|
|
}
|
|
else {
|
|
modes = toMeterLinkModeSet(type);
|
|
}
|
|
|
|
mt = toMeterType(type);
|
|
|
|
if (mt == MeterType::UNKNOWN) error("Not a valid meter type \"%s\"\n", type.c_str());
|
|
if (!isValidId(id)) error("Not a valid meter id \"%s\"\n", id.c_str());
|
|
if (!isValidKey(key)) error("Not a valid meter key \"%s\"\n", key.c_str());
|
|
vector<string> no_meter_shells;
|
|
c->meters.push_back(MeterInfo(name, type, id, key, modes, no_meter_shells));
|
|
}
|
|
|
|
return unique_ptr<Configuration>(c);
|
|
}
|