/*************************************************************************************************
* The command line interface of miscellaneous utilities
* Copyright (C) 2009-2012 FAL Labs
* This file is part of Kyoto Cabinet.
* 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 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 "cmdcommon.h"
// global variables
const char* g_progname; // program name
// function prototypes
int main(int argc, char** argv);
static void usage();
static int32_t runhex(int argc, char** argv);
static int32_t runenc(int argc, char** argv);
static int32_t runciph(int argc, char** argv);
static int32_t runcomp(int argc, char** argv);
static int32_t runhash(int argc, char** argv);
static int32_t runregex(int argc, char** argv);
static int32_t runconf(int argc, char** argv);
static int32_t prochex(const char* file, bool dec);
static int32_t procenc(const char* file, int32_t mode, bool dec);
static int32_t procciph(const char* file, const char* key);
static int32_t proccomp(const char* file, int32_t mode, bool dec);
static int32_t prochash(const char* file, int32_t mode);
static int32_t procregex(const char* file, const char* pattern, const char* alt, int32_t opts);
static int32_t procconf(int32_t mode);
// main routine
int main(int argc, char** argv) {
g_progname = argv[0];
kc::setstdiobin();
if (argc < 2) usage();
int32_t rv = 0;
if (!std::strcmp(argv[1], "hex")) {
rv = runhex(argc, argv);
} else if (!std::strcmp(argv[1], "enc")) {
rv = runenc(argc, argv);
} else if (!std::strcmp(argv[1], "ciph")) {
rv = runciph(argc, argv);
} else if (!std::strcmp(argv[1], "comp")) {
rv = runcomp(argc, argv);
} else if (!std::strcmp(argv[1], "hash")) {
rv = runhash(argc, argv);
} else if (!std::strcmp(argv[1], "regex")) {
rv = runregex(argc, argv);
} else if (!std::strcmp(argv[1], "conf")) {
rv = runconf(argc, argv);
} else if (!std::strcmp(argv[1], "version") || !std::strcmp(argv[1], "--version")) {
printversion();
rv = 0;
} else {
usage();
}
return rv;
}
// print the usage and exit
static void usage() {
eprintf("%s: command line interface of miscellaneous utilities of Kyoto Cabinet\n",
g_progname);
eprintf("\n");
eprintf("usage:\n");
eprintf(" %s hex [-d] [file]\n", g_progname);
eprintf(" %s enc [-hex|-url|-quote] [-d] [file]\n", g_progname);
eprintf(" %s ciph [-key str] [file]\n", g_progname);
eprintf(" %s comp [-def|-gz|-lzo|-lzma] [-d] [file]\n", g_progname);
eprintf(" %s hash [-fnv|-path|-crc] [file]\n", g_progname);
eprintf(" %s regex [-alt str] [-ic] pattern [file]\n", g_progname);
eprintf(" %s conf [-v|-i|-l|-p]\n", g_progname);
eprintf(" %s version\n", g_progname);
eprintf("\n");
std::exit(1);
}
// parse arguments of hex command
static int32_t runhex(int argc, char** argv) {
bool argbrk = false;
const char* file = NULL;
bool dec = false;
for (int32_t i = 2; i < argc; i++) {
if (!argbrk && argv[i][0] == '-') {
if (!std::strcmp(argv[i], "--")) {
argbrk = true;
} else if (!std::strcmp(argv[i], "-d")) {
dec = true;
} else {
usage();
}
} else if (!file) {
argbrk = true;
file = argv[i];
} else {
usage();
}
}
int32_t rv = prochex(file, dec);
return rv;
}
// parse arguments of enc command
static int32_t runenc(int argc, char** argv) {
bool argbrk = false;
const char* file = NULL;
int32_t mode = 0;
bool dec = false;
for (int32_t i = 2; i < argc; i++) {
if (!argbrk && argv[i][0] == '-') {
if (!std::strcmp(argv[i], "--")) {
argbrk = true;
} else if (!std::strcmp(argv[i], "-hex")) {
mode = 1;
} else if (!std::strcmp(argv[i], "-url")) {
mode = 2;
} else if (!std::strcmp(argv[i], "-quote")) {
mode = 3;
} else if (!std::strcmp(argv[i], "-d")) {
dec = true;
} else {
usage();
}
} else if (!file) {
argbrk = true;
file = argv[i];
} else {
usage();
}
}
int32_t rv = procenc(file, mode, dec);
return rv;
}
// parse arguments of ciph command
static int32_t runciph(int argc, char** argv) {
bool argbrk = false;
const char* file = NULL;
const char* key = "";
for (int32_t i = 2; i < argc; i++) {
if (!argbrk && argv[i][0] == '-') {
if (!std::strcmp(argv[i], "--")) {
argbrk = true;
} else if (!std::strcmp(argv[i], "-key")) {
if (++i >= argc) usage();
key = argv[i];
} else {
usage();
}
} else if (!file) {
argbrk = true;
file = argv[i];
} else {
usage();
}
}
int32_t rv = procciph(file, key);
return rv;
}
// parse arguments of comp command
static int32_t runcomp(int argc, char** argv) {
bool argbrk = false;
const char* file = NULL;
int32_t mode = 0;
bool dec = false;
for (int32_t i = 2; i < argc; i++) {
if (!argbrk && argv[i][0] == '-') {
if (!std::strcmp(argv[i], "--")) {
argbrk = true;
} else if (!std::strcmp(argv[i], "-def")) {
mode = 1;
} else if (!std::strcmp(argv[i], "-gz")) {
mode = 2;
} else if (!std::strcmp(argv[i], "-lzo")) {
mode = 3;
} else if (!std::strcmp(argv[i], "-lzma")) {
mode = 4;
} else if (!std::strcmp(argv[i], "-d")) {
dec = true;
} else {
usage();
}
} else if (!file) {
argbrk = true;
file = argv[i];
} else {
usage();
}
}
int32_t rv = proccomp(file, mode, dec);
return rv;
}
// parse arguments of hash command
static int32_t runhash(int argc, char** argv) {
bool argbrk = false;
const char* file = NULL;
int32_t mode = 0;
for (int32_t i = 2; i < argc; i++) {
if (!argbrk && argv[i][0] == '-') {
if (!std::strcmp(argv[i], "--")) {
argbrk = true;
} else if (!std::strcmp(argv[i], "-fnv")) {
mode = 1;
} else if (!std::strcmp(argv[i], "-path")) {
mode = 2;
} else if (!std::strcmp(argv[i], "-crc")) {
mode = 3;
} else {
usage();
}
} else if (!file) {
argbrk = true;
file = argv[i];
} else {
usage();
}
}
int32_t rv = prochash(file, mode);
return rv;
}
// parse arguments of regex command
static int32_t runregex(int argc, char** argv) {
bool argbrk = false;
const char* pattern = NULL;
const char* file = NULL;
const char* alt = NULL;
int32_t opts = 0;
for (int32_t i = 2; i < argc; i++) {
if (!argbrk && argv[i][0] == '-') {
if (!std::strcmp(argv[i], "--")) {
argbrk = true;
} else if (!std::strcmp(argv[i], "-alt")) {
if (++i >= argc) usage();
alt = argv[i];
} else if (!std::strcmp(argv[i], "-ic")) {
opts |= kc::Regex::IGNCASE;
} else {
usage();
}
} else if (!pattern) {
argbrk = true;
pattern = argv[i];
} else if (!file) {
file = argv[i];
} else {
usage();
}
}
if (!pattern) usage();
int32_t rv = procregex(file, pattern, alt, opts);
return rv;
}
// parse arguments of conf command
static int32_t runconf(int argc, char** argv) {
bool argbrk = false;
int32_t mode = 0;
for (int32_t i = 2; i < argc; i++) {
if (!argbrk && argv[i][0] == '-') {
if (!std::strcmp(argv[i], "--")) {
argbrk = true;
} else if (!std::strcmp(argv[i], "-v")) {
mode = 'v';
} else if (!std::strcmp(argv[i], "-i")) {
mode = 'i';
} else if (!std::strcmp(argv[i], "-l")) {
mode = 'l';
} else if (!std::strcmp(argv[i], "-p")) {
mode = 'p';
} else {
usage();
}
} else {
argbrk = true;
usage();
}
}
int32_t rv = procconf(mode);
return rv;
}
// perform hex command
static int32_t prochex(const char* file, bool dec) {
const char* istr = file && *file == '@' ? file + 1 : NULL;
std::istream *is;
std::ifstream ifs;
std::istringstream iss(istr ? istr : "");
if (file) {
if (istr) {
is = &iss;
} else {
ifs.open(file, std::ios_base::in | std::ios_base::binary);
if (!ifs) {
eprintf("%s: %s: open error\n", g_progname, file);
return 1;
}
is = &ifs;
}
} else {
is = &std::cin;
}
if (dec) {
char c;
while (is->get(c)) {
int32_t cc = (unsigned char)c;
int32_t num = -1;
if (cc >= '0' && cc <= '9') {
num = cc - '0';
} else if (cc >= 'a' && cc <= 'f') {
num = cc - 'a' + 10;
} else if (cc >= 'A' && cc <= 'F') {
num = cc - 'A' + 10;
}
if (num >= 0) {
if (is->get(c)) {
cc = (unsigned char)c;
if (cc >= '0' && cc <= '9') {
num = num * 0x10 + cc - '0';
} else if (cc >= 'a' && cc <= 'f') {
num = num * 0x10 + cc - 'a' + 10;
} else if (cc >= 'A' && cc <= 'F') {
num = num * 0x10 + cc - 'A' + 10;
}
std::cout << (char)num;
} else {
std::cout << (char)num;
break;
}
}
}
if (istr) std::cout << std::endl;
} else {
bool mid = false;
char c;
while (is->get(c)) {
if (mid) std::cout << ' ';
int32_t cc = (unsigned char)c;
int32_t num = (cc >> 4);
if (num < 10) {
std::cout << (char)('0' + num);
} else {
std::cout << (char)('a' + num - 10);
}
num = (cc & 0x0f);
if (num < 10) {
std::cout << (char)('0' + num);
} else {
std::cout << (char)('a' + num - 10);
}
mid = true;
}
std::cout << std::endl;
}
return 0;
}
// perform enc command
static int32_t procenc(const char* file, int32_t mode, bool dec) {
const char* istr = file && *file == '@' ? file + 1 : NULL;
std::istream *is;
std::ifstream ifs;
std::istringstream iss(istr ? istr : "");
if (file) {
if (istr) {
is = &iss;
} else {
ifs.open(file, std::ios_base::in | std::ios_base::binary);
if (!ifs) {
eprintf("%s: %s: open error\n", g_progname, file);
return 1;
}
is = &ifs;
}
} else {
is = &std::cin;
}
std::ostringstream oss;
char c;
while (is->get(c)) {
oss.put(c);
}
const std::string& ostr = oss.str();
bool err = false;
switch (mode) {
default: {
if (dec) {
size_t zsiz;
char* zbuf = kc::basedecode(ostr.c_str(), &zsiz);
std::cout.write(zbuf, zsiz);
delete[] zbuf;
if (istr) std::cout << std::endl;
} else {
char* zbuf = kc::baseencode(ostr.data(), ostr.size());
std::cout << zbuf;
delete[] zbuf;
std::cout << std::endl;
}
break;
}
case 1: {
if (dec) {
size_t zsiz;
char* zbuf = kc::hexdecode(ostr.c_str(), &zsiz);
std::cout.write(zbuf, zsiz);
delete[] zbuf;
if (istr) std::cout << std::endl;
} else {
char* zbuf = kc::hexencode(ostr.data(), ostr.size());
std::cout << zbuf;
delete[] zbuf;
std::cout << std::endl;
}
break;
}
case 2: {
if (dec) {
size_t zsiz;
char* zbuf = kc::urldecode(ostr.c_str(), &zsiz);
std::cout.write(zbuf, zsiz);
delete[] zbuf;
if (istr) std::cout << std::endl;
} else {
char* zbuf = kc::urlencode(ostr.data(), ostr.size());
std::cout << zbuf;
delete[] zbuf;
std::cout << std::endl;
}
break;
}
case 3: {
if (dec) {
size_t zsiz;
char* zbuf = kc::quotedecode(ostr.c_str(), &zsiz);
std::cout.write(zbuf, zsiz);
delete[] zbuf;
if (istr) std::cout << std::endl;
} else {
char* zbuf = kc::quoteencode(ostr.data(), ostr.size());
std::cout << zbuf;
delete[] zbuf;
std::cout << std::endl;
}
break;
}
}
return err ? 1 : 0;
}
// perform ciph command
static int32_t procciph(const char* file, const char* key) {
const char* istr = file && *file == '@' ? file + 1 : NULL;
std::istream *is;
std::ifstream ifs;
std::istringstream iss(istr ? istr : "");
if (file) {
if (istr) {
is = &iss;
} else {
ifs.open(file, std::ios_base::in | std::ios_base::binary);
if (!ifs) {
eprintf("%s: %s: open error\n", g_progname, file);
return 1;
}
is = &ifs;
}
} else {
is = &std::cin;
}
std::ostringstream oss;
char c;
while (is->get(c)) {
oss.put(c);
}
const std::string& ostr = oss.str();
char* cbuf = new char[ostr.size()];
kc::arccipher(ostr.data(), ostr.size(), key, std::strlen(key), cbuf);
std::cout.write(cbuf, ostr.size());
delete[] cbuf;
return 0;
}
// perform comp command
static int32_t proccomp(const char* file, int32_t mode, bool dec) {
const char* istr = file && *file == '@' ? file + 1 : NULL;
std::istream *is;
std::ifstream ifs;
std::istringstream iss(istr ? istr : "");
if (file) {
if (istr) {
is = &iss;
} else {
ifs.open(file, std::ios_base::in | std::ios_base::binary);
if (!ifs) {
eprintf("%s: %s: open error\n", g_progname, file);
return 1;
}
is = &ifs;
}
} else {
is = &std::cin;
}
std::ostringstream oss;
char c;
while (is->get(c)) {
oss.put(c);
}
const std::string& ostr = oss.str();
bool err = false;
switch (mode) {
default: {
kc::ZLIB::Mode zmode;
switch (mode) {
default: zmode = kc::ZLIB::RAW; break;
case 1: zmode = kc::ZLIB::DEFLATE; break;
case 2: zmode = kc::ZLIB::GZIP; break;
}
if (dec) {
size_t zsiz;
char* zbuf = kc::ZLIB::decompress(ostr.data(), ostr.size(), &zsiz, zmode);
if (zbuf) {
std::cout.write(zbuf, zsiz);
delete[] zbuf;
} else {
eprintf("%s: decompression failed\n", g_progname);
err = true;
}
} else {
size_t zsiz;
char* zbuf = kc::ZLIB::compress(ostr.data(), ostr.size(), &zsiz, zmode);
if (zbuf) {
std::cout.write(zbuf, zsiz);
delete[] zbuf;
} else {
eprintf("%s: compression failed\n", g_progname);
err = true;
}
}
break;
}
case 3: {
kc::LZO::Mode zmode = kc::LZO::RAW;
if (dec) {
size_t zsiz;
char* zbuf = kc::LZO::decompress(ostr.data(), ostr.size(), &zsiz, zmode);
if (zbuf) {
std::cout.write(zbuf, zsiz);
delete[] zbuf;
} else {
eprintf("%s: decompression failed\n", g_progname);
err = true;
}
} else {
size_t zsiz;
char* zbuf = kc::LZO::compress(ostr.data(), ostr.size(), &zsiz, zmode);
if (zbuf) {
std::cout.write(zbuf, zsiz);
delete[] zbuf;
} else {
eprintf("%s: compression failed\n", g_progname);
err = true;
}
}
break;
}
case 4: {
kc::LZMA::Mode zmode = kc::LZMA::RAW;
if (dec) {
size_t zsiz;
char* zbuf = kc::LZMA::decompress(ostr.data(), ostr.size(), &zsiz, zmode);
if (zbuf) {
std::cout.write(zbuf, zsiz);
delete[] zbuf;
} else {
eprintf("%s: decompression failed\n", g_progname);
err = true;
}
} else {
size_t zsiz;
char* zbuf = kc::LZMA::compress(ostr.data(), ostr.size(), &zsiz, zmode);
if (zbuf) {
std::cout.write(zbuf, zsiz);
delete[] zbuf;
} else {
eprintf("%s: compression failed\n", g_progname);
err = true;
}
}
break;
}
}
return err ? 1 : 0;
}
// perform hash command
static int32_t prochash(const char* file, int32_t mode) {
const char* istr = file && *file == '@' ? file + 1 : NULL;
std::istream *is;
std::ifstream ifs;
std::istringstream iss(istr ? istr : "");
if (file) {
if (istr) {
is = &iss;
} else {
ifs.open(file, std::ios_base::in | std::ios_base::binary);
if (!ifs) {
eprintf("%s: %s: open error\n", g_progname, file);
return 1;
}
is = &ifs;
}
} else {
is = &std::cin;
}
std::ostringstream oss;
char c;
while (is->get(c)) {
oss.put(c);
}
const std::string& ostr = oss.str();
switch (mode) {
default: {
uint64_t hash = kc::hashmurmur(ostr.data(), ostr.size());
oprintf("%016llx\n", (unsigned long long)hash);
break;
}
case 1: {
uint64_t hash = kc::hashfnv(ostr.data(), ostr.size());
oprintf("%016llx\n", (unsigned long long)hash);
break;
}
case 2: {
char name[kc::NUMBUFSIZ];
uint32_t hash = kc::hashpath(ostr.data(), ostr.size(), name);
oprintf("%s\t%08lx\n", name, (unsigned long)hash);
break;
}
case 3: {
uint32_t hash = kc::ZLIB::calculate_crc(ostr.data(), ostr.size());
oprintf("%08x\n", (unsigned)hash);
break;
}
}
return 0;
}
// perform regex command
static int32_t procregex(const char* file, const char* pattern, const char* alt, int32_t opts) {
const char* istr = file && *file == '@' ? file + 1 : NULL;
std::istream *is;
std::ifstream ifs;
std::istringstream iss(istr ? istr : "");
if (file) {
if (istr) {
is = &iss;
} else {
ifs.open(file, std::ios_base::in | std::ios_base::binary);
if (!ifs) {
eprintf("%s: %s: open error\n", g_progname, file);
return 1;
}
is = &ifs;
}
} else {
is = &std::cin;
}
if (alt) {
kc::Regex regex;
if (!regex.compile(pattern, opts)) {
eprintf("%s: %s: compilation failed\n", g_progname, pattern);
return 1;
}
std::string altstr(alt);
std::string line;
while (mygetline(is, &line)) {
std::cout << regex.replace(line, altstr) << std::endl;
}
} else {
kc::Regex regex;
if (!regex.compile(pattern, opts | kc::Regex::MATCHONLY)) {
eprintf("%s: %s: compilation failed\n", g_progname, pattern);
return 1;
}
std::string line;
while (mygetline(is, &line)) {
if (regex.match(line)) std::cout << line << std::endl;
}
}
return 0;
}
// perform conf command
static int32_t procconf(int32_t mode) {
switch (mode) {
case 'v': {
oprintf("%s\n", kc::VERSION);
break;
}
case 'i': {
oprintf("%s\n", _KC_APPINC);
break;
}
case 'l': {
oprintf("%s\n", _KC_APPLIBS);
break;
}
case 'p': {
oprintf("%s\n", _KC_BINDIR);
break;
}
default: {
oprintf("VERSION: %s\n", kc::VERSION);
oprintf("LIBVER: %d\n", kc::LIBVER);
oprintf("LIBREV: %d\n", kc::LIBREV);
oprintf("FMTVER: %d\n", kc::FMTVER);
oprintf("OSNAME: %s\n", kc::OSNAME);
oprintf("BIGEND: %d\n", kc::BIGEND);
oprintf("CLOCKTICK: %d\n", kc::CLOCKTICK);
oprintf("PAGESIZ: %d\n", kc::PAGESIZ);
oprintf("FEATURES: %s\n", kc::FEATURES);
oprintf("TYPES: void*=%d short=%d int=%d long=%d long_long=%d size_t=%d"
" float=%d double=%d long_double=%d\n",
(int)sizeof(void*), (int)sizeof(short), (int)sizeof(int), (int)sizeof(long),
(int)sizeof(long long), (int)sizeof(size_t),
(int)sizeof(float), (int)sizeof(double), (int)sizeof(long double));
std::map<std::string, std::string> info;
kc::getsysinfo(&info);
if (info["mem_total"].size() > 0)
oprintf("MEMORY: total=%s free=%s cached=%s\n",
info["mem_total"].c_str(), info["mem_free"].c_str(),
info["mem_cached"].c_str());
if (std::strcmp(_KC_PREFIX, "*")) {
oprintf("prefix: %s\n", _KC_PREFIX);
oprintf("includedir: %s\n", _KC_INCLUDEDIR);
oprintf("libdir: %s\n", _KC_LIBDIR);
oprintf("bindir: %s\n", _KC_BINDIR);
oprintf("libexecdir: %s\n", _KC_LIBEXECDIR);
oprintf("appinc: %s\n", _KC_APPINC);
oprintf("applibs: %s\n", _KC_APPLIBS);
}
break;
}
}
return 0;
}
// END OF FILE