1 /*
2  * Copyright (c) 2017 Intel Corporation
3  *
4  * SPDX-License-Identifier: Apache-2.0
5  */
6 #include <string.h>
7 #include <zephyr/types.h>
8 #include <stdbool.h>
9 #include <zephyr/ztest.h>
10 
11 /* Don't confuse doxygen with missing files */
12 /**
13  * @cond INTERNAL_HIDDEN
14  */
15 
16 /* The file.bin.inc contains characters from 0 to 255 */
17 static const unsigned char inc_file[] = {
18 #include <file.bin.inc>
19 };
20 
21 /* The file.bin.partial.inc contains characters from 100 to 141 */
22 static const unsigned char partial_inc_file[] = {
23 #include <file.bin.partial.inc>
24 };
25 
26 static const unsigned char no_mtime_gz_inc_file[] = {
27 #include <file.bin.gz.inc>
28 };
29 
30 static const unsigned char mtime_gz_inc_file[] = {
31 #include <file.bin.mtime.gz.inc>
32 };
33 
34 static const unsigned char partial_gz_inc_file[] = {
35 #include <file.bin.partial.gz.inc>
36 };
37 
38 /**
39  * @endcond
40  */
41 
42 static const unsigned char mtime_zero[4] = { 0, 0, 0, 0 };
43 
44 /* Must match the --gzip-mtime=test_val passed to
45  * generate_inc_file_for_target() */
46 static const unsigned char mtime_test_val[4] = { 42, 0, 0, 0 };
47 
48 static const unsigned char compressed_inc_file[] = {
49 	0x1f, 0x8b, 0x08, 0x00, 0xDD, 0xDD, 0xDD, 0xDD,
50 	0x02, 0xff, 0x01, 0x00, 0x01, 0xff, 0xfe, 0x00,
51 	0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
52 	0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f, 0x10,
53 	0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18,
54 	0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f, 0x20,
55 	0x21, 0x22, 0x23, 0x24, 0x25, 0x26, 0x27, 0x28,
56 	0x29, 0x2a, 0x2b, 0x2c, 0x2d, 0x2e, 0x2f, 0x30,
57 	0x31, 0x32, 0x33, 0x34, 0x35, 0x36, 0x37, 0x38,
58 	0x39, 0x3a, 0x3b, 0x3c, 0x3d, 0x3e, 0x3f, 0x40,
59 	0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47, 0x48,
60 	0x49, 0x4a, 0x4b, 0x4c, 0x4d, 0x4e, 0x4f, 0x50,
61 	0x51, 0x52, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58,
62 	0x59, 0x5a, 0x5b, 0x5c, 0x5d, 0x5e, 0x5f, 0x60,
63 	0x61, 0x62, 0x63, 0x64, 0x65, 0x66, 0x67, 0x68,
64 	0x69, 0x6a, 0x6b, 0x6c, 0x6d, 0x6e, 0x6f, 0x70,
65 	0x71, 0x72, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78,
66 	0x79, 0x7a, 0x7b, 0x7c, 0x7d, 0x7e, 0x7f, 0x80,
67 	0x81, 0x82, 0x83, 0x84, 0x85, 0x86, 0x87, 0x88,
68 	0x89, 0x8a, 0x8b, 0x8c, 0x8d, 0x8e, 0x8f, 0x90,
69 	0x91, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98,
70 	0x99, 0x9a, 0x9b, 0x9c, 0x9d, 0x9e, 0x9f, 0xa0,
71 	0xa1, 0xa2, 0xa3, 0xa4, 0xa5, 0xa6, 0xa7, 0xa8,
72 	0xa9, 0xaa, 0xab, 0xac, 0xad, 0xae, 0xaf, 0xb0,
73 	0xb1, 0xb2, 0xb3, 0xb4, 0xb5, 0xb6, 0xb7, 0xb8,
74 	0xb9, 0xba, 0xbb, 0xbc, 0xbd, 0xbe, 0xbf, 0xc0,
75 	0xc1, 0xc2, 0xc3, 0xc4, 0xc5, 0xc6, 0xc7, 0xc8,
76 	0xc9, 0xca, 0xcb, 0xcc, 0xcd, 0xce, 0xcf, 0xd0,
77 	0xd1, 0xd2, 0xd3, 0xd4, 0xd5, 0xd6, 0xd7, 0xd8,
78 	0xd9, 0xda, 0xdb, 0xdc, 0xdd, 0xde, 0xdf, 0xe0,
79 	0xe1, 0xe2, 0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8,
80 	0xe9, 0xea, 0xeb, 0xec, 0xed, 0xee, 0xef, 0xf0,
81 	0xf1, 0xf2, 0xf3, 0xf4, 0xf5, 0xf6, 0xf7, 0xf8,
82 	0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff, 0x73,
83 	0x8c, 0x05, 0x29, 0x00, 0x01, 0x00, 0x00,
84 };
85 
86 static const unsigned char compressed_partial_inc_file[] = {
87 	0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00,
88 	0x02, 0xff, 0x4b, 0x49, 0x4d, 0x4b, 0xcf, 0xc8,
89 	0xcc, 0xca, 0xce, 0xc9, 0xcd, 0xcb, 0x2f, 0x28,
90 	0x2c, 0x2a, 0x2e, 0x29, 0x2d, 0x2b, 0xaf, 0xa8,
91 	0xac, 0xaa, 0xae, 0xa9, 0xad, 0xab, 0x6f, 0x68,
92 	0x6c, 0x6a, 0x6e, 0x69, 0x6d, 0x6b, 0xef, 0xe8,
93 	0xec, 0xea, 0xee, 0xe9, 0x05, 0x00, 0xe7, 0xa1,
94 	0xc1, 0x22, 0x2a, 0x00, 0x00, 0x00,
95 };
96 
ZTEST(gen_inc_file,test_gen_inc_file)97 ZTEST(gen_inc_file, test_gen_inc_file)
98 {
99 	int i;
100 
101 	zassert_equal(sizeof(inc_file), 256, "Invalid size file");
102 
103 	for (i = 0; i < sizeof(inc_file); i++) {
104 		zassert_equal(inc_file[i], i, "Invalid value in inc file");
105 	}
106 }
107 
ZTEST(gen_inc_file,test_gen_partial_inc_file)108 ZTEST(gen_inc_file, test_gen_partial_inc_file)
109 {
110 	int i;
111 
112 	zassert_equal(sizeof(partial_inc_file), 42, "Invalid partial size file");
113 
114 	for (i = 0; i < sizeof(partial_inc_file); i++) {
115 		zassert_equal(partial_inc_file[i], i + 100, "Invalid value in partial inc file");
116 	}
117 }
118 
do_test_gen_gz_inc_file(const unsigned char gz_inc_file[],const unsigned char ref_file[],int ref_size,const unsigned char mtime[4])119 static void do_test_gen_gz_inc_file(const unsigned char gz_inc_file[],
120 				    const unsigned char ref_file[], int ref_size,
121 				    const unsigned char mtime[4])
122 {
123 	int i;
124 
125 	for (i = 0; i < ref_size; i++) {
126 		if (4 <= i && i < 8) {
127 			/* Modification time field (4 bytes) in
128 			 * the gzip header.
129 			 */
130 			if (mtime != NULL) { /* NULL arg = random "now" */
131 				zassert_equal(gz_inc_file[i], mtime[i-4],
132 					      "Invalid mtime in inc file");
133 			}
134 
135 			continue;
136 		}
137 
138 		zassert_equal(gz_inc_file[i], ref_file[i],
139 			      "Invalid value in inc file");
140 	}
141 }
142 
ZTEST(gen_inc_file,test_gen_gz_inc_file_no_mtime)143 ZTEST(gen_inc_file, test_gen_gz_inc_file_no_mtime)
144 {
145 	zassert_equal(sizeof(no_mtime_gz_inc_file), sizeof(compressed_inc_file),
146 		      "Invalid compressed file size");
147 
148 	do_test_gen_gz_inc_file(no_mtime_gz_inc_file, compressed_inc_file,
149 				sizeof(compressed_inc_file), mtime_zero);
150 }
151 
ZTEST(gen_inc_file,test_gen_gz_inc_file_mtime_arg)152 ZTEST(gen_inc_file, test_gen_gz_inc_file_mtime_arg)
153 {
154 	zassert_equal(sizeof(mtime_gz_inc_file), sizeof(compressed_inc_file),
155 		      "Invalid compressed file size");
156 
157 	do_test_gen_gz_inc_file(mtime_gz_inc_file, compressed_inc_file,
158 				sizeof(compressed_inc_file), mtime_test_val);
159 }
160 
ZTEST(gen_inc_file,test_gen_gz_inc_partial_file)161 ZTEST(gen_inc_file, test_gen_gz_inc_partial_file)
162 {
163 	zassert_equal(sizeof(partial_gz_inc_file), sizeof(compressed_partial_inc_file),
164 		      "Invalid partial compressed file size");
165 
166 	do_test_gen_gz_inc_file(partial_gz_inc_file, compressed_partial_inc_file,
167 				sizeof(compressed_partial_inc_file), NULL);
168 }
169 
170 
171 ZTEST_SUITE(gen_inc_file, NULL, NULL, NULL, NULL, NULL);
172