sockdrive.c File Reference

#include "mbconfig.h"
#include <stdio.h>
#include <stdlib.h>
#include "ac/getopt.h"
#include <errno.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <sys/un.h>
#include <netdb.h>
#include <string.h>
#include <signal.h>
#include "sock.h"

Include dependency graph for sockdrive.c:

Go to the source code of this file.

Functions
static void	usage (void)
int	main (int argc, char *argv[])

Variables
const unsigned short int	PORT = 5555
const char *	SERVERHOST = "localhost"
const char *	SERVERPATH = "/var/mbdyn/mbdyn.sock"
static int	sasl = 0

Function Documentation

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 95 of file sockdrive.c.

References buf, getopt(), host, mbdyn_make_inet_socket(), mbdyn_make_named_socket(), optarg, optind, path, PORT, port, SERVERHOST, and usage().

{
        int sock;

        char *path = NULL;
        char *host = (char *)SERVERHOST;
        unsigned short int port = PORT;

        char *user = NULL;
        char *cred = NULL;
        char *mech = NULL;

        int inc = 0;
        int imp = 0;
        char *label;
        char *value = NULL;

        FILE *fd;


        while (1) {
                int opt;

                opt = getopt (argc, argv, "D:h:i:I:m:p:P:S:w:W");

                if (opt == EOF) {
                        break;
                }

                switch (opt) {
                case 'D':
                        user = optarg;
                        break;

                case 'h':
                        host = optarg;
                        break;

                case 'i':
                        if (strcasecmp(optarg, "yes") == 0) {
                                inc = 1;
                        } else if (strcasecmp(optarg, "no") == 0) {
                                inc = -1;
                        }
                        break;

                case 'I':
                        if (strcasecmp(optarg, "yes") == 0) {
                                imp = 1;
                        } else if (strcasecmp(optarg, "no") == 0) {
                                imp = -1;
                        }
                        break;

                case 'm':
                        mech = optarg;
#ifndef HAVE_SASL2
                        fprintf(stderr, "SASL not supported\n");
#endif /* ! HAVE_SASL2 */
                        break;

                case 'p':
                        port = atoi(optarg);
                        break;

                case 'P':
                        path = optarg;
                        break;

                case 'S':
                        sasl++;
#ifndef HAVE_SASL2
                        fprintf(stderr, "SASL not supported\n");
                        exit(EXIT_FAILURE);
#endif /* ! HAVE_SASL2 */
                        break;

                case 'w':
                        cred = strdup(optarg);
                        break;

                case 'W':
                        if (cred) {
                                free(cred);
                        }

                        if (user) {
                                char buf[1024];

                                snprintf(buf, sizeof(buf), "Password for user \"%s\": ", user);
                                cred = getpass(buf);

                        } else {
                                cred = getpass("Password: ");
                        }

                        if (cred) {
                                cred = strdup(cred);
                        }
                        break;
                }
        }

        if ((argc - optind) < 1) {
                usage();
                exit(EXIT_SUCCESS);
        }
        label = argv[optind];

        if ((argc - optind) > 1) {
                value = argv[optind + 1];
        }

        if (path) {
                sock = mbdyn_make_named_socket(0, path, 0, NULL);
        } else {
                sock = mbdyn_make_inet_socket(0, host, port, 0, NULL);
        }
        if (sock < 0) {
                fprintf(stderr, "socket initialization error\n");
                exit(EXIT_FAILURE);
        }

        if (sasl) {
#ifdef HAVE_SASL2
                printf("initializing SASL data...\n");

                mbdyn_sasl.use_sasl = MBDYN_SASL_CLIENT;
                mbdyn_sasl.sasl_flags = MBDYN_SASL_FLAG_CRITICAL | MBDYN_SASL_FLAG_USERAUTHZ;
                mbdyn_sasl.sasl_mech = mech;
                mbdyn_sasl.sasl_user = user;
                mbdyn_sasl.sasl_cred = cred;
                mbdyn_sasl.sasl_hostname = host;

                if (mbdyn_sasl_client_init(&mbdyn_sasl) != SASL_OK) {
                        fprintf(stderr, "SASL init failed\n");
                        exit(EXIT_FAILURE);
                }

                if (mbdyn_sasl_client_auth(sock, NULL, &mbdyn_sasl) != SASL_OK) {
                        fprintf(stderr, "SASL auth failed\n");
                        exit(EXIT_FAILURE);
                }
#endif /* HAVE_SASL2 */
        }

        fd = fdopen(sock, "w");

        if (!sasl) {
                if (user) {
                        fprintf(fd, "user: %s\n", user);
                        if (cred) {
                                fprintf(fd, "password: %s\n", cred);
                                memset(cred, '\0', strlen(cred));
                        }
                }
        }

        fprintf(fd, "label: %s\n", label);

        if (inc == 1) {
                fprintf(fd, "inc: yes\n");
        } else if (inc == -1) {
                fprintf(fd, "inc: no\n");
        }

        if (imp == 1) {
                fprintf(fd, "imp: yes\n");
        } else if (imp == -1) {
                fprintf(fd, "imp: no\n");
        }

        if (value != NULL) {
                fprintf(fd, "value: %s\n", value);
        }

        fprintf(fd, ".\n");

        fclose(fd);

        exit(EXIT_SUCCESS);
}

Here is the call graph for this function:

static void usage ( void  )

static

Definition at line 66 of file sockdrive.c.

Referenced by main().

{
        fprintf(stderr,
                "\n\tusage: sockdrive [h:p:D:w:Wi:I:] <label> [<value>]\n\n"
                "\t\t-D <user>\tuser name\n"
                "\t\t-h <host>\thost name\n"
                "\t\t-i {yes|no}\tincremental (i.e. value[<label>] += <value>)\n"
                "\t\t-I {yes|no}\timpulsive (reset after one step)\n"
                "\t\t-m <mech>\tSASL mechanism (needs -S)\n"
                "\t\t-p <port>\tport number\n"
                "\t\t-P <path>\tpath of named pipe\n"
                "\t\t-S\t\tenable SASL auth"
#ifndef HAVE_SASL2
                        " (not supported)"
#endif /* ! HAVE_SASL2 */
                "\n"
                "\t\t-w <cred>\tuser credentials\n"
                "\t\t-W\t\tprompt for user credentials\n"
                "\t\t<label>:\tfile drive (base 1) index to modify\n"
                "\n"
                "\t\t<value>:\tnew value (or increment if -i)\n\n");
}

Variable Documentation

const unsigned short int PORT = 5555

Definition at line 61 of file sockdrive.c.

Referenced by main().

int sasl = 0

static

Definition at line 89 of file sockdrive.c.

const char* SERVERHOST = "localhost"

Definition at line 62 of file sockdrive.c.

Referenced by main().

const char* SERVERPATH = "/var/mbdyn/mbdyn.sock"

Definition at line 63 of file sockdrive.c.