*/
 #define LENOVO_DEBUG_CMD_GUID "7FF47003-3B6C-4E5E-A227-E979824A85D1"
 
+/*
+ * Name:
+ *  Lenovo_OpcodeIF
+ * Description:
+ *  Opcode interface which provides the ability to set multiple
+ *  parameters and then trigger an action with a final command.
+ *  This is particularly useful for simplifying setting passwords.
+ *  With this support comes the ability to set System, HDD and NVMe
+ *  passwords.
+ *  This is currently available on ThinkCenter and ThinkStations platforms
+ */
+#define LENOVO_OPCODE_IF_GUID "DFDDEF2C-57D4-48ce-B196-0FB787D90836"
+
 #define TLMI_POP_PWD (1 << 0)
 #define TLMI_PAP_PWD (1 << 1)
+#define TLMI_HDD_PWD (1 << 2)
+#define TLMI_SYS_PWD (1 << 3)
 #define to_tlmi_pwd_setting(kobj)  container_of(kobj, struct tlmi_pwd_setting, kobj)
 #define to_tlmi_attr_setting(kobj)  container_of(kobj, struct tlmi_attr_setting, kobj)
 
        [TLMI_ENCODING_ASCII] = "ascii",
        [TLMI_ENCODING_SCANCODE] = "scancode",
 };
+static const char * const level_options[] = {
+       [TLMI_LEVEL_USER] = "user",
+       [TLMI_LEVEL_MASTER] = "master",
+};
 static struct think_lmi tlmi_priv;
 static struct class *fw_attr_class;
 
        struct acpi_buffer output = { ACPI_ALLOCATE_BUFFER, NULL };
        const union acpi_object *obj;
        acpi_status status;
+       int copy_size;
 
        if (!tlmi_priv.can_get_password_settings)
                return -EOPNOTSUPP;
         * The size of thinkpad_wmi_pcfg on ThinkStation is larger than ThinkPad.
         * To make the driver compatible on different brands, we permit it to get
         * the data in below case.
+        * Settings must have at minimum the core fields available
         */
-       if (obj->buffer.length < sizeof(struct tlmi_pwdcfg)) {
+       if (obj->buffer.length < sizeof(struct tlmi_pwdcfg_core)) {
                pr_warn("Unknown pwdcfg buffer length %d\n", obj->buffer.length);
                kfree(obj);
                return -EIO;
        }
-       memcpy(pwdcfg, obj->buffer.pointer, sizeof(struct tlmi_pwdcfg));
+
+       copy_size = obj->buffer.length < sizeof(struct tlmi_pwdcfg) ?
+               obj->buffer.length : sizeof(struct tlmi_pwdcfg);
+       memcpy(pwdcfg, obj->buffer.pointer, copy_size);
        kfree(obj);
+
+       if (WARN_ON(pwdcfg->core.max_length >= TLMI_PWD_BUFSIZE))
+               pwdcfg->core.max_length = TLMI_PWD_BUFSIZE - 1;
        return 0;
 }
 
                                password);
 }
 
+static int tlmi_opcode_setting(char *setting, const char *value)
+{
+       char *opcode_str;
+       int ret;
+
+       opcode_str = kasprintf(GFP_KERNEL, "%s:%s;", setting, value);
+       if (!opcode_str)
+               return -ENOMEM;
+
+       ret = tlmi_simple_call(LENOVO_OPCODE_IF_GUID, opcode_str);
+       kfree(opcode_str);
+       return ret;
+}
+
 static int tlmi_setting(int item, char **value, const char *guid_string)
 {
        struct acpi_buffer output = { ACPI_ALLOCATE_BUFFER, NULL };
                goto out;
        }
 
-       /* Format: 'PasswordType,CurrentPw,NewPw,Encoding,KbdLang;' */
-       auth_str = kasprintf(GFP_KERNEL, "%s,%s,%s,%s,%s;",
-                setting->pwd_type, setting->password, new_pwd,
-                encoding_options[setting->encoding], setting->kbdlang);
-       if (!auth_str) {
-               ret = -ENOMEM;
-               goto out;
+       /* If opcode support is present use that interface */
+       if (tlmi_priv.opcode_support) {
+               char pwd_type[8];
+
+               /* Special handling required for HDD and NVMe passwords */
+               if (setting == tlmi_priv.pwd_hdd) {
+                       if (setting->level == TLMI_LEVEL_USER)
+                               sprintf(pwd_type, "uhdp%d", setting->index);
+                       else
+                               sprintf(pwd_type, "mhdp%d", setting->index);
+               } else if (setting == tlmi_priv.pwd_nvme) {
+                       if (setting->level == TLMI_LEVEL_USER)
+                               sprintf(pwd_type, "unvp%d", setting->index);
+                       else
+                               sprintf(pwd_type, "mnvp%d", setting->index);
+               } else {
+                       sprintf(pwd_type, "%s", setting->pwd_type);
+               }
+
+               ret = tlmi_opcode_setting("WmiOpcodePasswordType", pwd_type);
+               if (ret)
+                       goto out;
+
+               if (tlmi_priv.pwd_admin->valid) {
+                       ret = tlmi_opcode_setting("WmiOpcodePasswordAdmin",
+                                       tlmi_priv.pwd_admin->password);
+                       if (ret)
+                               goto out;
+               }
+               ret = tlmi_opcode_setting("WmiOpcodePasswordCurrent01", setting->password);
+               if (ret)
+                       goto out;
+               ret = tlmi_opcode_setting("WmiOpcodePasswordNew01", new_pwd);
+               if (ret)
+                       goto out;
+               ret = tlmi_simple_call(LENOVO_OPCODE_IF_GUID, "WmiOpcodePasswordSetUpdate;");
+       } else {
+               /* Format: 'PasswordType,CurrentPw,NewPw,Encoding,KbdLang;' */
+               auth_str = kasprintf(GFP_KERNEL, "%s,%s,%s,%s,%s;",
+                               setting->pwd_type, setting->password, new_pwd,
+                               encoding_options[setting->encoding], setting->kbdlang);
+               if (!auth_str) {
+                       ret = -ENOMEM;
+                       goto out;
+               }
+               ret = tlmi_simple_call(LENOVO_SET_BIOS_PASSWORD_GUID, auth_str);
+               kfree(auth_str);
        }
-       ret = tlmi_simple_call(LENOVO_SET_BIOS_PASSWORD_GUID, auth_str);
-       kfree(auth_str);
 out:
        kfree(new_pwd);
        return ret ?: count;
 }
 static struct kobj_attribute auth_role = __ATTR_RO(role);
 
+static ssize_t index_show(struct kobject *kobj, struct kobj_attribute *attr,
+                        char *buf)
+{
+       struct tlmi_pwd_setting *setting = to_tlmi_pwd_setting(kobj);
+
+       return sysfs_emit(buf, "%d\n", setting->index);
+}
+
+static ssize_t index_store(struct kobject *kobj,
+                                 struct kobj_attribute *attr,
+                                 const char *buf, size_t count)
+{
+       struct tlmi_pwd_setting *setting = to_tlmi_pwd_setting(kobj);
+       int err, val;
+
+       err = kstrtoint(buf, 10, &val);
+       if (err < 0)
+               return err;
+
+       if (val > TLMI_INDEX_MAX)
+               return -EINVAL;
+
+       setting->index = val;
+       return count;
+}
+
+static struct kobj_attribute auth_index = __ATTR_RW(index);
+
+static ssize_t level_show(struct kobject *kobj, struct kobj_attribute *attr,
+                        char *buf)
+{
+       struct tlmi_pwd_setting *setting = to_tlmi_pwd_setting(kobj);
+
+       return sysfs_emit(buf, "%s\n", level_options[setting->level]);
+}
+
+static ssize_t level_store(struct kobject *kobj,
+                                 struct kobj_attribute *attr,
+                                 const char *buf, size_t count)
+{
+       struct tlmi_pwd_setting *setting = to_tlmi_pwd_setting(kobj);
+       int i;
+
+       /* Scan for a matching profile */
+       i = sysfs_match_string(level_options, buf);
+       if (i < 0)
+               return -EINVAL;
+
+       setting->level = i;
+       return count;
+}
+
+static struct kobj_attribute auth_level = __ATTR_RW(level);
+
+static umode_t auth_attr_is_visible(struct kobject *kobj,
+                                            struct attribute *attr, int n)
+{
+       struct tlmi_pwd_setting *setting = to_tlmi_pwd_setting(kobj);
+
+       /*We only want to display level and index settings on HDD/NVMe */
+       if ((attr == (struct attribute *)&auth_index) ||
+                       (attr == (struct attribute *)&auth_level)) {
+               if ((setting == tlmi_priv.pwd_hdd) || (setting == tlmi_priv.pwd_nvme))
+                       return attr->mode;
+               return 0;
+       }
+       return attr->mode;
+}
+
 static struct attribute *auth_attrs[] = {
        &auth_is_pass_set.attr,
        &auth_min_pass_length.attr,
        &auth_mechanism.attr,
        &auth_encoding.attr,
        &auth_kbdlang.attr,
+       &auth_index.attr,
+       &auth_level.attr,
        NULL
 };
 
 static const struct attribute_group auth_attr_group = {
+       .is_visible = auth_attr_is_visible,
        .attrs = auth_attrs,
 };
 
        kobject_put(&tlmi_priv.pwd_admin->kobj);
        sysfs_remove_group(&tlmi_priv.pwd_power->kobj, &auth_attr_group);
        kobject_put(&tlmi_priv.pwd_power->kobj);
+
+       if (tlmi_priv.opcode_support) {
+               sysfs_remove_group(&tlmi_priv.pwd_system->kobj, &auth_attr_group);
+               kobject_put(&tlmi_priv.pwd_system->kobj);
+               sysfs_remove_group(&tlmi_priv.pwd_hdd->kobj, &auth_attr_group);
+               kobject_put(&tlmi_priv.pwd_hdd->kobj);
+               sysfs_remove_group(&tlmi_priv.pwd_nvme->kobj, &auth_attr_group);
+               kobject_put(&tlmi_priv.pwd_nvme->kobj);
+       }
+
        kset_unregister(tlmi_priv.authentication_kset);
 }
 
                goto fail_create_attr;
 
        tlmi_priv.pwd_power->kobj.kset = tlmi_priv.authentication_kset;
-       ret = kobject_add(&tlmi_priv.pwd_power->kobj, NULL, "%s", "System");
+       ret = kobject_add(&tlmi_priv.pwd_power->kobj, NULL, "%s", "Power-on");
        if (ret)
                goto fail_create_attr;
 
        if (ret)
                goto fail_create_attr;
 
+       if (tlmi_priv.opcode_support) {
+               tlmi_priv.pwd_system->kobj.kset = tlmi_priv.authentication_kset;
+               ret = kobject_add(&tlmi_priv.pwd_system->kobj, NULL, "%s", "System");
+               if (ret)
+                       goto fail_create_attr;
+
+               ret = sysfs_create_group(&tlmi_priv.pwd_system->kobj, &auth_attr_group);
+               if (ret)
+                       goto fail_create_attr;
+
+               tlmi_priv.pwd_hdd->kobj.kset = tlmi_priv.authentication_kset;
+               ret = kobject_add(&tlmi_priv.pwd_hdd->kobj, NULL, "%s", "HDD");
+               if (ret)
+                       goto fail_create_attr;
+
+               ret = sysfs_create_group(&tlmi_priv.pwd_hdd->kobj, &auth_attr_group);
+               if (ret)
+                       goto fail_create_attr;
+
+               tlmi_priv.pwd_nvme->kobj.kset = tlmi_priv.authentication_kset;
+               ret = kobject_add(&tlmi_priv.pwd_nvme->kobj, NULL, "%s", "NVMe");
+               if (ret)
+                       goto fail_create_attr;
+
+               ret = sysfs_create_group(&tlmi_priv.pwd_nvme->kobj, &auth_attr_group);
+               if (ret)
+                       goto fail_create_attr;
+       }
+
        return ret;
 
 fail_create_attr:
 }
 
 /* ---- Base Driver -------------------------------------------------------- */
+static struct tlmi_pwd_setting *tlmi_create_auth(const char *pwd_type,
+                           const char *pwd_role)
+{
+       struct tlmi_pwd_setting *new_pwd;
+
+       new_pwd = kzalloc(sizeof(struct tlmi_pwd_setting), GFP_KERNEL);
+       if (!new_pwd)
+               return NULL;
+
+       strscpy(new_pwd->kbdlang, "us", TLMI_LANG_MAXLEN);
+       new_pwd->encoding = TLMI_ENCODING_ASCII;
+       new_pwd->pwd_type = pwd_type;
+       new_pwd->role = pwd_role;
+       new_pwd->minlen = tlmi_priv.pwdcfg.core.min_length;
+       new_pwd->maxlen = tlmi_priv.pwdcfg.core.max_length;
+       new_pwd->index = 0;
+       return new_pwd;
+}
+
 static int tlmi_analyze(void)
 {
-       struct tlmi_pwdcfg pwdcfg;
        acpi_status status;
        int i, ret;
 
        if (wmi_has_guid(LENOVO_DEBUG_CMD_GUID))
                tlmi_priv.can_debug_cmd = true;
 
+       if (wmi_has_guid(LENOVO_OPCODE_IF_GUID))
+               tlmi_priv.opcode_support = true;
+
        /*
         * Try to find the number of valid settings of this machine
         * and use it to create sysfs attributes.
        }
 
        /* Create password setting structure */
-       ret = tlmi_get_pwd_settings(&pwdcfg);
+       ret = tlmi_get_pwd_settings(&tlmi_priv.pwdcfg);
        if (ret)
                goto fail_clear_attr;
 
-       tlmi_priv.pwd_admin = kzalloc(sizeof(struct tlmi_pwd_setting), GFP_KERNEL);
+       tlmi_priv.pwd_admin = tlmi_create_auth("pap", "bios-admin");
        if (!tlmi_priv.pwd_admin) {
                ret = -ENOMEM;
                goto fail_clear_attr;
        }
-       strscpy(tlmi_priv.pwd_admin->kbdlang, "us", TLMI_LANG_MAXLEN);
-       tlmi_priv.pwd_admin->encoding = TLMI_ENCODING_ASCII;
-       tlmi_priv.pwd_admin->pwd_type = "pap";
-       tlmi_priv.pwd_admin->role = "bios-admin";
-       tlmi_priv.pwd_admin->minlen = pwdcfg.min_length;
-       if (WARN_ON(pwdcfg.max_length >= TLMI_PWD_BUFSIZE))
-               pwdcfg.max_length = TLMI_PWD_BUFSIZE - 1;
-       tlmi_priv.pwd_admin->maxlen = pwdcfg.max_length;
-       if (pwdcfg.password_state & TLMI_PAP_PWD)
+       if (tlmi_priv.pwdcfg.core.password_state & TLMI_PAP_PWD)
                tlmi_priv.pwd_admin->valid = true;
 
        kobject_init(&tlmi_priv.pwd_admin->kobj, &tlmi_pwd_setting_ktype);
 
-       tlmi_priv.pwd_power = kzalloc(sizeof(struct tlmi_pwd_setting), GFP_KERNEL);
+       tlmi_priv.pwd_power = tlmi_create_auth("pop", "power-on");
        if (!tlmi_priv.pwd_power) {
                ret = -ENOMEM;
-               goto fail_free_pwd_admin;
+               goto fail_clear_attr;
        }
-       strscpy(tlmi_priv.pwd_power->kbdlang, "us", TLMI_LANG_MAXLEN);
-       tlmi_priv.pwd_power->encoding = TLMI_ENCODING_ASCII;
-       tlmi_priv.pwd_power->pwd_type = "pop";
-       tlmi_priv.pwd_power->role = "power-on";
-       tlmi_priv.pwd_power->minlen = pwdcfg.min_length;
-       tlmi_priv.pwd_power->maxlen = pwdcfg.max_length;
-
-       if (pwdcfg.password_state & TLMI_POP_PWD)
+       if (tlmi_priv.pwdcfg.core.password_state & TLMI_POP_PWD)
                tlmi_priv.pwd_power->valid = true;
 
        kobject_init(&tlmi_priv.pwd_power->kobj, &tlmi_pwd_setting_ktype);
 
+       if (tlmi_priv.opcode_support) {
+               tlmi_priv.pwd_system = tlmi_create_auth("sys", "system");
+               if (!tlmi_priv.pwd_system) {
+                       ret = -ENOMEM;
+                       goto fail_clear_attr;
+               }
+               if (tlmi_priv.pwdcfg.core.password_state & TLMI_SYS_PWD)
+                       tlmi_priv.pwd_system->valid = true;
+
+               kobject_init(&tlmi_priv.pwd_system->kobj, &tlmi_pwd_setting_ktype);
+
+               tlmi_priv.pwd_hdd = tlmi_create_auth("hdd", "hdd");
+               if (!tlmi_priv.pwd_hdd) {
+                       ret = -ENOMEM;
+                       goto fail_clear_attr;
+               }
+               kobject_init(&tlmi_priv.pwd_hdd->kobj, &tlmi_pwd_setting_ktype);
+
+               tlmi_priv.pwd_nvme = tlmi_create_auth("nvm", "nvme");
+               if (!tlmi_priv.pwd_nvme) {
+                       ret = -ENOMEM;
+                       goto fail_clear_attr;
+               }
+               kobject_init(&tlmi_priv.pwd_nvme->kobj, &tlmi_pwd_setting_ktype);
+
+               if (tlmi_priv.pwdcfg.core.password_state & TLMI_HDD_PWD) {
+                       /* Check if PWD is configured and set index to first drive found */
+                       if (tlmi_priv.pwdcfg.ext.hdd_user_password ||
+                                       tlmi_priv.pwdcfg.ext.hdd_master_password) {
+                               tlmi_priv.pwd_hdd->valid = true;
+                               if (tlmi_priv.pwdcfg.ext.hdd_master_password)
+                                       tlmi_priv.pwd_hdd->index =
+                                               ffs(tlmi_priv.pwdcfg.ext.hdd_master_password) - 1;
+                               else
+                                       tlmi_priv.pwd_hdd->index =
+                                               ffs(tlmi_priv.pwdcfg.ext.hdd_user_password) - 1;
+                       }
+                       if (tlmi_priv.pwdcfg.ext.nvme_user_password ||
+                                       tlmi_priv.pwdcfg.ext.nvme_master_password) {
+                               tlmi_priv.pwd_nvme->valid = true;
+                               if (tlmi_priv.pwdcfg.ext.nvme_master_password)
+                                       tlmi_priv.pwd_nvme->index =
+                                               ffs(tlmi_priv.pwdcfg.ext.nvme_master_password) - 1;
+                               else
+                                       tlmi_priv.pwd_nvme->index =
+                                               ffs(tlmi_priv.pwdcfg.ext.nvme_user_password) - 1;
+                       }
+               }
+       }
        return 0;
 
-fail_free_pwd_admin:
-       kfree(tlmi_priv.pwd_admin);
 fail_clear_attr:
        for (i = 0; i < TLMI_SETTINGS_COUNT; ++i) {
                if (tlmi_priv.setting[i]) {
                        kfree(tlmi_priv.setting[i]);
                }
        }
+       kfree(tlmi_priv.pwd_admin);
+       kfree(tlmi_priv.pwd_power);
+       kfree(tlmi_priv.pwd_system);
+       kfree(tlmi_priv.pwd_hdd);
+       kfree(tlmi_priv.pwd_nvme);
        return ret;
 }