mirror of
https://github.com/ceph/ceph
synced 2024-12-29 15:03:33 +00:00
143 lines
3.4 KiB
C++
143 lines
3.4 KiB
C++
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
|
|
// vim: ts=8 sw=2 smarttab
|
|
/*
|
|
* Ceph - scalable distributed file system
|
|
*
|
|
* Copyright (C) 2004-2006 Sage Weil <sage@newdream.net>
|
|
*
|
|
* This is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License version 2.1, as published by the Free Software
|
|
* Foundation. See file COPYING.
|
|
*
|
|
*/
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
|
|
#include <sys/stat.h>
|
|
#include <iostream>
|
|
#include <string>
|
|
using namespace std;
|
|
|
|
#include "config.h"
|
|
|
|
#include "mon/MonMap.h"
|
|
|
|
void usage()
|
|
{
|
|
cout << " usage: [--print] [--create [--clobber]] [--add 1.2.3.4:567] [--rm 1.2.3.4:567] <mapfilename>" << std::endl;
|
|
exit(1);
|
|
}
|
|
|
|
int main(int argc, const char **argv)
|
|
{
|
|
vector<const char*> args;
|
|
argv_to_vec(argc, argv, args);
|
|
DEFINE_CONF_VARS(usage);
|
|
|
|
const char *me = argv[0];
|
|
|
|
const char *fn = 0;
|
|
bool print = false;
|
|
bool create = false;
|
|
bool clobber = false;
|
|
bool modified = false;
|
|
list<entity_addr_t> add, rm;
|
|
|
|
FOR_EACH_ARG(args) {
|
|
if (CONF_ARG_EQ("print", '\0')) {
|
|
CONF_SAFE_SET_ARG_VAL(&print, OPT_BOOL);
|
|
} else if (CONF_ARG_EQ("create", '\0')) {
|
|
CONF_SAFE_SET_ARG_VAL(&create, OPT_BOOL);
|
|
} else if (CONF_ARG_EQ("clobber", '\0')) {
|
|
CONF_SAFE_SET_ARG_VAL(&clobber, OPT_BOOL);
|
|
} else if (CONF_ARG_EQ("add", '\0') ||
|
|
CONF_ARG_EQ("rm", '\0')) {
|
|
bool is_add=CONF_ARG_EQ("add", '\0');
|
|
if (++i >= args.size())
|
|
usage();
|
|
entity_addr_t addr;
|
|
if (!parse_ip_port(args[i], addr)) {
|
|
cerr << me << ": invalid ip:port '" << args[i] << "'" << std::endl;
|
|
return -1;
|
|
}
|
|
//inst.name = entity_name_t::MON(monmap.size());
|
|
if (is_add)
|
|
add.push_back(addr);
|
|
else
|
|
rm.push_back(addr);
|
|
modified = true;
|
|
} else if (!fn)
|
|
fn = args[i];
|
|
else {
|
|
cout << "invalid argument: '" << args[i] << "'" << std::endl;
|
|
usage();
|
|
}
|
|
}
|
|
if (!fn)
|
|
usage();
|
|
|
|
MonMap monmap;
|
|
|
|
cout << me << ": monmap file " << fn << std::endl;
|
|
|
|
int r = 0;
|
|
if (!(create && clobber))
|
|
r = monmap.read(fn);
|
|
|
|
char buf[80];
|
|
if (!create && r < 0) {
|
|
cerr << me << ": couldn't open " << fn << ": " << strerror_r(errno, buf, sizeof(buf)) << std::endl;
|
|
return -1;
|
|
}
|
|
else if (create && !clobber && r == 0) {
|
|
cerr << me << ": " << fn << " exists, --clobber to overwrite" << std::endl;
|
|
return -1;
|
|
}
|
|
|
|
if (create) {
|
|
srand(getpid() + time(0));
|
|
monmap.generate_fsid();
|
|
cout << me << ": generated fsid " << monmap.fsid << std::endl;
|
|
modified++;
|
|
}
|
|
|
|
for (list<entity_addr_t>::iterator p = add.begin(); p != add.end(); p++)
|
|
monmap.add(*p);
|
|
for (list<entity_addr_t>::iterator p = rm.begin(); p != rm.end(); p++) {
|
|
cout << me << ": removing " << *p << std::endl;
|
|
if (!monmap.remove(*p)) {
|
|
cerr << me << ": map does not contain " << *p << std::endl;
|
|
usage();
|
|
}
|
|
}
|
|
|
|
if (!print && !modified)
|
|
usage();
|
|
|
|
if (modified)
|
|
monmap.epoch++;
|
|
|
|
if (print)
|
|
monmap.print(cout);
|
|
|
|
if (modified) {
|
|
// write it out
|
|
cout << me << ": writing epoch " << monmap.epoch
|
|
<< " to " << fn
|
|
<< " (" << monmap.size() << " monitors)"
|
|
<< std::endl;
|
|
int r = monmap.write(fn);
|
|
if (r < 0) {
|
|
cerr << "monmaptool: error writing to '" << fn << "': " << strerror_r(-r, buf, sizeof(buf)) << std::endl;
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
|
|
return 0;
|
|
}
|