// Copyright (C) 2018 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 apex

import (
 "fmt"

 "android/soong/android"

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

//go:generate go run ../../blueprint/gobtools/codegen

var String = proptools.String

func init() {
 registerApexKeyBuildComponents(android.InitRegistrationContext)
}

func registerApexKeyBuildComponents(ctx android.RegistrationContext) {
 ctx.RegisterModuleType("apex_key", ApexKeyFactory)
 ctx.RegisterModuleType("all_apex_certs", allApexCertsFactory)
}

// @auto-generate: gob
type ApexKeyInfo struct {
 PublicKeyFile  android.Path
 PrivateKeyFile android.Path
}

var ApexKeyInfoProvider = blueprint.NewProvider[ApexKeyInfo]()

type apexKey struct {
 android.ModuleBase

 properties apexKeyProperties

 publicKeyFile  android.Path
 privateKeyFile android.Path
}

type apexKeyProperties struct {
 // Path or module to the public key file in avbpubkey format. Installed to the device.
 // Base name of the file is used as the ID for the key.
 Public_key *string `android:"path"`
 // Path or module to the private key file in pem format. Used to sign APEXs.
 Private_key *string `android:"path"`

 // Whether this key is installable to one of the partitions. Defualt: true.
 Installable *bool
}

func ApexKeyFactory() android.Module {
 module := &apexKey{}
 module.AddProperties(&module.properties)
 android.InitAndroidArchModule(module, android.DeviceSupported, android.MultilibCommon)
 return module
}

func (m *apexKey) installable() bool {
 return false
}

func (m *apexKey) GenerateAndroidBuildActions(ctx android.ModuleContext) {
 // If the keys are from other modules (i.e. :module syntax) respect it.
 // Otherwise, try to locate the key files in the default cert dir or
 // in the local module dir
 if android.SrcIsModule(String(m.properties.Public_key)) != "" {
  m.publicKeyFile = android.PathForModuleSrc(ctx, String(m.properties.Public_key))
 } else {
  m.publicKeyFile = ctx.Config().ApexKeyDir(ctx).Join(ctx, String(m.properties.Public_key))
  // If not found, fall back to the local key pairs
  if !android.ExistentPathForSource(ctx, m.publicKeyFile.String()).Valid() {
   m.publicKeyFile = android.PathForModuleSrc(ctx, String(m.properties.Public_key))
  }
 }

 if android.SrcIsModule(String(m.properties.Private_key)) != "" {
  m.privateKeyFile = android.PathForModuleSrc(ctx, String(m.properties.Private_key))
 } else {
  m.privateKeyFile = ctx.Config().ApexKeyDir(ctx).Join(ctx, String(m.properties.Private_key))
  if !android.ExistentPathForSource(ctx, m.privateKeyFile.String()).Valid() {
   m.privateKeyFile = android.PathForModuleSrc(ctx, String(m.properties.Private_key))
  }
 }

 pubKeyName := m.publicKeyFile.Base()[0 : len(m.publicKeyFile.Base())-len(m.publicKeyFile.Ext())]
 privKeyName := m.privateKeyFile.Base()[0 : len(m.privateKeyFile.Base())-len(m.privateKeyFile.Ext())]

 if m.properties.Public_key != nil && m.properties.Private_key != nil && pubKeyName != privKeyName {
  ctx.ModuleErrorf("public_key %q (keyname:%q) and private_key %q (keyname:%q) do not have same keyname",
   m.publicKeyFile.String(), pubKeyName, m.privateKeyFile, privKeyName)
  return
 }

 android.SetProvider(ctx, ApexKeyInfoProvider, ApexKeyInfo{
  PublicKeyFile:  m.publicKeyFile,
  PrivateKeyFile: m.privateKeyFile,
 })
}

type apexKeyEntry struct {
 name                 string
 presigned            bool
 publicKey            string
 privateKey           string
 containerCertificate string
 containerPrivateKey  string
 partition            string
 signTool             string
}

func (e apexKeyEntry) String() string {
 signTool := ""
 if e.signTool != "" {
  signTool = fmt.Sprintf(" sign_tool=%q", e.signTool)
 }
 format := "name=%q public_key=%q private_key=%q container_certificate=%q container_private_key=%q partition=%q%s\n"
 if e.presigned {
  return fmt.Sprintf(format, e.name, "PRESIGNED", "PRESIGNED", "PRESIGNED", "PRESIGNED", e.partition, signTool)
 } else {
  return fmt.Sprintf(format, e.name, e.publicKey, e.privateKey, e.containerCertificate, e.containerPrivateKey, e.partition, signTool)
 }
}

func apexKeyEntryFor(ctx android.ModuleContext, module android.Module) apexKeyEntry {
 switch m := module.(type) {
 case *apexBundle:
  pem, key := m.getCertificateAndPrivateKey(ctx)
  return apexKeyEntry{
   name:                 m.Name() + ".apex",
   presigned:            false,
   publicKey:            m.publicKeyFile.String(),
   privateKey:           m.privateKeyFile.String(),
   containerCertificate: pem.String(),
   containerPrivateKey:  key.String(),
   partition:            m.PartitionTag(ctx.DeviceConfig()),
   signTool:             proptools.String(m.properties.Custom_sign_tool),
  }
 case *Prebuilt:
  return apexKeyEntry{
   name:      m.InstallFilename(),
   presigned: true,
   partition: m.PartitionTag(ctx.DeviceConfig()),
  }
 case *ApexSet:
  return apexKeyEntry{
   name:      m.InstallFilename(),
   presigned: true,
   partition: m.PartitionTag(ctx.DeviceConfig()),
  }
 }
 panic(fmt.Errorf("unknown type(%t) for apexKeyEntry", module))
}

func writeApexKeys(ctx android.ModuleContext, module android.Module) android.WritablePath {
 path := android.PathForModuleOut(ctx, "apexkeys.txt")
 entry := apexKeyEntryFor(ctx, module)
 android.WriteFileRuleVerbatim(ctx, path, entry.String())
 return path
}

var (
 pemToDer = pctx.AndroidStaticRule("pem_to_der",
  blueprint.RuleParams{
   Command2: blueprint.NewCommand(
    `$openssl x509 -inform PEM -outform DER -in $in -out $out`,
   ),
   CommandDeps: []string{"${openssl}"},
   Description: "Convert certificate from PEM to DER format",
  },
 )
)

// all_apex_certs is a singleton module that collects the certs of all apexes in the tree.
// It provides two types of output files
// 1. .pem: This is usually the checked-in x509 certificate in PEM format
// 2. .der: This is DER format of the certificate, and is generated from the PEM certificate using `openssl x509`
func allApexCertsFactory() android.Module {
 m := &allApexCerts{}
 android.InitAndroidArchModule(m, android.DeviceSupported, android.MultilibCommon)
 return m
}

type allApexCerts struct {
 android.ModuleBase
}

func (_ *allApexCerts) GenerateAndroidBuildActions(ctx android.ModuleContext) {
 if ctx.ModuleName() != "all_apex_certs" || ctx.Namespace().Path != "." {
  ctx.ModuleErrorf(`all_apex_certs can only be used by a single module named "all_apex_certs" in the root namespace.`)
 }

 var avbpubkeys android.Paths
 var certificatesPem android.Paths
 ctx.VisitDirectDepsProxy(func(m android.ModuleProxy) {
  if apex, ok := android.OtherModuleProvider(ctx, m, android.ApexBundleTypeInfoProvider); ok {
   pem := apex.Pem
   if !android.ExistentPathForSource(ctx, pem.String()).Valid() {
    if ctx.Config().AllowMissingDependencies() {
     return
    } else {
     ctx.ModuleErrorf("Path %s is not valid\n", pem.String())
    }
   }
   certificatesPem = append(certificatesPem, pem)
   // avbpubkey for signing the apex payload
   avbpubkeys = append(avbpubkeys, apex.Key)
  }
 })
 certificatesPem = android.SortedUniquePaths(certificatesPem) // For hermiticity
 avbpubkeys = android.SortedUniquePaths(avbpubkeys)           // For hermiticity
 var certificatesDer android.Paths
 for index, certificatePem := range certificatesPem {
  certificateDer := android.PathForModuleOut(ctx, fmt.Sprintf("x509.%v.der", index))
  ctx.Build(pctx, android.BuildParams{
   Rule:   pemToDer,
   Input:  certificatePem,
   Output: certificateDer,
  })
  certificatesDer = append(certificatesDer, certificateDer)
 }
 ctx.SetOutputFiles(certificatesPem, ".pem")
 ctx.SetOutputFiles(certificatesDer, ".der")
 ctx.SetOutputFiles(avbpubkeys, ".avbpubkey")
}