--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/main.c Tue Sep 13 22:13:02 2011 -0700
@@ -0,0 +1,130 @@
+/* vim: set noet nosta sw=4 ts=4 ft=c : */
+/*
+Copyright (c) 2011, Mahlon E. Smith <mahlon@martini.nu>
+All rights reserved.
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+ * Redistributions of source code must retain the above copyright
+ notice, this list of conditions and the following disclaimer.
+
+ * Redistributions in binary form must reproduce the above copyright
+ notice, this list of conditions and the following disclaimer in the
+ documentation and/or other materials provided with the distribution.
+
+ * Neither the name of Mahlon E. Smith nor the names of his
+ contributors may be used to endorse or promote products derived
+ from this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
+EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
+DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+*/
+
+/*
+ * TODO
+ *
+ * empty struct not necessary?
+ * inet_pton( AF_INET, *char src, dest )
+ * an option to run the DB out of memory?
+ * PRAGMA user_version = 1;
+ *
+ */
+
+#include "volta.h"
+unsigned short int debugmode;
+
+
+/*
+ * Parse command line options, perform actions, and enter accept loop.
+ *
+ */
+int
+main( int argc, char *argv[] )
+{
+ /* opt action flags */
+ struct {
+ unsigned char init;
+ unsigned char dump;
+ } actions = {0};
+
+#ifdef DEBUG
+ /* debugmode set at compile time,
+ * default to display everything */
+ debugmode = 99;
+#else
+ debugmode = 0;
+#endif
+
+ /* get_opt vars */
+ int opt = 0;
+ opterr = 0;
+
+ /* parse options */
+ while ( (opt = getopt( argc, argv, "a:d:hv" )) != -1 ) {
+ switch ( opt ) {
+
+ /* action */
+ case 'a':
+ if ( strcmp( optarg, "init" ) == 0 ) actions.init++;
+ if ( strcmp( optarg, "dump" ) == 0 ) actions.dump++;
+ break;
+
+ /* debug */
+ case 'd':
+ if ( optarg[2] == '-' ) {
+ argc++; argv -= 1;
+ debugmode = 1;
+ }
+ sscanf( optarg, "%hu", &debugmode );
+ break;
+
+ /* help */
+ case 'h':
+ usage( argv[0] );
+ return( 0 );
+
+ /* version */
+ case 'v':
+ printf( "%s %s\n", PROG, VERSION );
+ return( 0 );
+
+ /* unknown option or option argument missing */
+ case '?':
+ switch( optopt ) {
+ case 'd': /* no debug argument, default to level 1 */
+ debugmode = 1;
+ break;
+ default:
+ usage( argv[0] );
+ return( 1 );
+ }
+
+ default:
+ break;
+ }
+ }
+ argc -= optind;
+ argv += optind;
+
+ /* perform any requested actions */
+ if ( actions.init ) {
+ db_initialize();
+ return( 0 );
+ }
+ if ( actions.dump ) {
+ debug( 1, LOC, "dump.\n" );
+ return( 0 );
+ }
+
+ /* enter stdin parsing loop */
+ return accept_loop();
+}
+