// SPDX-License-Identifier: GPL-2.0-only
/*
 * POWER Platform specific code for non-volatile SED key access
 * Copyright (C) 2022 IBM Corporation
 *
 * Define operations for SED Opal to read/write keys
 * from POWER LPAR Platform KeyStore(PLPKS).
 *
 * Self Encrypting Drives(SED) key storage using PLPKS
 */

#include <linux/kernel.h>
#include <linux/slab.h>
#include <linux/string.h>
#include <linux/ioctl.h>
#include <linux/sed-opal-key.h>
#include <asm/plpks.h>

static bool plpks_sed_initialized = false;
static bool plpks_sed_available = false;

/*
 * structure that contains all SED data
 */
struct plpks_sed_object_data {
 u_char version;
 u_char pad1[7];
 u_long authority;
 u_long range;
 u_int  key_len;
 u_char key[32];
};

#define PLPKS_SED_OBJECT_DATA_V0        0
#define PLPKS_SED_MANGLED_LABEL         "/default/pri"
#define PLPKS_SED_COMPONENT             "sed-opal"
#define PLPKS_SED_KEY                   "opal-boot-pin"

/*
 * authority is admin1 and range is global
 */
#define PLPKS_SED_AUTHORITY  0x0000000900010001
#define PLPKS_SED_RANGE      0x0000080200000001

static void plpks_init_var(struct plpks_var *var, char *keyname)
{
 if (!plpks_sed_initialized) {
  plpks_sed_initialized = true;
  plpks_sed_available = plpks_is_available();
  if (!plpks_sed_available)
   pr_err("SED: plpks not available\n");
 }

 var->name = keyname;
 var->namelen = strlen(keyname);
 if (strcmp(PLPKS_SED_KEY, keyname) == 0) {
  var->name = PLPKS_SED_MANGLED_LABEL;
  var->namelen = strlen(keyname);
 }
 var->policy = PLPKS_WORLDREADABLE;
 var->os = PLPKS_VAR_COMMON;
 var->data = NULL;
 var->datalen = 0;
 var->component = PLPKS_SED_COMPONENT;
}

/*
 * Read the SED Opal key from PLPKS given the label
 */
int sed_read_key(char *keyname, char *key, u_int *keylen)
{
 struct plpks_var var;
 struct plpks_sed_object_data data;
 int ret;
 u_int len;

 plpks_init_var(&var, keyname);

 if (!plpks_sed_available)
  return -EOPNOTSUPP;

 var.data = (u8 *)&data;
 var.datalen = sizeof(data);

 ret = plpks_read_os_var(&var);
 if (ret != 0)
  return ret;

 len = min_t(u16, be32_to_cpu(data.key_len), var.datalen);
 memcpy(key, data.key, len);
 key[len] = '\0';
 *keylen = len;

 return 0;
}

/*
 * Write the SED Opal key to PLPKS given the label
 */
int sed_write_key(char *keyname, char *key, u_int keylen)
{
 struct plpks_var var;
 struct plpks_sed_object_data data;
 struct plpks_var_name vname;

 plpks_init_var(&var, keyname);

 if (!plpks_sed_available)
  return -EOPNOTSUPP;

 var.datalen = sizeof(struct plpks_sed_object_data);
 var.data = (u8 *)&data;

 /* initialize SED object */
 data.version = PLPKS_SED_OBJECT_DATA_V0;
 data.authority = cpu_to_be64(PLPKS_SED_AUTHORITY);
 data.range = cpu_to_be64(PLPKS_SED_RANGE);
 memset(&data.pad1, '\0', sizeof(data.pad1));
 data.key_len = cpu_to_be32(keylen);
 memcpy(data.key, (char *)key, keylen);

 /*
  * Key update requires remove first. The return value
  * is ignored since it's okay if the key doesn't exist.
 */
 vname.namelen = var.namelen;
 vname.name = var.name;
 plpks_remove_var(var.component, var.os, vname);

 return plpks_write_var(var);
}