int main(int argc, char **argv) { int port = 0; char *user = NULL; char *passwd = NULL; char *host = NULL; char *dbname = NULL; int trace = 0; int describe = 0; int functions = 0; int useinserts = 0; int c; Mapi mid; int quiet = 0; stream *out; char user_set_as_flag = 0; char *table = NULL; static struct option long_options[] = { {"host", 1, 0, 'h'}, {"port", 1, 0, 'p'}, {"database", 1, 0, 'd'}, {"describe", 0, 0, 'D'}, {"functions", 0, 0, 'f'}, {"table", 1, 0, 't'}, {"inserts", 0, 0, 'N'}, {"Xdebug", 0, 0, 'X'}, {"user", 1, 0, 'u'}, {"quiet", 0, 0, 'q'}, {"help", 0, 0, '?'}, {0, 0, 0, 0} }; parse_dotmonetdb(&user, &passwd, NULL, NULL, NULL, NULL); while ((c = getopt_long(argc, argv, "h:p:d:Dft:NXu:q?", long_options, NULL)) != -1) { switch (c) { case 'u': if (user) free(user); user = strdup(optarg); user_set_as_flag = 1; break; case 'h': host = optarg; break; case 'p': assert(optarg != NULL); port = atoi(optarg); break; case 'd': dbname = optarg; break; case 'D': describe = 1; break; case 'N': useinserts = 1; break; case 'f': if (table) usage(argv[0], -1); functions = 1; break; case 't': if (table || functions) usage(argv[0], -1); table = optarg; break; case 'q': quiet = 1; break; case 'X': trace = MAPI_TRACE; break; case '?': /* a bit of a hack: look at the option that the current `c' is based on and see if we recognize it: if -? or --help, exit with 0, else with -1 */ usage(argv[0], strcmp(argv[optind - 1], "-?") == 0 || strcmp(argv[optind - 1], "--help") == 0 ? 0 : -1); default: usage(argv[0], -1); } } if (optind == argc - 1) dbname = argv[optind]; else if (optind != argc) usage(argv[0], -1); /* when config file would provide defaults */ if (user_set_as_flag) passwd = NULL; if (user == NULL) user = simple_prompt("user", BUFSIZ, 1, prompt_getlogin()); if (passwd == NULL) passwd = simple_prompt("password", BUFSIZ, 0, NULL); mid = mapi_connect(host, port, user, passwd, "sql", dbname); if (user) free(user); if (passwd) free(passwd); if (mid == NULL) { fprintf(stderr, "failed to allocate Mapi structure\n"); exit(2); } if (mapi_error(mid)) { mapi_explain(mid, stderr); exit(2); } if (!quiet) { char *motd = mapi_get_motd(mid); if (motd) fprintf(stderr, "%s", motd); } mapi_trace(mid, trace); mapi_cache_limit(mid, 10000); out = file_wastream(stdout, "stdout"); if (out == NULL) { fprintf(stderr, "failed to allocate stream\n"); exit(2); } if (!quiet) { char buf[27]; time_t t = time(0); char *p; #ifdef HAVE_CTIME_R3 ctime_r(&t, buf, sizeof(buf)); #else #ifdef HAVE_CTIME_R ctime_r(&t, buf); #else strncpy(buf, ctime(&t), sizeof(buf)); #endif #endif if ((p = strrchr(buf, '\n')) != NULL) *p = 0; mnstr_printf(out, "-- msqldump %s %s%s %s\n", describe ? "describe" : "dump", functions ? "functions" : table ? "table " : "database", table ? table : "", buf); dump_version(mid, out, "--"); } if (functions) c = dump_functions(mid, out, NULL, NULL); else if (table) c = dump_table(mid, NULL, table, out, describe, 1, useinserts); else c = dump_database(mid, out, describe, useinserts); mnstr_flush(out); mapi_destroy(mid); if (mnstr_errnr(out)) { fprintf(stderr, "%s: %s", argv[0], mnstr_error(out)); return 1; } mnstr_destroy(out); return c; }
int main(int argc, char **argv) { ssize_t n; size_t len, buflen; char *host = NULL; int portnr = 0; char *uri = NULL; char *user = NULL; char *password = NULL; char buf[BUFSIZ], *buffer, *e, *response; int done = 0; EventRecord event; static struct option long_options[15] = { { "dbname", 1, 0, 'd' }, { "user", 1, 0, 'u' }, { "port", 1, 0, 'p' }, { "password", 1, 0, 'P' }, { "host", 1, 0, 'h' }, { "help", 0, 0, '?' }, { "output", 1, 0, 'o' }, { "queries", 1, 0, 'q' }, { "debug", 0, 0, 'D' }, { 0, 0, 0, 0 } }; /* parse config file first, command line options override */ parse_dotmonetdb(&user, &password, &dbname, NULL, NULL, NULL, NULL); while (1) { int option_index = 0; int c = getopt_long(argc, argv, "d:u:p:P:h:?:o:q:D", long_options, &option_index); if (c == -1) break; switch (c) { case 'D': debug = 1; break; case 'd': if (dbname) free(dbname); dbname = strdup(optarg); break; case 'u': if (user) free(user); user = strdup(optarg); /* force password prompt */ if (password) free(password); password = NULL; break; case 'P': if (password) free(password); password = strdup(optarg); break; case 'p': if (optarg) portnr = atoi(optarg); break; case 'q': if (optarg) querypool = atoi(optarg) > 0? atoi(optarg):1; break; case 'h': host = optarg; break; case '?': usageTachograph(); /* a bit of a hack: look at the option that the current `c' is based on and see if we recognize it: if -? or --help, exit with 0, else with -1 */ exit(strcmp(argv[optind - 1], "-?") == 0 || strcmp(argv[optind - 1], "--help") == 0 ? 0 : -1); default: usageTachograph(); exit(-1); } } if( dbname == NULL){ usageTachograph(); exit(-1); } if (dbname != NULL && strncmp(dbname, "mapi:monetdb://", 15) == 0) { uri = dbname; dbname = NULL; } #ifdef SIGPIPE signal(SIGPIPE, stopListening); #endif #ifdef SIGHUP signal(SIGHUP, stopListening); #endif #ifdef SIGQUIT signal(SIGQUIT, stopListening); #endif signal(SIGINT, stopListening); signal(SIGTERM, stopListening); close(0); if (user == NULL) user = simple_prompt("user", BUFSIZ, 1, prompt_getlogin()); if (password == NULL) password = simple_prompt("password", BUFSIZ, 0, NULL); /* our hostname, how remote servers have to contact us */ gethostname(hostname, sizeof(hostname)); /* set up the profiler */ if (uri) dbh = mapi_mapiuri(uri, user, password, "mal"); else dbh = mapi_mapi(host, portnr, user, password, "mal", dbname); if (dbh == NULL || mapi_error(dbh)) die(dbh, hdl); mapi_reconnect(dbh); if (mapi_error(dbh)) die(dbh, hdl); host = strdup(mapi_get_host(dbh)); if(debug) fprintf(stderr,"-- connection with server %s\n", uri ? uri : host); snprintf(buf,BUFSIZ-1,"profiler.setheartbeat(0);"); if( debug) fprintf(stderr,"-- %s\n",buf); doQ(buf); snprintf(buf, BUFSIZ, " profiler.openstream(0);"); if( debug) fprintf(stderr,"-- %s\n",buf); doQ(buf); len = 0; buflen = BUFSIZ; buffer = (char *) malloc(buflen); if( buffer == NULL){ fprintf(stderr,"Could not create input buffer\n"); exit(-1); } conn = mapi_get_from(dbh); while ((n = mnstr_read(conn, buffer + len, 1, buflen - len-1)) >= 0) { if (n == 0 && (n = mnstr_read(conn, buffer + len, 1, buflen - len-1)) <= 0) break; buffer[len + n] = 0; response = buffer; while ((e = strchr(response, '\n')) != NULL) { *e = 0; if(debug) printf("%s\n", response); done= keyvalueparser(response, &event); if( done == 1){ update(&event); } else if( done == 0){ if (debug ) fprintf(stderr, "PARSE %d:%s\n", done, response); } response = e + 1; } /* handle the case that the line is too long to * fit in the buffer */ if( response == buffer){ char *new = (char *) realloc(buffer, buflen + BUFSIZ); if( new == NULL){ fprintf(stderr,"Could not extend input buffer\n"); exit(-1); } buffer = new; buflen += BUFSIZ; len += n; } /* handle the case the buffer contains more than one * line, and the last line is not completely read yet. * Copy the first part of the incomplete line to the * beginning of the buffer */ else if (*response) {