[PATCH] Implement support for command line argument parsing using <argp.h>

Sergio Durigan Junior sergiodj en sergiodj.net
Sab Mar 15 20:11:12 UTC 2014


Oi de novo :-),

Eu sou um fã da <argp.h>, uso ela em vários projetos pessoais, e achei
que fosse uma boa adição ao rnetclient.  Eu optei por criar um switch
chamado -d|--declaration pra que o usuário possa passar o nome do
arquivo da declaração, mas também mantive a compatibilidade com versões
anteriores que aceitam o arquivo passado diretamente (i.e., sem switches
nem nada).

O patch é simples, não tem muito truque, mas reviews são obviamente bem
vindos.  Queria vê-lo incluído antes do release de hoje :-).

Falou!

-- 
Sergio

commit 28a8c5807a9fdfff747d9a8e95d6cb986f56e681
Author: Sergio Durigan Junior <sergiodj en sergiodj.net>
Date:   Sat Mar 15 16:28:39 2014 -0300

    Implement argument parsing via argp.h.
    
    This commit implements argument parsing from the command line via
    argp.h. So far, I have added only one parameter ('-d', or
    '--declaracao') which is used for providing the filename of the
    declaration.  I have also marked this argument as optional, so that the
    user will still be able to provide the filename without using '-d'.
    
    As an auxiliary support, I enabled the generation of the config.h header
    file, so that the program can obtain the package version and the bug
    report address directly from configure.ac.

diff --git a/.gitignore b/.gitignore
index c037aae..c62646f 100644
--- a/.gitignore
+++ b/.gitignore
@@ -8,6 +8,7 @@ build-aux
 config.log
 config.status
 configure
+config.h*
 
 # Object files
 *.o
diff --git a/bootstrap.sh b/bootstrap.sh
index 41e9823..1bf9968 100755
--- a/bootstrap.sh
+++ b/bootstrap.sh
@@ -10,5 +10,6 @@ AUX_DIR=build-aux
 test -d $AUX_DIR || mkdir -p $AUX_DIR
 
 aclocal
+autoheader
 autoconf
 automake --add-missing --copy --force --foreign
diff --git a/configure.ac b/configure.ac
index c7a85dc..e06ecdf 100644
--- a/configure.ac
+++ b/configure.ac
@@ -7,4 +7,5 @@ PKG_CHECK_MODULES(GNUTLS, gnutls >= 1.4.0, , AC_MSG_ERROR(Could not find gnutls)
 AM_PATH_LIBGCRYPT(,,AC_MSG_ERROR(Could not find gcrypt))
 LIBS="$LIBGCRYPT_LIBS $GNUTLS_LIBS $LIBS -lz"
 CFLAGS="$LIBGCRYPT_CFLAGS $GNUTLS_CFLAGS $CFLAGS"
+AC_CONFIG_HEADERS([config.h])
 AC_OUTPUT(Makefile)
diff --git a/rnetclient.c b/rnetclient.c
index d264c72..eb5a391 100644
--- a/rnetclient.c
+++ b/rnetclient.c
@@ -27,10 +27,81 @@
 #include <netdb.h>
 #include <gnutls/gnutls.h>
 #include <zlib.h>
+#include <argp.h>
+#include "config.h"
 #include "decfile.h"
 #include "rnet_message.h"
 #include "rnet_encode.h"
 
+/* Program version and bug report address.  */
+
+const char *argp_program_version = PACKAGE_VERSION;
+const char *argp_program_bug_address = PACKAGE_BUGREPORT;
+
+/* Documentation strings.  */
+
+static const char rnetclient_doc[] =
+	"Send the Brazilian Income Tax Declaration to the Brazilian "
+	"Tax Authority";
+static const char rnetclient_args_doc[] =
+	"[-d|--declaration] FILE";
+
+/* Description and definition of each option accepted by the program.  */
+
+static const struct argp_option rnetclient_options_desc[] = {
+	{ "declaration", 'd', "FILE", OPTION_ARG_OPTIONAL,
+	  "The Income Tax Declaration file that will be sent.",
+	  0 },
+
+	{ NULL },
+};
+
+struct rnetclient_args {
+	/* File representing the declaration.  */
+	char *file;
+};
+
+/* Parser for command line arguments.  */
+
+static error_t rnetclient_parse_opt(int key, char *arg, struct argp_state *state)
+{
+	struct rnetclient_args *a = state->input;
+	switch (key) {
+	case 'd':
+		/* The user has explicitly provided a filename through
+		   the '-d' switch.  */
+		a->file = arg;
+		break;
+
+	case ARGP_KEY_ARG:
+		/* The user has possibly provided a filename without
+		   using any switches (e.g., by running './rnetclient
+		   file').  */
+		a->file = arg;
+		break;
+
+	case ARGP_KEY_END:
+		/* We have reached the end of the argument parsing.
+		   Let's check if the user has provided a filename.  */
+		if (arg == NULL && a->file == NULL)
+			argp_error(state,
+				   "You need to provide the Income Tax Declaration "
+				   "filename.");
+	}
+
+	return 0;
+}
+
+/* Control struct used by argp.  */
+
+static struct argp rnetclient_argp = {
+	rnetclient_options_desc,
+	rnetclient_parse_opt,
+	rnetclient_args_doc,
+	rnetclient_doc,
+	NULL, NULL, NULL
+};
+
 static size_t chars2len (unsigned char buf[2]) {
 	return (buf[0] << 8 | buf[1]);
 }
@@ -172,12 +243,6 @@ static int handshake(int c)
 	return 0;
 }
 
-static void usage(void)
-{
-	fprintf(stderr, "rnetclient [filename]\n");
-	exit(1);
-}
-
 static int rnet_send(gnutls_session_t session, char *buffer, size_t len, int header)
 {
 	int r = 0;
@@ -329,17 +394,17 @@ int main(int argc, char **argv)
 	int r;
 	struct rnet_decfile *decfile;
 	struct rnet_message *message = NULL;
+	struct rnetclient_args rnet_args;
 	gnutls_session_t session;
 	int finish = 0;
 	char *cpf;
 	
-	if (argc < 2) {
-		usage();
-	}
+	memset (&rnet_args, 0, sizeof (rnet_args));
+	argp_parse (&rnetclient_argp, argc, argv, 0, NULL, &rnet_args);
 
-	decfile = rnet_decfile_open(argv[1]);
+	decfile = rnet_decfile_open(rnet_args.file);
 	if (!decfile) {
-		fprintf(stderr, "could not parse %s: %s\n", argv[1], strerror(errno));
+		fprintf(stderr, "could not parse file \"%s\": %s\n", rnet_args.file, strerror(errno));
 		exit(1);
 	}
 


Más información sobre la lista de distribución Softwares-impostos