int
main(int argc, char **argv)
{
int sep = __pmPathSeparator();
int sts;
int c;
char *p;
char pmnsfile[MAXPATHLEN];
char outfname[MAXPATHLEN];
struct stat sbuf;
if ((p = getenv("PMNS_DEFAULT")) != NULL) {
strncpy(pmnsfile, p, MAXPATHLEN);
pmnsfile[MAXPATHLEN-1]= '\0';
} else {
snprintf(pmnsfile, sizeof(pmnsfile), "%s%c" "pmns" "%c" "root",
pmGetConfig("PCP_VAR_DIR"), sep, sep);
}
while ((c = pmgetopt_r(argc, argv, &opts)) != EOF) {
switch (c) {
case 'd': /* duplicate PMIDs are OK */
fprintf(stderr, "%s: Warning: -d deprecated, duplicate PMNS names allowed by default\n", pmProgname);
break;
case 'D': /* debug flag */
if ((sts = __pmParseDebug(opts.optarg)) < 0) {
fprintf(stderr, "%s: unrecognized debug flag specification (%s)\n",
pmProgname, opts.optarg);
opts.errors++;
} else {
pmDebug |= sts;
}
break;
case 'n': /* alternative name space file */
strncpy(pmnsfile, opts.optarg, MAXPATHLEN);
pmnsfile[MAXPATHLEN-1]= '\0';
break;
case '?':
default:
opts.errors++;
break;
}
}
if (opts.errors || opts.optind > argc - 1) {
pmUsageMessage(&opts);
exit(1);
}
if ((sts = pmLoadASCIINameSpace(pmnsfile, 1)) < 0) {
fprintf(stderr, "%s: Error: pmLoadASCIINameSpace(%s, 1): %s\n",
pmProgname, pmnsfile, pmErrStr(sts));
exit(1);
}
{
__pmnsTree *t;
t = __pmExportPMNS();
if (t == NULL) {
/* sanity check - shouldn't ever happen */
fprintf(stderr, "Exported PMNS is NULL !");
exit(1);
}
root = t->root;
}
while (opts.optind < argc) {
delpmns(root, fullname = argv[opts.optind]);
opts.optind++;
}
/*
* from here on, ignore SIGHUP, SIGINT and SIGTERM to protect
* the integrity of the new ouput file
*/
__pmSetSignalHandler(SIGHUP, SIG_IGN);
__pmSetSignalHandler(SIGINT, SIG_IGN);
__pmSetSignalHandler(SIGTERM, SIG_IGN);
snprintf(outfname, sizeof(outfname), "%s.new", pmnsfile);
if ((outf = fopen(outfname, "w")) == NULL) {
fprintf(stderr, "%s: Error: cannot open PMNS file \"%s\" for writing: %s\n",
pmProgname, outfname, osstrerror());
exit(1);
}
if (stat(pmnsfile, &sbuf) == 0) {
/*
* preserve the mode and ownership of any existing PMNS file
*/
chmod(outfname, sbuf.st_mode & ~S_IFMT);
#if defined(HAVE_CHOWN)
if (chown(outfname, sbuf.st_uid, sbuf.st_gid) < 0)
fprintf(stderr, "%s: chown(%s, ...) failed: %s\n",
pmProgname, outfname, osstrerror());
#endif
}
pmns_output(root, outf);
fclose(outf);
/* rename the PMNS */
if (rename2(outfname, pmnsfile) == -1) {
fprintf(stderr, "%s: cannot rename \"%s\" to \"%s\": %s\n",
pmProgname, outfname, pmnsfile, osstrerror());
/* remove the new PMNS */
unlink(outfname);
exit(1);
}
exit(0);
}
int
main(int argc, char **argv)
{
int sts;
int first = 1;
int c;
int j;
int force = 0;
int asis = 0;
int dupok = 0;
__pmnsNode *tmp;
umask((mode_t)022); /* anything else is pretty silly */
while ((c = pmgetopt_r(argc, argv, &opts)) != EOF) {
switch (c) {
case 'a':
asis = 1;
break;
case 'd': /* duplicate PMIDs are OK */
dupok = 1;
break;
case 'D': /* debug flag */
if ((sts = __pmParseDebug(opts.optarg)) < 0) {
pmprintf("%s: unrecognized debug flag specification (%s)\n",
pmProgname, opts.optarg);
opts.errors++;
} else {
pmDebug |= sts;
}
break;
case 'f': /* force ... unlink file first */
force = 1;
break;
case 'v':
verbose++;
break;
case '?':
default:
opts.errors++;
break;
}
}
if (opts.errors || opts.optind > argc - 2) {
pmUsageMessage(&opts);
exit(1);
}
if (force)
unlink(argv[argc-1]);
if (access(argv[argc-1], F_OK) == 0) {
fprintf(stderr, "%s: Error: output PMNS file \"%s\" already exists!\nYou must either remove it first, or use -f\n",
pmProgname, argv[argc-1]);
exit(1);
}
/*
* from here on, ignore SIGHUP, SIGINT and SIGTERM to protect
* the integrity of the new ouput file
*/
__pmSetSignalHandler(SIGHUP, SIG_IGN);
__pmSetSignalHandler(SIGINT, SIG_IGN);
__pmSetSignalHandler(SIGTERM, SIG_IGN);
if ((outf = fopen(argv[argc-1], "w+")) == NULL) {
fprintf(stderr, "%s: Error: cannot create output PMNS file \"%s\": %s\n", pmProgname, argv[argc-1], osstrerror());
exit(1);
}
if (!asis)
sortargs(&argv[opts.optind], argc - opts.optind - 1);
j = opts.optind;
while (j < argc-1) {
if (verbose)
printf("%s:\n", argv[j]);
if ((sts = pmLoadASCIINameSpace(argv[j], dupok)) < 0) {
fprintf(stderr, "%s: Error: pmLoadNameSpace(%s): %s\n",
pmProgname, argv[j], pmErrStr(sts));
exit(1);
}
{
__pmnsTree *t;
t = __pmExportPMNS();
if (t == NULL) {
/* sanity check - shouldn't ever happen */
fprintf(stderr, "Exported PMNS is NULL !");
exit(1);
}
tmp = t->root;
}
if (first) {
root = tmp;
first = 0;
}
else {
pmns_traverse(tmp, 0, "", merge);
}
j++;
}
pmns_output(root, outf);
fclose(outf);
/*
* now load the merged PMNS to check for errors ...
*/
if ((sts = pmLoadASCIINameSpace(argv[argc-1], dupok)) < 0) {
fprintf(stderr, "%s: Error: pmLoadNameSpace(%s): %s\n",
pmProgname, argv[argc-1], pmErrStr(sts));
exit(1);
}
exit(0);
}
