libopkg: make checksum_bin2hex() and checksum_hex2bin() cope with NULL
[oweals/opkg-lede.git] / libopkg / file_util.c
index f5519c85d58a3cb685481e0db757ef39feaa2a8a..b1d793023154ff5f7da0f9ef76efc6601454b718 100644 (file)
@@ -1,7 +1,8 @@
 /* file_util.c - convenience routines for common stat operations
 
-   Carl D. Worth
+   Copyright (C) 2009 Ubiq Technologies <graham.gower@gmail.com>
 
+   Carl D. Worth
    Copyright (C) 2001 University of Southern California
 
    This program is free software; you can redistribute it and/or
    General Public License for more details.
 */
 
-#include "includes.h"
+#include "config.h"
+
+#include <stdio.h>
 #include <sys/types.h>
 #include <sys/stat.h>
 #include <dirent.h>
+#include <unistd.h>
+#include <ctype.h>
 
 #include "sprintf_alloc.h"
 #include "file_util.h"
+#ifdef HAVE_MD5
 #include "md5.h"
+#endif
 #include "libbb/libbb.h"
-#undef strlen
 
 #if defined HAVE_SHA256
 #include "sha256.h"
 #endif
 
-int
-file_exists(const char *file_name)
+int file_exists(const char *file_name)
 {
        struct stat st;
 
@@ -41,8 +46,7 @@ file_exists(const char *file_name)
        return 1;
 }
 
-int
-file_is_dir(const char *file_name)
+int file_is_dir(const char *file_name)
 {
        struct stat st;
 
@@ -59,8 +63,7 @@ file_is_dir(const char *file_name)
 
    Return value is NULL if the file is at EOF when called.
 */
-char *
-file_read_line_alloc(FILE *fp)
+char *file_read_line_alloc(FILE * fp)
 {
        char buf[BUFSIZ];
        unsigned int buf_len;
@@ -79,7 +82,7 @@ file_read_line_alloc(FILE *fp)
                }
                if (line) {
                        line_size += buf_len;
-                       line = xrealloc(line, line_size+1);
+                       line = xrealloc(line, line_size + 1);
                        strncat(line, buf, line_size);
                } else {
                        line_size = buf_len + 1;
@@ -92,8 +95,7 @@ file_read_line_alloc(FILE *fp)
        return line;
 }
 
-int
-file_move(const char *src, const char *dest)
+int file_move(const char *src, const char *dest)
 {
        int err;
 
@@ -106,135 +108,196 @@ file_move(const char *src, const char *dest)
                                unlink(src);
                } else {
                        opkg_perror(ERROR, "Failed to rename %s to %s",
-                               src, dest);
+                                   src, dest);
                }
        }
 
        return err;
 }
 
-int
-file_copy(const char *src, const char *dest)
+int file_copy(const char *src, const char *dest)
 {
        int err;
 
        err = copy_file(src, dest, FILEUTILS_FORCE | FILEUTILS_PRESERVE_STATUS);
        if (err)
-               opkg_msg(ERROR, "Failed to copy file %s to %s.\n",
-                               src, dest);
+               opkg_msg(ERROR, "Failed to copy file %s to %s.\n", src, dest);
 
        return err;
 }
 
-int
-file_mkdir_hier(const char *path, long mode)
+int file_mkdir_hier(const char *path, long mode)
 {
        return make_directory(path, mode, FILEUTILS_RECUR);
 }
 
+#ifdef HAVE_MD5
 char *file_md5sum_alloc(const char *file_name)
 {
-    static const int md5sum_bin_len = 16;
-    static const int md5sum_hex_len = 32;
-
-    static const unsigned char bin2hex[16] = {
-       '0', '1', '2', '3',
-       '4', '5', '6', '7',
-       '8', '9', 'a', 'b',
-       'c', 'd', 'e', 'f'
-    };
-
-    int i, err;
-    FILE *file;
-    char *md5sum_hex;
-    unsigned char md5sum_bin[md5sum_bin_len];
-
-    md5sum_hex = xcalloc(1, md5sum_hex_len + 1);
-
-    file = fopen(file_name, "r");
-    if (file == NULL) {
-       opkg_perror(ERROR, "Failed to open file %s", file_name);
-       free(md5sum_hex);
-       return NULL;
-    }
-
-    err = md5_stream(file, md5sum_bin);
-    if (err) {
-       opkg_msg(ERROR, "Could't compute md5sum for %s.\n", file_name);
+       static const int md5sum_bin_len = 16;
+       static const int md5sum_hex_len = 32;
+
+       static const unsigned char bin2hex[16] = {
+               '0', '1', '2', '3',
+               '4', '5', '6', '7',
+               '8', '9', 'a', 'b',
+               'c', 'd', 'e', 'f'
+       };
+
+       int i, err;
+       FILE *file;
+       char *md5sum_hex;
+       unsigned char md5sum_bin[md5sum_bin_len];
+
+       md5sum_hex = xcalloc(1, md5sum_hex_len + 1);
+
+       file = fopen(file_name, "r");
+       if (file == NULL) {
+               opkg_perror(ERROR, "Failed to open file %s", file_name);
+               free(md5sum_hex);
+               return NULL;
+       }
+
+       err = md5_stream(file, md5sum_bin);
+       if (err) {
+               opkg_msg(ERROR, "Could't compute md5sum for %s.\n", file_name);
+               fclose(file);
+               free(md5sum_hex);
+               return NULL;
+       }
+
        fclose(file);
-       free(md5sum_hex);
-       return NULL;
-    }
-
-    fclose(file);
-
-    for (i=0; i < md5sum_bin_len; i++) {
-       md5sum_hex[i*2] = bin2hex[md5sum_bin[i] >> 4];
-       md5sum_hex[i*2+1] = bin2hex[md5sum_bin[i] & 0xf];
-    }
-    
-    md5sum_hex[md5sum_hex_len] = '\0';
-    
-    return md5sum_hex;
+
+       for (i = 0; i < md5sum_bin_len; i++) {
+               md5sum_hex[i * 2] = bin2hex[md5sum_bin[i] >> 4];
+               md5sum_hex[i * 2 + 1] = bin2hex[md5sum_bin[i] & 0xf];
+       }
+
+       md5sum_hex[md5sum_hex_len] = '\0';
+
+       return md5sum_hex;
 }
+#endif
 
 #ifdef HAVE_SHA256
 char *file_sha256sum_alloc(const char *file_name)
 {
-    static const int sha256sum_bin_len = 32;
-    static const int sha256sum_hex_len = 64;
-
-    static const unsigned char bin2hex[16] = {
-       '0', '1', '2', '3',
-       '4', '5', '6', '7',
-       '8', '9', 'a', 'b',
-       'c', 'd', 'e', 'f'
-    };
-
-    int i, err;
-    FILE *file;
-    char *sha256sum_hex;
-    unsigned char sha256sum_bin[sha256sum_bin_len];
-
-    sha256sum_hex = xcalloc(1, sha256sum_hex_len + 1);
-
-    file = fopen(file_name, "r");
-    if (file == NULL) {
-       opkg_perror(ERROR, "Failed to open file %s", file_name);
-       free(sha256sum_hex);
-       return NULL;
-    }
-
-    err = sha256_stream(file, sha256sum_bin);
-    if (err) {
-       opkg_msg(ERROR, "Could't compute sha256sum for %s.\n", file_name);
+       static const int sha256sum_bin_len = 32;
+       static const int sha256sum_hex_len = 64;
+
+       static const unsigned char bin2hex[16] = {
+               '0', '1', '2', '3',
+               '4', '5', '6', '7',
+               '8', '9', 'a', 'b',
+               'c', 'd', 'e', 'f'
+       };
+
+       int i, err;
+       FILE *file;
+       char *sha256sum_hex;
+       unsigned char sha256sum_bin[sha256sum_bin_len];
+
+       sha256sum_hex = xcalloc(1, sha256sum_hex_len + 1);
+
+       file = fopen(file_name, "r");
+       if (file == NULL) {
+               opkg_perror(ERROR, "Failed to open file %s", file_name);
+               free(sha256sum_hex);
+               return NULL;
+       }
+
+       err = sha256_stream(file, sha256sum_bin);
+       if (err) {
+               opkg_msg(ERROR, "Could't compute sha256sum for %s.\n",
+                        file_name);
+               fclose(file);
+               free(sha256sum_hex);
+               return NULL;
+       }
+
        fclose(file);
-       free(sha256sum_hex);
-       return NULL;
-    }
-
-    fclose(file);
-
-    for (i=0; i < sha256sum_bin_len; i++) {
-       sha256sum_hex[i*2] = bin2hex[sha256sum_bin[i] >> 4];
-       sha256sum_hex[i*2+1] = bin2hex[sha256sum_bin[i] & 0xf];
-    }
-    
-    sha256sum_hex[sha256sum_hex_len] = '\0';
-    
-    return sha256sum_hex;
+
+       for (i = 0; i < sha256sum_bin_len; i++) {
+               sha256sum_hex[i * 2] = bin2hex[sha256sum_bin[i] >> 4];
+               sha256sum_hex[i * 2 + 1] = bin2hex[sha256sum_bin[i] & 0xf];
+       }
+
+       sha256sum_hex[sha256sum_hex_len] = '\0';
+
+       return sha256sum_hex;
 }
 
 #endif
 
+char *checksum_bin2hex(const char *src, size_t len)
+{
+       unsigned char *p;
+       static unsigned char buf[65];
+       const unsigned char *s = (unsigned char *)src;
+       static const unsigned char bin2hex[16] = {
+               '0', '1', '2', '3',
+               '4', '5', '6', '7',
+               '8', '9', 'a', 'b',
+               'c', 'd', 'e', 'f'
+       };
+
+       if (!s || len > 32)
+               return NULL;
+
+       for (p = buf; len > 0; s++, len--) {
+               *p++ = bin2hex[*s / 16];
+               *p++ = bin2hex[*s % 16];
+       }
+
+       *p = 0;
+
+       return (char *)buf;
+}
+
+char *checksum_hex2bin(const char *src, size_t *len)
+{
+       size_t slen;
+       unsigned char *p;
+       const unsigned char *s = (unsigned char *)src;
+       static unsigned char buf[32];
+
+       if (!src) {
+               *len = 0;
+               return NULL;
+       }
+
+       while (isspace(*src))
+               src++;
 
-int
-rm_r(const char *path)
+       slen = strlen(src);
+
+       if (slen > 64) {
+               *len = 0;
+               return NULL;
+       }
+
+#define hex(c) \
+       (c >= 'a' ? (c - 'a') : (c >= 'A' ? (c - 'A') : (c - '0')))
+
+       for (p = buf, *len = 0;
+            slen > 0 && isxdigit(s[0]) && isxdigit(s[1]);
+            slen--, s += 2, (*len)++)
+               *p++ = hex(s[0]) * 16 + hex(s[1]);
+
+       return (char *)buf;
+}
+
+int rm_r(const char *path)
 {
        int ret = 0;
        DIR *dir;
        struct dirent *dent;
 
+       if (path == NULL) {
+               opkg_perror(ERROR, "Missing directory parameter");
+               return -1;
+       }
+
        dir = opendir(path);
        if (dir == NULL) {
                opkg_perror(ERROR, "Failed to open dir %s", path);
@@ -252,7 +315,7 @@ rm_r(const char *path)
                if ((dent = readdir(dir)) == NULL) {
                        if (errno) {
                                opkg_perror(ERROR, "Failed to read dir %s",
-                                               path);
+                                           path);
                                ret = -1;
                        }
                        break;
@@ -272,7 +335,7 @@ rm_r(const char *path)
                        struct stat st;
                        if ((ret = lstat(dent->d_name, &st)) == -1) {
                                opkg_perror(ERROR, "Failed to lstat %s",
-                                               dent->d_name);
+                                           dent->d_name);
                                break;
                        }
                        if (S_ISDIR(st.st_mode)) {
@@ -293,7 +356,7 @@ rm_r(const char *path)
                opkg_perror(ERROR, "Failed to change to dir %s/..", path);
        }
 
-       if (rmdir(path) == -1 ) {
+       if (rmdir(path) == -1) {
                ret = -1;
                opkg_perror(ERROR, "Failed to remove dir %s", path);
        }