wmii/cmd/wmiir2.c

/*
 * (C)opyright MMIV-MMV Anselm R. Garbe <garbeam at gmail dot com>
 * See LICENSE file for license details.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "../libixp2/ixp.h"

#include <cext.h>

typedef struct {
    char *name;
    int (*cmd) (char **argv);
    int min_argc;
} Command;

static int xcreate(char **argv);
static int xread(char **argv);
static int xwrite(char **argv);
static int xremove(char **argv);
static Command cmds[] = {
    {"create", xcreate, 2},
    {"read", xread, 1},
    {"write", xwrite, 2},
    {"remove", xremove, 1},
    {0, 0}
};
static IXPClient c = { 0 };

static char *version[] = {
    "wmiir - window manager improved remote - " VERSION "\n"
        " (C)opyright MMIV-MMV Anselm R. Garbe\n", 0
};

static void
usage()
{
    fprintf(stderr, "%s",
            "usage: wmiir [-a <server address>] [-v]\n"
            "      -a    server address (default: $WMIIR_SOCKET)\n"
            "      -v    version info\n"
            "commands read from stdin:\n"
            "      create <file> <content>   -- creates file\n"
            "      read   <file/directory>   -- reads file/directory contents\n"
            "      write  <file> <content>   -- writes content to a file\n"
            "      remove <file/directory>   -- removes file\n");
    exit(1);
}

static unsigned int
write_data(unsigned int fid, unsigned char *data, unsigned int count)
{
    unsigned int len, i, runs = count / c.fcall.iounit;

    for(i = 0; i <= runs; i++) {
        /* write */
        len = count - (i * c.fcall.iounit);
        if(len > c.fcall.iounit)
            len = c.fcall.iounit;
        if(ixp_client_write
           (&c, fid, i * c.fcall.iounit, len,
            &data[i * c.fcall.iounit]) != count) {
            fprintf(stderr, "wmiir: cannot write file: %s\n", c.errstr);
            return 0;
        }
    }
    return count;
}

static int
xcreate(char **argv)
{
    unsigned int fid;
    char *p = strrchr(argv[0], '/');

    fid = c.root_fid << 2;
    /* walk to bottom-most directory */
    *p = 0;
    if(!ixp_client_walk(&c, fid, argv[0])) {
        fprintf(stderr, "wmiir: cannot walk to %s: %s\n", argv[0],
                c.errstr);
        return 1;
    }
    /* create */
    p++;
    if(!ixp_client_create(&c, fid, p, (unsigned int) 0xff, IXP_OWRITE)) {
        fprintf(stderr, "wmiir: cannot create file: %s\n", c.errstr);
        return 1;
    }
    write_data(fid, (unsigned char *) argv[1], strlen(argv[1]));
    return !ixp_client_close(&c, fid);
}

static int
xwrite(char **argv)
{
    /* open */
    unsigned int fid = c.root_fid << 2;
    if(!ixp_client_open(&c, fid, argv[0], IXP_OWRITE)) {
        fprintf(stderr, "wmiir: cannot open file: %s\n", c.errstr);
        return 1;
    }
    write_data(fid, (unsigned char *) argv[1], strlen(argv[1]));
    return !ixp_client_close(&c, fid);
}

static void
print_directory(void *result, unsigned int msize)
{
    void *p = result;
    static Stat stat, zerostat = { 0 };
    unsigned int len = 0;
    do {
        p = ixp_dec_stat(p, &stat);
        len += stat.size + sizeof(unsigned short);
        if(stat.qid.type == IXP_QTDIR)
            fprintf(stdout, "%s/\n", stat.name);
        else
            fprintf(stdout, "%s\n", stat.name);
        stat = zerostat;
    }
    while(len < msize);
}

static int
xread(char **argv)
{
    unsigned int count, fid = c.root_fid << 2;
    int is_directory = FALSE;
    static unsigned char result[IXP_MAX_MSG];

    /* open */
    if(!ixp_client_open(&c, fid, argv[0], IXP_OREAD)) {
        fprintf(stderr, "wmiir: cannot open file: %s\n", c.errstr);
        return 1;
    }
    is_directory = !c.fcall.nwqid || (c.fcall.qid.type == IXP_QTDIR);
    /* read */
    if(!(count = ixp_client_read(&c, fid, 0, result, IXP_MAX_MSG))
       && c.errstr) {
        fprintf(stderr, "wmiir: cannot read file: %s\n", c.errstr);
        return 1;
    }
    if(count) {
        if(is_directory)
            print_directory(result, count);
        else {
            unsigned int i;
            for(i = 0; i < count; i++)
                putchar(result[i]);
        }
    }
    return !ixp_client_close(&c, fid);
}

static int
xremove(char **argv)
{
    unsigned int fid;

    /* remove */
    fid = c.root_fid << 2;
    if(!ixp_client_remove(&c, fid, argv[0])) {
        fprintf(stderr, "wmiir: cannot remove file: %s\n", c.errstr);
        return 1;
    }
    return 0;
}

static int
perform_cmd(int argc, char **argv)
{
    int i;
    for(i = 0; cmds[i].name; i++)
        if(!strncmp(cmds[i].name, argv[0], strlen(cmds[i].name))) {
            if(cmds[i].min_argc <= argc)
                return cmds[i].cmd(&argv[1]);
            else
                usage();
        }
    /* bogus command */
    return 1;
}

int
main(int argc, char *argv[])
{
    int i = 0, ret;
    char line[4096];
    char *sockfile = getenv("WMIIR_SOCKET");
    char *stdin_argv[3];
    int stdin_argc;

    /* command line args */
    if(argc > 1) {
        for(i = 1; (i < argc) && (argv[i][0] == '-'); i++) {
            switch (argv[i][1]) {
            case 'v':
                fprintf(stderr, "%s", version[0]);
                exit(0);
                break;
            case 'a':
                if(i + 1 < argc)
                    sockfile = argv[++i];
                else
                    usage();
                break;
            default:
                usage();
                break;
            }
        }
    }
    if(!sockfile) {
        fprintf(stderr, "%s",
                "wmiir: error: WMIIR_SOCKET environment not set\n");
        usage();
    }
    /* open socket */
    if(!ixp_client_init(&c, sockfile)) {
        fprintf(stderr, "wmiir: %s\n", c.errstr);
        exit(1);
    }
    /* wether perform directly or read from stdin */
    while(fgets(line, 4096, stdin))
        if((stdin_argc = cext_tokenize(stdin_argv, 3, line, ' '))) {
            if((ret = perform_cmd(stdin_argc, stdin_argv)))
                break;
        }

    /* close socket */
    ixp_client_deinit(&c);

    return ret;
}