1 /* $NetBSD: file_test.c,v 1.3 2025/01/26 16:25:49 christos Exp $ */ 2 3 /* 4 * Copyright (C) Internet Systems Consortium, Inc. ("ISC") 5 * 6 * SPDX-License-Identifier: MPL-2.0 7 * 8 * This Source Code Form is subject to the terms of the Mozilla Public 9 * License, v. 2.0. If a copy of the MPL was not distributed with this 10 * file, you can obtain one at https://mozilla.org/MPL/2.0/. 11 * 12 * See the COPYRIGHT file distributed with this work for additional 13 * information regarding copyright ownership. 14 */ 15 16 #include <fcntl.h> 17 #include <inttypes.h> 18 #include <sched.h> /* IWYU pragma: keep */ 19 #include <setjmp.h> 20 #include <stdarg.h> 21 #include <stddef.h> 22 #include <stdlib.h> 23 #include <string.h> 24 #include <unistd.h> 25 26 #define UNIT_TESTING 27 #include <cmocka.h> 28 29 #include <isc/file.h> 30 #include <isc/result.h> 31 #include <isc/util.h> 32 33 #include <tests/isc.h> 34 35 #define NAME "internal" 36 #define SHA "3bed2cb3a3acf7b6a8ef408420cc682d5520e26976d354254f528c965612054f" 37 #define TRUNC_SHA "3bed2cb3a3acf7b6" 38 39 #define BAD1 "in/internal" 40 #define BADHASH1 "8bbb97a888791399" 41 42 #define BAD2 "Internal" 43 #define BADHASH2 "2ea1842b445b0c81" 44 45 #define F(x) "testdata/file/" x ".test" 46 47 static void 48 touch(const char *filename) { 49 int fd; 50 51 unlink(filename); 52 fd = creat(filename, 0644); 53 if (fd != -1) { 54 close(fd); 55 } 56 } 57 58 /* test sanitized filenames */ 59 ISC_RUN_TEST_IMPL(isc_file_sanitize) { 60 isc_result_t result; 61 char buf[1024]; 62 63 UNUSED(state); 64 65 assert_return_code(chdir(TESTS_DIR), 0); 66 67 result = isc_file_sanitize("testdata/file", NAME, "test", buf, 1024); 68 assert_int_equal(result, ISC_R_SUCCESS); 69 assert_int_equal(strcmp(buf, F(NAME)), 0); 70 71 touch(F(TRUNC_SHA)); 72 result = isc_file_sanitize("testdata/file", NAME, "test", buf, 1024); 73 assert_int_equal(result, ISC_R_SUCCESS); 74 assert_int_equal(strcmp(buf, F(TRUNC_SHA)), 0); 75 76 touch(F(SHA)); 77 result = isc_file_sanitize("testdata/file", NAME, "test", buf, 1024); 78 assert_int_equal(result, ISC_R_SUCCESS); 79 assert_int_equal(strcmp(buf, F(SHA)), 0); 80 81 result = isc_file_sanitize("testdata/file", BAD1, "test", buf, 1024); 82 assert_int_equal(result, ISC_R_SUCCESS); 83 assert_int_equal(strcmp(buf, F(BADHASH1)), 0); 84 85 result = isc_file_sanitize("testdata/file", BAD2, "test", buf, 1024); 86 assert_int_equal(result, ISC_R_SUCCESS); 87 assert_int_equal(strcmp(buf, F(BADHASH2)), 0); 88 89 unlink(F(TRUNC_SHA)); 90 unlink(F(SHA)); 91 } 92 93 /* test filename templates */ 94 ISC_RUN_TEST_IMPL(isc_file_template) { 95 isc_result_t result; 96 char buf[1024]; 97 98 UNUSED(state); 99 100 assert_return_code(chdir(TESTS_DIR), 0); 101 102 result = isc_file_template("/absolute/path", "file-XXXXXXXX", buf, 103 sizeof(buf)); 104 assert_int_equal(result, ISC_R_SUCCESS); 105 assert_string_equal(buf, "/absolute/file-XXXXXXXX"); 106 107 result = isc_file_template("relative/path", "file-XXXXXXXX", buf, 108 sizeof(buf)); 109 assert_int_equal(result, ISC_R_SUCCESS); 110 assert_string_equal(buf, "relative/file-XXXXXXXX"); 111 112 result = isc_file_template("/trailing/slash/", "file-XXXXXXXX", buf, 113 sizeof(buf)); 114 assert_int_equal(result, ISC_R_SUCCESS); 115 assert_string_equal(buf, "/trailing/slash/file-XXXXXXXX"); 116 117 result = isc_file_template("relative/trailing/slash/", "file-XXXXXXXX", 118 buf, sizeof(buf)); 119 assert_int_equal(result, ISC_R_SUCCESS); 120 assert_string_equal(buf, "relative/trailing/slash/file-XXXXXXXX"); 121 122 result = isc_file_template("/", "file-XXXXXXXX", buf, sizeof(buf)); 123 assert_int_equal(result, ISC_R_SUCCESS); 124 assert_string_equal(buf, "/file-XXXXXXXX"); 125 126 result = isc_file_template("noslash", "file-XXXXXXXX", buf, 127 sizeof(buf)); 128 assert_int_equal(result, ISC_R_SUCCESS); 129 assert_string_equal(buf, "file-XXXXXXXX"); 130 131 result = isc_file_template(NULL, "file-XXXXXXXX", buf, sizeof(buf)); 132 assert_int_equal(result, ISC_R_SUCCESS); 133 assert_string_equal(buf, "file-XXXXXXXX"); 134 } 135 136 ISC_TEST_LIST_START 137 138 ISC_TEST_ENTRY(isc_file_sanitize) 139 ISC_TEST_ENTRY(isc_file_template) 140 141 ISC_TEST_LIST_END 142 143 ISC_TEST_MAIN 144