com.reandroid.archive.model.LocalFileDirectory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ARSCLib Show documentation
Android binary resources read/write library
There is a newer version: 1.3.5
/*
 *  Copyright (C) 2022 github.com/REAndroid
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.reandroid.archive.model;

import com.reandroid.archive.ArchiveEntry;
import com.reandroid.archive.ArchiveException;
import com.reandroid.archive.block.*;
import com.reandroid.archive.block.ApkSignatureBlock;
import com.reandroid.archive.io.ZipInput;
import com.reandroid.arsc.io.BlockReader;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;

public class LocalFileDirectory {
    private final CentralFileDirectory centralFileDirectory;
    private final List headerList;
    private ApkSignatureBlock apkSignatureBlock;
    public LocalFileDirectory(CentralFileDirectory centralFileDirectory){
        this.centralFileDirectory = centralFileDirectory;
        this.headerList = new ArrayList<>(centralFileDirectory.count() + 2);
    }
    public void visit(ZipInput zipInput) throws IOException {
        visitLocalFile(zipInput);
        visitApkSigBlock(zipInput);
    }
    private void visitLocalFile(ZipInput zipInput) throws IOException {
        List headerList = this.getHeaderList();
        long offset;
        int index = 0;
        CentralFileDirectory centralFileDirectory = getCentralFileDirectory();
        long length = zipInput.getLength();
        InputStream inputStream = zipInput.getInputStream(0, length);
        for(CentralEntryHeader ceh : centralFileDirectory.getHeaderList()){
            offset = ceh.getLocalRelativeOffset();
            inputStream.reset();
            offset = inputStream.skip(offset);
            LocalFileHeader lfh = LocalFileHeader.read(inputStream);
            if(lfh == null){
                throw new ArchiveException("Error reading LFH at "
                        + offset + ", for CEH = " + ceh.getFileName());
            }
            offset = offset + lfh.countBytes();
            ceh.setFileOffset(offset);

            lfh.setCentralEntryHeader(ceh);

            inputStream.skip(lfh.getDataSize());

            DataDescriptor dataDescriptor = null;
            if(lfh.hasDataDescriptor()){
                dataDescriptor = new DataDescriptor();
                int read = dataDescriptor.readBytes(inputStream);
                if(read != dataDescriptor.countBytes()){
                    dataDescriptor = null;
                }
            }
            lfh.setDataDescriptor(dataDescriptor);
            lfh.setIndex(index);

            headerList.add(lfh);

            index++;
        }
    }
    private void visitApkSigBlock(ZipInput zipInput) throws IOException{
        CentralFileDirectory cfd = getCentralFileDirectory();
        SignatureFooter footer = cfd.getSignatureFooter();
        if(footer == null || !footer.isValid()){
            return;
        }
        EndRecord endRecord = cfd.getEndRecord();
        long length = footer.getSignatureSize() + 8;
        long offset = endRecord.getOffsetOfCentralDirectory() - length;
        ApkSignatureBlock apkSignatureBlock = new ApkSignatureBlock(footer);
        apkSignatureBlock.readBytes(new BlockReader(zipInput.getInputStream(offset, length)));
        this.apkSignatureBlock = apkSignatureBlock;
    }
    public ApkSignatureBlock getApkSigBlock() {
        return apkSignatureBlock;
    }
    public CentralFileDirectory getCentralFileDirectory() {
        return centralFileDirectory;
    }
    public List getHeaderList() {
        return headerList;
    }
    public ArchiveEntry[] buildArchiveEntryList(){
        List headerList = getHeaderList();
        int size = headerList.size();
        ArchiveEntry[] entryList = new ArchiveEntry[size];
        for(int i = 0; i < size; i++){
            LocalFileHeader lfh = headerList.get(i);
            CentralEntryHeader ceh = lfh.getCentralEntryHeader();
            if(ceh == null){
                continue;
            }
            entryList[i] = new ArchiveEntry(lfh);
        }
        return entryList;
    }
}