1 /*
2 * Copyright (C) 2016 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include "TestHelpers.h"
18
19 #include "ziparchive/zip_archive.h"
20
21 using ::testing::AssertionFailure;
22 using ::testing::AssertionResult;
23 using ::testing::AssertionSuccess;
24
25 namespace android {
26
ReadFileFromZipToString(const std::string & zip_path,const std::string & file,std::string * out_contents)27 AssertionResult ReadFileFromZipToString(const std::string& zip_path, const std::string& file,
28 std::string* out_contents) {
29 out_contents->clear();
30 ::ZipArchiveHandle handle;
31 int32_t result = OpenArchive(zip_path.c_str(), &handle);
32 if (result != 0) {
33 return AssertionFailure() << "Failed to open zip '" << zip_path
34 << "': " << ::ErrorCodeString(result);
35 }
36
37 ::ZipEntry entry;
38 result = ::FindEntry(handle, file.c_str(), &entry);
39 if (result != 0) {
40 ::CloseArchive(handle);
41 return AssertionFailure() << "Could not find file '" << file << "' in zip '" << zip_path
42 << "' : " << ::ErrorCodeString(result);
43 }
44
45 out_contents->resize(entry.uncompressed_length);
46 result = ::ExtractToMemory(
47 handle, &entry, const_cast<uint8_t*>(reinterpret_cast<const uint8_t*>(out_contents->data())),
48 out_contents->size());
49 if (result != 0) {
50 ::CloseArchive(handle);
51 return AssertionFailure() << "Failed to extract file '" << file << "' from zip '" << zip_path
52 << "': " << ::ErrorCodeString(result);
53 }
54
55 ::CloseArchive(handle);
56 return AssertionSuccess();
57 }
58
IsStringEqual(const ResTable & table,uint32_t resource_id,const char * expected_str)59 AssertionResult IsStringEqual(const ResTable& table, uint32_t resource_id,
60 const char* expected_str) {
61 Res_value val;
62 ssize_t block = table.getResource(resource_id, &val, MAY_NOT_BE_BAG);
63 if (block < 0) {
64 return AssertionFailure() << "could not find resource";
65 }
66
67 if (val.dataType != Res_value::TYPE_STRING) {
68 return AssertionFailure() << "resource is not a string";
69 }
70
71 const ResStringPool* pool = table.getTableStringBlock(block);
72 if (pool == NULL) {
73 return AssertionFailure() << "table has no string pool for block " << block;
74 }
75
76 const String8 actual_str = pool->string8ObjectAt(val.data);
77 if (String8(expected_str) != actual_str) {
78 return AssertionFailure() << actual_str.string();
79 }
80 return AssertionSuccess() << actual_str.string();
81 }
82
83 } // namespace android
84