1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
/*-
* See the file LICENSE for redistribution information.
*
* Copyright (c) 2005, 2015 Oracle and/or its affiliates. All rights reserved.
*
* $Id$
*/
#include "csv.h"
#include "csv_local.h"
#include "csv_extern.h"
static int usage(void);
/*
* Globals
*/
DB_ENV *dbenv; /* Database environment */
DB *db; /* Primary database */
int verbose; /* Program verbosity */
char *progname; /* Program name */
int
main(int argc, char *argv[])
{
int ch, done, ret, t_ret;
char **clist, **clp, *home;
/* Initialize globals. */
dbenv = NULL;
db = NULL;
if ((progname = strrchr(argv[0], '/')) == NULL)
progname = argv[0];
else
++progname;
verbose = 0;
/* Initialize arguments. */
home = NULL;
ret = 0;
/* Allocate enough room for command-list arguments. */
if ((clp = clist =
(char **)calloc((size_t)argc + 1, sizeof(char *))) == NULL) {
fprintf(stderr, "%s: %s\n", progname, strerror(ENOMEM));
return (EXIT_FAILURE);
}
/* Process arguments. */
while ((ch = getopt(argc, argv, "c:h:v")) != EOF)
switch (ch) {
case 'c':
*clp++ = optarg;
break;
case 'h':
home = optarg;
break;
case 'v':
++verbose;
break;
case '?':
default:
return (usage());
}
argc -= optind;
argv += optind;
if (*argv != NULL)
return (usage());
/* Create or join the database environment. */
if (csv_env_open(home, 1) != 0)
return (EXIT_FAILURE);
/* Handle the queries. */
if (clp == clist)
ret = query_interactive();
else
for (clp = clist, done = 0; *clp != NULL && !done; ++clp)
if ((ret = query(*clp, &done)) != 0)
break;
/* Close the database environment. */
if ((t_ret = csv_env_close()) != 0 && ret == 0)
ret = t_ret;
return (ret == 0 ? EXIT_SUCCESS : EXIT_FAILURE);
}
/*
* usage --
* Program usage message.
*/
static int
usage(void)
{
(void)fprintf(stderr, "usage: %s [-v] [-c cmd] [-h home]\n", progname);
return (EXIT_FAILURE);
}
|