Inital commit
This commit is contained in:
Olof hagsand
parent
edc5e091bb
commit
d6e393ea58
145 changed files with 58117 additions and 0 deletions
241
apps/dbctrl/dbctrl_main.c
Normal file
241
apps/dbctrl/dbctrl_main.c
Normal file
|
|
@ -0,0 +1,241 @@
|
|||
/*
|
||||
*
|
||||
Copyright (C) 2009-2016 Olof Hagsand and Benny Holmgren
|
||||
|
||||
This file is part of CLICON.
|
||||
|
||||
CLICON 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.
|
||||
|
||||
CLICON 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 CLICON; see the file COPYING. If not, see
|
||||
<http://www.gnu.org/licenses/>.
|
||||
|
||||
*/
|
||||
|
||||
#ifdef HAVE_CONFIG_H
|
||||
#include "clicon_config.h" /* generated by config & autoconf */
|
||||
#endif
|
||||
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
#include <stdarg.h>
|
||||
#include <errno.h>
|
||||
#include <signal.h>
|
||||
#include <fcntl.h>
|
||||
#include <time.h>
|
||||
#include <syslog.h>
|
||||
#include <sys/time.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/param.h>
|
||||
#include <sys/types.h>
|
||||
#include <sys/stat.h>
|
||||
#include <netinet/in.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <netinet/in.h>
|
||||
|
||||
/* cligen */
|
||||
#include <cligen/cligen.h>
|
||||
|
||||
/* clicon */
|
||||
#include <clicon/clicon.h>
|
||||
|
||||
/* Command line options to be passed to getopt(3) */
|
||||
#define DBCTRL_OPTS "hDd:pbn:r:m:Zi"
|
||||
|
||||
/*! Write database contents to file, xml database variant
|
||||
* @param[in] dbspec If set make a sanity check if key dont match (just)
|
||||
*/
|
||||
static int
|
||||
dump_database(FILE *f,
|
||||
char *dbname,
|
||||
char *rxkey)
|
||||
{
|
||||
int retval = 0;
|
||||
int npairs;
|
||||
struct db_pair *pairs;
|
||||
|
||||
/* Default is match all */
|
||||
if (rxkey == NULL)
|
||||
rxkey = "^.*$";
|
||||
|
||||
/* Get all keys/values for vector */
|
||||
if ((npairs = db_regexp(dbname, rxkey, __FUNCTION__, &pairs, 0)) < 0)
|
||||
return -1;
|
||||
|
||||
for (npairs--; npairs >= 0; npairs--)
|
||||
fprintf(f, "%s %s\n", pairs[npairs].dp_key,
|
||||
pairs[npairs].dp_val?pairs[npairs].dp_val:"");
|
||||
unchunk_group(__FUNCTION__);
|
||||
return retval;
|
||||
}
|
||||
|
||||
|
||||
/*
|
||||
* remove_entry
|
||||
*/
|
||||
static int
|
||||
remove_entry(char *dbname, char *key)
|
||||
{
|
||||
return db_del(dbname, key);
|
||||
}
|
||||
|
||||
/*
|
||||
* usage
|
||||
*/
|
||||
static void
|
||||
usage(char *argv0)
|
||||
{
|
||||
fprintf(stderr, "usage:%s\n"
|
||||
"where options are\n"
|
||||
"\t-h\t\tHelp\n"
|
||||
"\t-D\t\tDebug\n"
|
||||
"\t-d <dbname>\tDatabase name (default: running_db)\n"
|
||||
"\t-p\t\tDump database on stdout\n"
|
||||
"\t-b\t\tBrief output, just print keys. Combine with -p or -m\n"
|
||||
"\t-n \"<key> <val>\" Add database entry\n"
|
||||
"\t-r <key>\tRemove database entry\n"
|
||||
"\t-m <regexp key>\tMatch regexp key in database\n"
|
||||
"\t-Z\t\tDelete database\n"
|
||||
"\t-i\t\tInit database\n",
|
||||
argv0
|
||||
);
|
||||
exit(0);
|
||||
}
|
||||
|
||||
int
|
||||
main(int argc, char **argv)
|
||||
{
|
||||
char c;
|
||||
int zapdb;
|
||||
int initdb;
|
||||
int dumpdb;
|
||||
int addent;
|
||||
int rment;
|
||||
char *matchkey = NULL;
|
||||
char *addstr;
|
||||
char rmkey[MAXPATHLEN];
|
||||
int brief;
|
||||
char dbname[MAXPATHLEN] = {0,};
|
||||
int use_syslog;
|
||||
yang_spec *yspec;
|
||||
clicon_handle h;
|
||||
|
||||
/* In the startup, logs to stderr & debug flag set later */
|
||||
clicon_log_init(__PROGRAM__, LOG_INFO, CLICON_LOG_STDERR);
|
||||
|
||||
/* Defaults */
|
||||
zapdb = 0;
|
||||
initdb = 0;
|
||||
dumpdb = 0;
|
||||
addent = 0;
|
||||
rment = 0;
|
||||
brief = 0;
|
||||
use_syslog = 0;
|
||||
addstr = NULL;
|
||||
memset(rmkey, '\0', sizeof(rmkey));
|
||||
|
||||
if ((h = clicon_handle_init()) == NULL)
|
||||
goto done;
|
||||
/* getopt in two steps, first find config-file before over-riding options. */
|
||||
while ((c = getopt(argc, argv, DBCTRL_OPTS)) != -1)
|
||||
switch (c) {
|
||||
case '?' :
|
||||
case 'h' : /* help */
|
||||
usage(argv[0]);
|
||||
break;
|
||||
case 'D' : /* debug */
|
||||
debug = 1;
|
||||
break;
|
||||
case 'S': /* Log on syslog */
|
||||
use_syslog = 1;
|
||||
break;
|
||||
}
|
||||
/*
|
||||
* Logs, error and debug to stderr or syslog, set debug level
|
||||
*/
|
||||
clicon_log_init(__PROGRAM__, debug?LOG_DEBUG:LOG_INFO,
|
||||
use_syslog?CLICON_LOG_SYSLOG:CLICON_LOG_STDERR);
|
||||
clicon_debug_init(debug, NULL);
|
||||
|
||||
/* Now rest of options */
|
||||
optind = 1;
|
||||
while ((c = getopt(argc, argv, DBCTRL_OPTS)) != -1)
|
||||
switch (c) {
|
||||
case 'Z': /* Zap database */
|
||||
zapdb++;
|
||||
break;
|
||||
case 'i': /* Init database */
|
||||
initdb++;
|
||||
break;
|
||||
case 'p': /* Dump/print database */
|
||||
dumpdb++;
|
||||
break;
|
||||
case 'b': /* Dump/print/match database brief (combone w -p or -m) */
|
||||
brief++;
|
||||
break;
|
||||
case 'd': /* dbname */
|
||||
if (!optarg || sscanf(optarg, "%s", dbname) != 1)
|
||||
usage(argv[0]);
|
||||
break;
|
||||
case 'n': /* add database entry */
|
||||
if (!optarg || !strlen(optarg) || (addstr = strdup(optarg)) == NULL)
|
||||
usage(argv[0]);
|
||||
/* XXX addign both key and value, for now only key */
|
||||
addent++;
|
||||
break;
|
||||
case 'r':
|
||||
if (!optarg || sscanf(optarg, "%s", rmkey) != 1)
|
||||
usage(argv[0]);
|
||||
rment++;
|
||||
break;
|
||||
case 'm':
|
||||
if (!optarg || !strlen(optarg) || (matchkey = strdup(optarg)) == NULL)
|
||||
usage(argv[0]);
|
||||
dumpdb++;
|
||||
break;
|
||||
case 'D': /* Processed earlier, ignore now. */
|
||||
case 'S':
|
||||
break;
|
||||
default:
|
||||
usage(argv[0]);
|
||||
break;
|
||||
}
|
||||
argc -= optind;
|
||||
argv += optind;
|
||||
|
||||
if (*dbname == '\0'){
|
||||
clicon_err(OE_FATAL, 0, "database not specified (with -d <db>): %s");
|
||||
goto done;
|
||||
}
|
||||
yspec = clicon_dbspec_yang(h);
|
||||
if (dumpdb){
|
||||
if (dump_database(stdout, dbname, matchkey)) {
|
||||
fprintf(stderr, "Match error\n");
|
||||
goto done;
|
||||
}
|
||||
}
|
||||
if (addent) /* add entry */
|
||||
if (xmldb_put_xkey(dbname, addstr, NULL, yspec, OP_REPLACE) < 0)
|
||||
goto done;
|
||||
if (rment)
|
||||
if (remove_entry(dbname, rmkey) < 0)
|
||||
goto done;
|
||||
if (zapdb) /* remove databases */
|
||||
unlink(dbname);
|
||||
if (initdb)
|
||||
if (db_init(dbname) < 0)
|
||||
goto done;
|
||||
|
||||
done:
|
||||
return 0;
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue