blob: 7409ec47007bbcc694f82bb3679c3bed76d34b0f [file] [log] [blame]
Elliott Hughes2faa5f12012-01-30 14:42:07 -08001/*
2 * Copyright (C) 2011 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 */
Brian Carlstromb0460ea2011-07-29 10:08:05 -070016
Elliott Hughes90a33692011-08-30 13:27:07 -070017#include "zip_archive.h"
18
Brian Carlstromb0460ea2011-07-29 10:08:05 -070019#include <fcntl.h>
20#include <sys/stat.h>
21#include <sys/types.h>
22
Elliott Hughes90a33692011-08-30 13:27:07 -070023#include "UniquePtr.h"
Brian Carlstromb0460ea2011-07-29 10:08:05 -070024#include "common_test.h"
Elliott Hughes90a33692011-08-30 13:27:07 -070025#include "os.h"
Brian Carlstromb0460ea2011-07-29 10:08:05 -070026
27namespace art {
28
Brian Carlstromf734cf52011-08-17 16:28:14 -070029class ZipArchiveTest : public CommonTest {};
Brian Carlstromb0460ea2011-07-29 10:08:05 -070030
Brian Carlstromb0460ea2011-07-29 10:08:05 -070031TEST_F(ZipArchiveTest, FindAndExtract) {
Elliott Hughes90a33692011-08-30 13:27:07 -070032 UniquePtr<ZipArchive> zip_archive(ZipArchive::Open(GetLibCoreDexFileName()));
33 ASSERT_TRUE(zip_archive.get() != false);
34 UniquePtr<ZipEntry> zip_entry(zip_archive->Find("classes.dex"));
35 ASSERT_TRUE(zip_entry.get() != false);
Brian Carlstromb0460ea2011-07-29 10:08:05 -070036
Brian Carlstromdb4d5402011-08-09 12:18:28 -070037 ScratchFile tmp;
Brian Carlstromb0460ea2011-07-29 10:08:05 -070038 ASSERT_NE(-1, tmp.GetFd());
Brian Carlstroma004aa92012-02-08 18:05:09 -080039 UniquePtr<File> file(OS::FileFromFd(tmp.GetFilename().c_str(), tmp.GetFd()));
Elliott Hughes90a33692011-08-30 13:27:07 -070040 ASSERT_TRUE(file.get() != NULL);
Brian Carlstrom89521892011-12-07 22:05:07 -080041 bool success = zip_entry->ExtractToFile(*file);
Brian Carlstromb0460ea2011-07-29 10:08:05 -070042 ASSERT_TRUE(success);
43 close(tmp.GetFd());
44
45 uint32_t computed_crc = crc32(0L, Z_NULL, 0);
Brian Carlstroma004aa92012-02-08 18:05:09 -080046 int fd = open(tmp.GetFilename().c_str(), O_RDONLY);
Brian Carlstromb0460ea2011-07-29 10:08:05 -070047 ASSERT_NE(-1, fd);
48 const size_t kBufSize = 32768;
49 uint8_t buf[kBufSize];
50 while (true) {
51 ssize_t bytes_read = TEMP_FAILURE_RETRY(read(fd, buf, kBufSize));
52 if (bytes_read == 0) {
53 break;
54 }
55 computed_crc = crc32(computed_crc, buf, bytes_read);
56 }
57 EXPECT_EQ(zip_entry->GetCrc32(), computed_crc);
58}
59
60} // namespace art