dav/xattrtool.c

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright 2019 Olaf Wintermann. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *   1. Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 *
 *   2. 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS 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 COPYRIGHT HOLDER OR 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.
 */

/*
 * small tool for accessing extended attributes
 */

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

#ifdef _WIN32
#include <Windows.h>
#endif

#include "libxattr.h"

int attrtool_list(int argc, char **argv, int values);
int attrtool_get(int argc, char **argv, int raw);
int attrtool_set(int argc, char **argv);
int attrtool_rm(int argc, char **argv);

void print_usage(char *cmd) {
    fprintf(stderr, "usage %s:\n", cmd);
    fprintf(stderr, "   list <file>\n");
    fprintf(stderr, "   listvalues <file>\n");
    fprintf(stderr, "   get <file> <name>\n");
    fprintf(stderr, "   set <file> <name> <value>\n");
    fprintf(stderr, "   remove <file> <name>\n");
}

int xattr_tool_main(int argc, char **argv);

#ifdef _WIN32
static char* wchar2utf8(const wchar_t *wstr, size_t wlen) {
    size_t maxlen = wlen * 4;
    char *ret = malloc(maxlen + 1);
    int ret_len = WideCharToMultiByte(
        CP_UTF8,
        0,
        wstr,
        wlen,
        ret,
        maxlen,
        NULL,
        NULL);
    ret[ret_len] = 0;
    return ret;
}

int wmain(int argc, wchar_t **argv) {
    char **argv_utf8 = calloc(argc, sizeof(char*));
    for(int i=0;i<argc;i++) {
        argv_utf8[i] = wchar2utf8(argv[i], wcslen(argv[i]));
    }
    
    int ret = xattr_tool_main(argc, argv_utf8);
    
    for(int i=0;i<argc;i++) {
        free(argv_utf8[i]);
    }
    free(argv_utf8);
    
    return ret;
}
#else
int main(int argc, char **argv) {
    return xattr_tool_main(argc, argv);
}
#endif

int xattr_tool_main(int argc, char **argv) {
    if(argc < 3) {
        print_usage(argv[0]);
        return 1;
    }
    
    if(!strcmp(argv[1], "list") || !strcmp(argv[1], "ls")) {
        return attrtool_list(argc, argv, 0);
    } else if(!strcmp(argv[1], "listvalues") || !strcmp(argv[1], "lsv")) {
        return attrtool_list(argc, argv, 1);
    } else if(!strcmp(argv[1], "get")) {
        return attrtool_get(argc, argv, 0);
    } else if(!strcmp(argv[1], "set")) {
        return attrtool_set(argc, argv);
    } else if(!strcmp(argv[1], "rm") || !strcmp(argv[1], "remove")) {
        return attrtool_rm(argc, argv);
    } else {
        fprintf(stderr, "Unknown command\n");
        print_usage(argv[0]);
    }
    
    return 1;
}

int attrtool_list(int argc, char **argv, int values) {
    ssize_t nelm = 0;
    char **list = xattr_list(argv[2], &nelm);
    if(nelm < 0) {
        perror("xattr_list");
        return 3;
    }
    
    for(int i=0;i<nelm;i++) {
        char *attr = list[i];
        if(values) {
            ssize_t valuelen = 0;
            char *value = xattr_get(argv[2], attr, &valuelen);
            if(valuelen < 0) {
                printf("%s: #ERROR: %s\n", attr, strerror(errno));
            } else {
                printf("%s: %.*s\n", attr, (int)valuelen, value);
            }
            
            if(value) {
                free(value);
            }
        } else {
            printf("%s\n", attr);
        }
        
        free(list[i]);
    }
    if(list) {
        free(list);
    }
    
    return 0;
}

int attrtool_get(int argc, char **argv, int raw) {
    if(argc < 4) {
        fprintf(stderr, "Too few arguments\n");
        print_usage(argv[1]);
        return 1;
    }
    
    char *file = argv[2];
    char *attr = argv[3];
    int ret = 0;
    
    ssize_t valuelen = 0;
    char *value = xattr_get(file, attr, &valuelen);
    if(value) {
        if(raw) {
            fwrite(value, 1, valuelen, stdout);
        } else {
            printf("%.*s\n", (int)valuelen, value);
        }
        free(value);
    } else {
        perror("xattr_get");
        ret = 3;
    }
    
    return ret;
}

int attrtool_set(int argc, char **argv) {
    if(argc < 4) {
        fprintf(stderr, "Too few arguments\n");
        print_usage(argv[1]);
        return 1;
    }
    
    size_t len = strlen(argv[4]);
    char *file = argv[2];
    char *attr = argv[3];
    char *value = argv[4];
    int ret = 0;
    
    if(xattr_set(file, attr, value, len)) {
        perror("xattr_set");
        ret = 3;
    }
    
    return ret;
}

int attrtool_rm(int argc, char **argv) {
    if(argc < 4) {
        fprintf(stderr, "Too few arguments\n");
        print_usage(argv[1]);
        return 1;
    }
    
    char *file = argv[2];
    char *attr = argv[3];
    int ret = 0;
    
    if(xattr_remove(file, attr)) {
        perror("xattr_remove");
        ret = 1;
    }
    
    return ret;
}