From 22dca23d52c836e40c79cb4042b867fdc06f6ca3 Mon Sep 17 00:00:00 2001 From: Denis Vlasenko Date: Sun, 3 Sep 2006 14:23:29 +0000 Subject: archival: added O_TRUNC so that when we overwrite files on unpack, we truncate them. Also spotted & fixed hard to trigger bug with extension handling. --- archival/unlzma.c | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'archival/unlzma.c') diff --git a/archival/unlzma.c b/archival/unlzma.c index fbd207ca1..00acea6e0 100644 --- a/archival/unlzma.c +++ b/archival/unlzma.c @@ -37,14 +37,15 @@ int unlzma_main(int argc, char **argv) if (filename) { struct stat stat_buf; - char *extension = filename + strlen(filename) - 5; - - if (strcmp(extension, ".lzma") != 0) { + /* bug: char *extension = filename + strlen(filename) - 5; */ + char *extension = strrchr(filename, '.'); + if (!extension || strcmp(extension, ".lzma") != 0) { bb_error_msg_and_die("Invalid extension"); } xstat(filename, &stat_buf); - *extension = 0; - dst_fd = xopen3(filename, O_WRONLY | O_CREAT, stat_buf.st_mode); + *extension = '\0'; + dst_fd = xopen3(filename, O_WRONLY | O_CREAT | O_TRUNC, + stat_buf.st_mode); } else dst_fd = STDOUT_FILENO; status = unlzma(src_fd, dst_fd); -- cgit v1.2.3