/*
 * Copyright (C) 2024 The Android Open Source Project
 *
 * 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.android.test;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.junit.Assume.assumeTrue;
import org.junit.Before;
import org.junit.After;

import android.platform.test.annotations.AppModeFull;

import com.android.tradefed.testtype.DeviceJUnit4ClassRunner;
import com.android.tradefed.testtype.junit4.BaseHostJUnit4Test;
import com.android.tradefed.testtype.junit4.DeviceTestRunOptions;
import com.android.tradefed.util.CommandResult;
import com.android.tradefed.util.RunUtil;

import org.junit.Test;
import org.junit.runner.RunWith;

import java.io.BufferedReader;
import java.io.StringReader;

@RunWith(DeviceJUnit4ClassRunner.class)
public class CopyEfsTest extends BaseHostJUnit4Test {

    @Before
    public void setUp() throws Exception {
        getDevice().enableAdbRoot();

        getDevice().executeShellCommand("rm -rf /data/local/tmp/efs_test");
        getDevice().executeShellCommand("mkdir -p /data/local/tmp/efs_test/mnt");
        getDevice().executeShellCommand("mkdir -p /data/local/tmp/efs_test/dump");
    }

    @Test
    @AppModeFull
    public void copyEfsTest() throws Exception {
        assumeTrue(getDevice().executeShellCommand("getconf PAGESIZE").trim().equals("4096"));

        testDumpF2FS("efs");
        testDumpF2FS("efs_backup");
        testDumpF2FS("modem_userdata");
    }

    private void testDumpF2FS(String name) throws Exception {
        getDevice().executeShellCommand(String.format("cp /dev/block/by-name/%s /data/local/tmp/efs_test/%s.img", name, name));

        // The device was mounted r/w. To get a clean image, we run fsck, and then mount to allow mount time fixes to happen.
        // We can then dump and mount read only to ensure the contents should be the same.
        getDevice().executeShellCommand(String.format("fsck.f2fs -f /data/local/tmp/efs_test/%s.img", name, name));
        CommandResult r = getDevice().executeShellV2Command(String.format("mount /data/local/tmp/efs_test/%s.img /data/local/tmp/efs_test/mnt", name));
        assertEquals(r.getExitCode().intValue(), 0);
        r = getDevice().executeShellV2Command("umount /data/local/tmp/efs_test/mnt");
        assertEquals(r.getExitCode().intValue(), 0);

        r = getDevice().executeShellV2Command(String.format("dump.f2fs -rfPo /data/local/tmp/efs_test/dump /data/local/tmp/efs_test/%s.img", name));
        assertEquals(r.getExitCode().intValue(), 0);
        r = getDevice().executeShellV2Command(String.format("mount -r /data/local/tmp/efs_test/%s.img /data/local/tmp/efs_test/mnt", name));
        assertEquals(r.getExitCode().intValue(), 0);

        assertEquals("", getDevice().executeShellCommand("diff -rq /data/local/tmp/efs_test/mnt /data/local/tmp/efs_test/dump"));
        // Remove timestamps at positions 6 and 7, because ls on device does not support --time-style
        // Remove totals because on disk block usage may change depending on filesystem
        String ls_cmd = "cd /data/local/tmp/efs_test/%s;ls -AlLnR . | awk {'$6=\"\";$7=\"\";if ($1 != \"total\"){print $0}'}";
        String mnt_ls = getDevice().executeShellCommand(String.format(ls_cmd, "mnt"));
        assertEquals(getDevice().executeShellCommand("echo $?"), "0\n");
        String dump_ls = getDevice().executeShellCommand(String.format(ls_cmd, "dump"));
        assertEquals(getDevice().executeShellCommand("echo $?"), "0\n");
        assertEquals(mnt_ls, dump_ls);

        getDevice().executeShellCommand("umount /data/local/tmp/efs_test/mnt");
        getDevice().executeShellCommand("rm -rf /data/local/tmp/efs_test/dump/*");
        getDevice().executeShellCommand("rm /data/local/tmp/efs_test/" + name + ".img");
    }

    @After
    public void tearDown() throws Exception {
        getDevice().executeShellCommand("umount /data/local/tmp/efs_test/mnt");
        getDevice().executeShellCommand("rm -rf /data/local/tmp/efs_test");
    }
}