// 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 filesystem

import (
 "fmt"
 "path/filepath"
 "strings"

 "android/soong/android"

 "github.com/google/blueprint"
 "github.com/google/blueprint/proptools"
)

func init() {
 pctx.HostBinToolVariable("fsverity_metadata_generator", "fsverity_metadata_generator")
 pctx.HostBinToolVariable("fsverity_manifest_generator", "fsverity_manifest_generator")
 pctx.HostBinToolVariable("fsverity", "fsverity")
}

var (
 buildFsverityMeta = pctx.AndroidStaticRule("build_fsverity_meta", blueprint.RuleParams{
  Command:         `$fsverity_metadata_generator --fsverity-path $fsverity --signature none --hash-alg sha256 --output $out $in`,
  CommandDeps:     []string{"$fsverity_metadata_generator", "$fsverity"},
  SandboxDisabled: true,
 })
 buildFsverityManifest = pctx.AndroidStaticRule("build_fsverity_manifest", blueprint.RuleParams{
  Command:         `$fsverity_manifest_generator --fsverity-path $fsverity --output $out @$in`,
  CommandDeps:     []string{"$fsverity_manifest_generator", "$fsverity"},
  SandboxDisabled: true,
 })
)

type fsverityProperties struct {
 // Patterns of files for fsverity metadata generation.  For each matched file, a .fsv_meta file
 // will be generated and included to the filesystem image.
 // etc/security/fsverity/BuildManifest.apk will also be generated which contains information
 // about generated .fsv_meta files.
 Inputs proptools.Configurable[[]string]

 // APK libraries to link against, for etc/security/fsverity/BuildManifest.apk
 Libs proptools.Configurable[[]string] `android:"path"`
}

// Mapping of a given fsverity file, which may be a real file or a symlink, and the on-device
// path it should have relative to the filesystem root.
type fsveritySrcDest struct {
 src  android.Path
 dest string
}

func (f *filesystem) writeManifestGeneratorListFile(
 ctx android.ModuleContext,
 outputPath android.WritablePath,
 matchedFiles []fsveritySrcDest,
 rootDir android.OutputPath,
 rebasedDir android.OutputPath,
) []android.Path {
 prefix, err := filepath.Rel(rootDir.String(), rebasedDir.String())
 if err != nil {
  panic("rebasedDir should be relative to rootDir")
 }
 if prefix == "." {
  prefix = ""
 }
 if f.PartitionType() == "system_ext" {
  // Use the equivalent of $PRODUCT_OUT as the base dir.
  // This ensures that the paths in build_manifest.pb contain on-device paths
  // e.g. system_ext/framework/javalib.jar
  // and not framework/javalib.jar.
  //
  // Although base-dir is outside the rootdir provided for packaging, this action
  // is hermetic since it uses `manifestGeneratorListPath` to filter the files to be written to build_manifest.pb
  prefix = "system_ext"
 }

 var deps []android.Path
 var buf strings.Builder
 for _, spec := range matchedFiles {
  src := spec.src.String()
  dst := filepath.Join(prefix, spec.dest)
  if strings.Contains(src, ",") {
   ctx.ModuleErrorf("Path cannot contain a comma: %s", src)
  }
  if strings.Contains(dst, ",") {
   ctx.ModuleErrorf("Path cannot contain a comma: %s", dst)
  }
  buf.WriteString(src)
  buf.WriteString(",")
  buf.WriteString(dst)
  buf.WriteString("\n")
  deps = append(deps, spec.src)
 }
 android.WriteFileRuleVerbatim(ctx, outputPath, buf.String())
 return deps
}

func (f *filesystem) buildFsverityMetadataFiles(
 ctx android.ModuleContext,
 builder *android.RuleBuilder,
 specs map[string]android.PackagingSpec,
 rootDir android.OutputPath,
 rebasedDir android.OutputPath,
 fullInstallPaths *[]FullInstallPathInfo,
 platformGeneratedFiles *[]string,
) {
 match := func(path string) bool {
  for _, pattern := range f.properties.Fsverity.Inputs.GetOrDefault(ctx, nil) {
   if matched, err := filepath.Match(pattern, path); matched {
    return true
   } else if err != nil {
    ctx.PropertyErrorf("fsverity.inputs", "bad pattern %q", pattern)
    return false
   }
  }
  return false
 }

 var matchedFiles []android.PackagingSpec
 var matchedSymlinks []android.PackagingSpec
 for _, relPath := range android.SortedKeys(specs) {
  if match(relPath) {
   spec := specs[relPath]
   if spec.SrcPath() != nil {
    matchedFiles = append(matchedFiles, spec)
   } else if spec.SymlinkTarget() != "" {
    matchedSymlinks = append(matchedSymlinks, spec)
   } else {
    ctx.ModuleErrorf("Expected a file or symlink for fsverity packaging spec")
   }
  }
 }

 if len(matchedFiles) == 0 && len(matchedSymlinks) == 0 {
  return
 }

 // STEP 1: generate .fsv_meta
 var fsverityFileSpecs []fsveritySrcDest
 for _, spec := range matchedFiles {
  rel := spec.RelPathInPackage() + ".fsv_meta"
  outPath := android.PathForModuleOut(ctx, "fsverity/meta_files", rel)
  destPath := rebasedDir.Join(ctx, rel)
  // srcPath is copied by CopySpecsToDir()
  ctx.Build(pctx, android.BuildParams{
   Rule:   buildFsverityMeta,
   Input:  spec.SrcPath(),
   Output: outPath,
  })
  builder.Command().Textf("cp").Input(outPath).Output(destPath)
  f.appendToEntry(ctx, destPath)
  *fullInstallPaths = append(*fullInstallPaths, FullInstallPathInfo{
   SourcePath:      destPath,
   FullInstallPath: android.PathForModuleInPartitionInstall(ctx, f.PartitionType(), rel),
  })
  fsverityFileSpecs = append(fsverityFileSpecs, fsveritySrcDest{
   src:  spec.SrcPath(),
   dest: spec.RelPathInPackage(),
  })
 }
 for _, spec := range matchedSymlinks {
  rel := spec.RelPathInPackage() + ".fsv_meta"
  outPath := android.PathForModuleOut(ctx, "fsverity/meta_files", rel)
  destPath := rebasedDir.Join(ctx, rel)
  target := spec.SymlinkTarget() + ".fsv_meta"
  ctx.Build(pctx, android.BuildParams{
   Rule:   android.Symlink,
   Output: outPath,
   Args: map[string]string{
    "fromPath": target,
   },
  })
  builder.Command().
   Textf("cp").
   Flag(ctx.Config().CpPreserveSymlinksFlags()).
   Input(outPath).
   Output(destPath)
  f.appendToEntry(ctx, destPath)
  *fullInstallPaths = append(*fullInstallPaths, FullInstallPathInfo{
   SymlinkTarget:   target,
   FullInstallPath: android.PathForModuleInPartitionInstall(ctx, f.PartitionType(), rel),
  })
  // The fsverity manifest tool needs to actually look at the symlink. But symlink
  // packagingSpecs are not actually created on disk, at least until the staging dir is
  // built for the partition. Create a fake one now so the tool can see it.
  realizedSymlink := android.PathForModuleOut(ctx, "fsverity/realized_symlinks", spec.RelPathInPackage())
  ctx.Build(pctx, android.BuildParams{
   Rule:   android.Symlink,
   Output: realizedSymlink,
   Args: map[string]string{
    "fromPath": spec.SymlinkTarget(),
   },
  })
  fsverityFileSpecs = append(fsverityFileSpecs, fsveritySrcDest{
   src:  realizedSymlink,
   dest: spec.RelPathInPackage(),
  })
 }

 // STEP 2: generate signed BuildManifest.apk
 // STEP 2-1: generate build_manifest.pb
 manifestGeneratorListPath := android.PathForModuleOut(ctx, "fsverity/fsverity_manifest.list")
 manifestDeps := f.writeManifestGeneratorListFile(ctx, manifestGeneratorListPath, fsverityFileSpecs, rootDir, rebasedDir)
 manifestPbPath := android.PathForModuleOut(ctx, "fsverity/build_manifest.pb")
 ctx.Build(pctx, android.BuildParams{
  Rule:      buildFsverityManifest,
  Input:     manifestGeneratorListPath,
  Implicits: manifestDeps,
  Output:    manifestPbPath,
 })

 // STEP 2-2: generate BuildManifest.apk (unsigned)
 apkNameSuffix := ""
 if f.PartitionType() == "system_ext" {
  //https://source.corp.google.com/h/googleplex-android/platform/build/+/e392d2b486c2d4187b20a72b1c67cc737ecbcca5:core/Makefile;l=3410;drc=ea8f34bc1d6e63656b4ec32f2391e9d54b3ebb6b;bpv=1;bpt=0
  apkNameSuffix = "SystemExt"
 }
 apkPath := android.PathForModuleOut(ctx, "fsverity", fmt.Sprintf("BuildManifest%s.apk", apkNameSuffix))
 idsigPath := android.PathForModuleOut(ctx, "fsverity", fmt.Sprintf("BuildManifest%s.apk.idsig", apkNameSuffix))
 manifestTemplatePath := android.PathForSource(ctx, "system/security/fsverity/AndroidManifest.xml")
 libs := android.PathsForModuleSrc(ctx, f.properties.Fsverity.Libs.GetOrDefault(ctx, nil))

 minSdkVersion := ctx.Config().PlatformSdkCodename()
 if minSdkVersion == "REL" {
  minSdkVersion = ctx.Config().PlatformSdkVersion().String()
 }

 apkBuilder := android.NewRuleBuilder(pctx, ctx).SandboxDisabled()

 // aapt2 doesn't support adding individual asset files. Create a temp directory to hold asset
 // files and pass it to aapt2.
 tmpAssetDir := android.PathForModuleOut(ctx, "fsverity/tmp_asset_dir")
 stagedManifestPbPath := tmpAssetDir.Join(ctx, "build_manifest.pb")
 apkBuilder.Command().
  Text("rm -rf").Text(tmpAssetDir.String()).
  Text("&&").
  Text("mkdir -p").Text(tmpAssetDir.String())
 apkBuilder.Command().Text("cp").Input(manifestPbPath).Output(stagedManifestPbPath)

 unsignedApkCommand := apkBuilder.Command().
  BuiltTool("aapt2").
  Text("link").
  FlagWithOutput("-o ", apkPath).
  FlagWithArg("-A ", tmpAssetDir.String()).Implicit(stagedManifestPbPath)
 for _, lib := range libs {
  unsignedApkCommand.FlagWithInput("-I ", lib)
 }
 unsignedApkCommand.
  FlagWithArg("--min-sdk-version ", minSdkVersion).
  FlagWithArg("--version-code ", ctx.Config().PlatformSdkVersion().String()).
  FlagWithArg("--version-name ", ctx.Config().AppsDefaultVersionName()).
  FlagWithInput("--manifest ", manifestTemplatePath).
  Text(" --rename-manifest-package com.android.security.fsverity_metadata." + f.partitionName())

 // STEP 2-3: sign BuildManifest.apk
 pemPath, keyPath := ctx.Config().DefaultAppCertificate(ctx)
 apkBuilder.Command().
  BuiltTool("apksigner").
  Text("sign").
  FlagWithArg("--in ", apkPath.String()).
  FlagWithInput("--cert ", pemPath).
  FlagWithInput("--key ", keyPath).
  ImplicitOutput(idsigPath)
 apkBuilder.Build(fmt.Sprintf("%s_fsverity_apk", ctx.ModuleName()), "build fsverity apk")

 // STEP 2-4: Install the apk into the staging directory
 installedApkPath := rebasedDir.Join(ctx, "etc", "security", "fsverity", fmt.Sprintf("BuildManifest%s.apk", apkNameSuffix))
 installedIdsigPath := rebasedDir.Join(ctx, "etc", "security", "fsverity", fmt.Sprintf("BuildManifest%s.apk.idsig", apkNameSuffix))
 builder.Command().Text("mkdir -p").Text(filepath.Dir(installedApkPath.String()))
 builder.Command().Text("cp").Input(apkPath).Text(installedApkPath.String())
 builder.Command().Text("cp").Input(idsigPath).Text(installedIdsigPath.String())

 apkFullInstallPath := android.PathForModuleInPartitionInstall(ctx, f.PartitionType(), fmt.Sprintf("etc/security/fsverity/BuildManifest%s.apk", apkNameSuffix))
 *fullInstallPaths = append(*fullInstallPaths, FullInstallPathInfo{
  SourcePath:      apkPath,
  FullInstallPath: apkFullInstallPath,
 })

 f.appendToEntry(ctx, installedApkPath)

 idsigFullInstallPath := android.PathForModuleInPartitionInstall(ctx, f.PartitionType(), fmt.Sprintf("etc/security/fsverity/BuildManifest%s.apk.idsig", apkNameSuffix))
 *fullInstallPaths = append(*fullInstallPaths, FullInstallPathInfo{
  SourcePath:      idsigPath,
  FullInstallPath: idsigFullInstallPath,
 })

 f.appendToEntry(ctx, installedIdsigPath)
 *platformGeneratedFiles = append(*platformGeneratedFiles, apkFullInstallPath.String(), idsigFullInstallPath.String())
}