X-Git-Url: https://wimlib.net/git/?p=wimlib;a=blobdiff_plain;f=src%2Futil.c;h=8f4e0c0be53756db28ef2974d43de4b1f08b8b3e;hp=d091c3e155fba957bd57d38fd72b1131cba7100d;hb=1dde5fb0d809f2f5e032e4d5241d1cb15ff3eb65;hpb=885632f08c75c1d7bb5d25436231c78f6ad7e0c0 diff --git a/src/util.c b/src/util.c index d091c3e1..8f4e0c0b 100644 --- a/src/util.c +++ b/src/util.c @@ -1,28 +1,32 @@ /* * util.c - * + */ + +/* * Copyright (C) 2010 Carl Thijssen * Copyright (C) 2012 Eric Biggers * - * wimlib - Library for working with WIM files + * This file is part of wimlib, a library for working with WIM files. * - * This library is free software; you can redistribute it and/or modify it under - * the terms of the GNU Lesser General Public License as published by the Free - * Software Foundation; either version 2.1 of the License, or (at your option) any - * later version. + * wimlib is free software; you can redistribute it and/or modify it under the + * terms of the GNU Lesser General Public License as published by the Free + * Software Foundation; either version 2.1 of the License, or (at your option) + * any later version. * - * This library is distributed in the hope that it will be useful, but WITHOUT ANY - * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A - * PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. + * wimlib is distributed in the hope that it will be useful, but WITHOUT ANY + * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR + * A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more + * details. * - * You should have received a copy of the GNU Lesser General Public License along - * with this library; if not, write to the Free Software Foundation, Inc., 59 - * Temple Place, Suite 330, Boston, MA 02111-1307 USA + * You should have received a copy of the GNU Lesser General Public License + * along with wimlib; if not, see http://www.gnu.org/licenses/. */ #include "wimlib_internal.h" #include "endianness.h" #include "sha1.h" +#include "timestamp.h" +#include #include @@ -33,8 +37,8 @@ #include #include -/* True if WIMLIB is to print an informational message when an error occurs. - * This can be turned off by calling wimlib_set_error_messages(false). */ +/* True if wimlib is to print an informational message when an error occurs. + * This can be turned off by calling wimlib_set_print_errors(false). */ #ifdef ENABLE_ERROR_MESSAGES #include bool __wimlib_print_errors = false; @@ -47,12 +51,46 @@ void wimlib_error(const char *format, ...) va_start(va, format); errno_save = errno; - fputs("ERROR: ", stderr); + fputs("[ERROR] ", stderr); + vfprintf(stderr, format, va); + putc('\n', stderr); + errno = errno_save; + va_end(va); + } +} + +void wimlib_error_with_errno(const char *format, ...) +{ + if (__wimlib_print_errors) { + va_list va; + int errno_save; + + va_start(va, format); + errno_save = errno; + fputs("[ERROR] ", stderr); + vfprintf(stderr, format, va); + fprintf(stderr, ": %s\n", strerror(errno_save)); + errno = errno_save; + va_end(va); + } +} + +void wimlib_warning(const char *format, ...) +{ + if (__wimlib_print_errors) { + va_list va; + int errno_save; + + va_start(va, format); + errno_save = errno; + fputs("[WARNING] ", stderr); vfprintf(stderr, format, va); + putc('\n', stderr); errno = errno_save; va_end(va); } } + #endif WIMLIBAPI int wimlib_set_print_errors(bool show_error_messages) @@ -89,6 +127,8 @@ static const char *error_strings[] = { = "Tried to add an image with a name that is already in use", [WIMLIB_ERR_INTEGRITY] = "The WIM failed an integrity check", + [WIMLIB_ERR_INVALID_CAPTURE_CONFIG] + = "The capture configuration string was invalid", [WIMLIB_ERR_INVALID_CHUNK_SIZE] = "The WIM is compressed but does not have a chunk " "size of 32768", @@ -103,8 +143,12 @@ static const char *error_strings[] = { = "Tried to select an image that does not exist in the WIM", [WIMLIB_ERR_INVALID_INTEGRITY_TABLE] = "The WIM's integrity table is invalid", + [WIMLIB_ERR_INVALID_LOOKUP_TABLE_ENTRY] + = "An entry in the WIM's lookup table is invalid", [WIMLIB_ERR_INVALID_PARAM] = "An invalid parameter was given", + [WIMLIB_ERR_INVALID_RESOURCE_HASH] + = "The SHA1 message digest of a WIM resource did not match the expected value", [WIMLIB_ERR_INVALID_RESOURCE_SIZE] = "A resource entry in the WIM is invalid", [WIMLIB_ERR_LINK] @@ -123,16 +167,24 @@ static const char *error_strings[] = { "identify a WIM file", [WIMLIB_ERR_NO_FILENAME] = "The WIM is not identified with a filename", + [WIMLIB_ERR_NTFS_3G] + = "NTFS-3g encountered an error (check errno)", [WIMLIB_ERR_OPEN] = "Failed to open a file", [WIMLIB_ERR_OPENDIR] = "Failed to open a directory", [WIMLIB_ERR_READ] = "Could not read data from a file", + [WIMLIB_ERR_READLINK] + = "Could not read the target of a symbolic link", [WIMLIB_ERR_RENAME] = "Could not rename a file", - [WIMLIB_ERR_SPLIT] - = "The WIM is part of a split WIM, which Wimlib does not support", + [WIMLIB_ERR_SPECIAL_FILE] + = "Encountered a special file that cannot be archived", + [WIMLIB_ERR_SPLIT_INVALID] + = "The WIM is part of an invalid split WIM", + [WIMLIB_ERR_SPLIT_UNSUPPORTED] + = "The WIM is part of a split WIM, which is not supported for this operation", [WIMLIB_ERR_STAT] = "Could not read the metadata for a file or directory", [WIMLIB_ERR_TIMEOUT] @@ -158,8 +210,8 @@ WIMLIBAPI const char *wimlib_get_error_string(enum wimlib_error_code code) #ifdef ENABLE_CUSTOM_MEMORY_ALLOCATOR -void *(*wimlib_malloc_func)(size_t) = malloc; -void (*wimlib_free_func)(void *) = free; +void *(*wimlib_malloc_func) (size_t) = malloc; +void (*wimlib_free_func) (void *) = free; void *(*wimlib_realloc_func)(void *, size_t) = realloc; void *wimlib_calloc(size_t nmemb, size_t size) @@ -199,10 +251,11 @@ WIMLIBAPI int wimlib_set_memory_allocator(void *(*malloc_func)(size_t), xml_set_memory_allocator(wimlib_malloc_func, wimlib_free_func, wimlib_realloc_func); + return 0; #else - ERROR("Cannot set custom memory allocator functions:\n"); - ERROR("wimlib was compiled with the " - "--without-custom-memory-allocator flag\n"); + ERROR("Cannot set custom memory allocator functions:"); + ERROR("wimlib was compiled with the --without-custom-memory-allocator " + "flag"); return WIMLIB_ERR_UNSUPPORTED; #endif } @@ -214,13 +267,13 @@ static iconv_t cd_utf16_to_utf8 = (iconv_t)(-1); /* Converts a string in the UTF-16 encoding to a newly allocated string in the * UTF-8 encoding. */ char *utf16_to_utf8(const char *utf16_str, size_t utf16_len, - size_t *utf8_len_ret) + size_t *utf8_len_ret) { if (cd_utf16_to_utf8 == (iconv_t)(-1)) { cd_utf16_to_utf8 = iconv_open("UTF-8", "UTF-16LE"); if (cd_utf16_to_utf8 == (iconv_t)-1) { - ERROR("Failed to get conversion descriptor for " - "converting UTF-16LE to UTF-8: %m\n"); + ERROR_WITH_ERRNO("Failed to get conversion descriptor " + "for converting UTF-16LE to UTF-8"); return NULL; } } @@ -237,8 +290,8 @@ char *utf16_to_utf8(const char *utf16_str, size_t utf16_len, &utf16_bytes_left, &utf8_str, &utf8_bytes_left); if (num_chars_converted == (size_t)(-1)) { - ERROR("Failed to convert UTF-16LE string to UTF-8 string: " - "%m\n"); + ERROR_WITH_ERRNO("Failed to convert UTF-16LE string to UTF-8 " + "string"); FREE(orig_utf8_str); return NULL; } @@ -254,14 +307,14 @@ static iconv_t cd_utf8_to_utf16 = (iconv_t)(-1); /* Converts a string in the UTF-8 encoding to a newly allocated string in the * UTF-16 encoding. */ -char *utf8_to_utf16(const char *utf8_str, size_t utf8_len, - size_t *utf16_len_ret) +char *utf8_to_utf16(const char *utf8_str, size_t utf8_len, + size_t *utf16_len_ret) { if (cd_utf8_to_utf16 == (iconv_t)(-1)) { cd_utf8_to_utf16 = iconv_open("UTF-16LE", "UTF-8"); if (cd_utf8_to_utf16 == (iconv_t)-1) { - ERROR("Failed to get conversion descriptor for " - "converting UTF-8 to UTF-16LE: %m\n"); + ERROR_WITH_ERRNO("Failed to get conversion descriptor " + "for converting UTF-8 to UTF-16LE"); return NULL; } } @@ -280,11 +333,8 @@ char *utf8_to_utf16(const char *utf8_str, size_t utf8_len, &utf8_bytes_left, &utf16_str, &utf16_bytes_left); if (num_chars_converted == (size_t)(-1)) { - ERROR("Failed to convert UTF-8 string to UTF-16LE string: " - "%s\n", - (errno == E2BIG) ? - "Not enough room in output buffer" : - strerror(errno)); + ERROR_WITH_ERRNO("Failed to convert UTF-8 string to UTF-16LE " + "string"); FREE(orig_utf16_str); return NULL; } @@ -345,10 +395,8 @@ void randomize_char_array_with_alnum(char p[], size_t n) } /* Fills @n bytes pointer to by @p with random numbers. */ -void randomize_byte_array(void *__p, size_t n) +void randomize_byte_array(u8 *p, size_t n) { - u8 *p = __p; - if (!seeded) { srand(time(NULL)); seeded = true; @@ -381,8 +429,13 @@ const char *path_basename(const char *path) p--; /* Trailing slashes. */ - while ((p != path - 1) && *p == '/') + while (1) { + if (p == path - 1) + return ""; + if (*p != '/') + break; p--; + } while ((p != path - 1) && *p != '/') p--; @@ -390,6 +443,20 @@ const char *path_basename(const char *path) return p + 1; } +/* + * Returns a pointer to the part of @path following the first colon in the last + * path component, or NULL if the last path component does not contain a colon. + */ +const char *path_stream_name(const char *path) +{ + const char *base = path_basename(path); + const char *stream_name = strchr(base, ':'); + if (!stream_name) + return NULL; + else + return stream_name + 1; +} + /* * Splits a file path into the part before the first '/', or the entire name if * there is no '/', and the part after the first sequence of '/' characters. @@ -450,21 +517,11 @@ void print_string(const void *string, size_t len) } } -/* Calculates the SHA1 message digest given the name of a file. - * @buf must point to a buffer of length 20 bytes into which the message digest - * is written. - */ -int sha1sum(const char *filename, void *buf) +u64 get_wim_timestamp() { - FILE *fp; - int ret; - - fp = fopen(filename, "rb"); - if (!fp) { - ERROR("Cannot open the file `%s' for reading: %m\n", filename); - return WIMLIB_ERR_OPEN; - } - ret = sha1_stream(fp, buf); - fclose(fp); - return ret; + struct timeval tv; + gettimeofday(&tv, NULL); + return timeval_to_wim_timestamp(&tv); } + +