xref: /openbmc/linux/block/sed-opal.c (revision cbdf59ad)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright © 2016 Intel Corporation
4  *
5  * Authors:
6  *    Scott  Bauer      <scott.bauer@intel.com>
7  *    Rafael Antognolli <rafael.antognolli@intel.com>
8  */
9 
10 #define pr_fmt(fmt) KBUILD_MODNAME ":OPAL: " fmt
11 
12 #include <linux/delay.h>
13 #include <linux/device.h>
14 #include <linux/kernel.h>
15 #include <linux/list.h>
16 #include <linux/genhd.h>
17 #include <linux/slab.h>
18 #include <linux/uaccess.h>
19 #include <uapi/linux/sed-opal.h>
20 #include <linux/sed-opal.h>
21 #include <linux/string.h>
22 #include <linux/kdev_t.h>
23 
24 #include "opal_proto.h"
25 
26 #define IO_BUFFER_LENGTH 2048
27 #define MAX_TOKS 64
28 
29 /* Number of bytes needed by cmd_finalize. */
30 #define CMD_FINALIZE_BYTES_NEEDED 7
31 
32 struct opal_step {
33 	int (*fn)(struct opal_dev *dev, void *data);
34 	void *data;
35 };
36 typedef int (cont_fn)(struct opal_dev *dev);
37 
38 enum opal_atom_width {
39 	OPAL_WIDTH_TINY,
40 	OPAL_WIDTH_SHORT,
41 	OPAL_WIDTH_MEDIUM,
42 	OPAL_WIDTH_LONG,
43 	OPAL_WIDTH_TOKEN
44 };
45 
46 /*
47  * On the parsed response, we don't store again the toks that are already
48  * stored in the response buffer. Instead, for each token, we just store a
49  * pointer to the position in the buffer where the token starts, and the size
50  * of the token in bytes.
51  */
52 struct opal_resp_tok {
53 	const u8 *pos;
54 	size_t len;
55 	enum opal_response_token type;
56 	enum opal_atom_width width;
57 	union {
58 		u64 u;
59 		s64 s;
60 	} stored;
61 };
62 
63 /*
64  * From the response header it's not possible to know how many tokens there are
65  * on the payload. So we hardcode that the maximum will be MAX_TOKS, and later
66  * if we start dealing with messages that have more than that, we can increase
67  * this number. This is done to avoid having to make two passes through the
68  * response, the first one counting how many tokens we have and the second one
69  * actually storing the positions.
70  */
71 struct parsed_resp {
72 	int num;
73 	struct opal_resp_tok toks[MAX_TOKS];
74 };
75 
76 struct opal_dev {
77 	bool supported;
78 	bool mbr_enabled;
79 
80 	void *data;
81 	sec_send_recv *send_recv;
82 
83 	struct mutex dev_lock;
84 	u16 comid;
85 	u32 hsn;
86 	u32 tsn;
87 	u64 align;
88 	u64 lowest_lba;
89 
90 	size_t pos;
91 	u8 cmd[IO_BUFFER_LENGTH];
92 	u8 resp[IO_BUFFER_LENGTH];
93 
94 	struct parsed_resp parsed;
95 	size_t prev_d_len;
96 	void *prev_data;
97 
98 	struct list_head unlk_lst;
99 };
100 
101 
102 static const u8 opaluid[][OPAL_UID_LENGTH] = {
103 	/* users */
104 	[OPAL_SMUID_UID] =
105 		{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff },
106 	[OPAL_THISSP_UID] =
107 		{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01 },
108 	[OPAL_ADMINSP_UID] =
109 		{ 0x00, 0x00, 0x02, 0x05, 0x00, 0x00, 0x00, 0x01 },
110 	[OPAL_LOCKINGSP_UID] =
111 		{ 0x00, 0x00, 0x02, 0x05, 0x00, 0x00, 0x00, 0x02 },
112 	[OPAL_ENTERPRISE_LOCKINGSP_UID] =
113 		{ 0x00, 0x00, 0x02, 0x05, 0x00, 0x01, 0x00, 0x01 },
114 	[OPAL_ANYBODY_UID] =
115 		{ 0x00, 0x00, 0x00, 0x09, 0x00, 0x00, 0x00, 0x01 },
116 	[OPAL_SID_UID] =
117 		{ 0x00, 0x00, 0x00, 0x09, 0x00, 0x00, 0x00, 0x06 },
118 	[OPAL_ADMIN1_UID] =
119 		{ 0x00, 0x00, 0x00, 0x09, 0x00, 0x01, 0x00, 0x01 },
120 	[OPAL_USER1_UID] =
121 		{ 0x00, 0x00, 0x00, 0x09, 0x00, 0x03, 0x00, 0x01 },
122 	[OPAL_USER2_UID] =
123 		{ 0x00, 0x00, 0x00, 0x09, 0x00, 0x03, 0x00, 0x02 },
124 	[OPAL_PSID_UID] =
125 		{ 0x00, 0x00, 0x00, 0x09, 0x00, 0x01, 0xff, 0x01 },
126 	[OPAL_ENTERPRISE_BANDMASTER0_UID] =
127 		{ 0x00, 0x00, 0x00, 0x09, 0x00, 0x00, 0x80, 0x01 },
128 	[OPAL_ENTERPRISE_ERASEMASTER_UID] =
129 		{ 0x00, 0x00, 0x00, 0x09, 0x00, 0x00, 0x84, 0x01 },
130 
131 	/* tables */
132 
133 	[OPAL_TABLE_TABLE]
134 		{ 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x01 },
135 	[OPAL_LOCKINGRANGE_GLOBAL] =
136 		{ 0x00, 0x00, 0x08, 0x02, 0x00, 0x00, 0x00, 0x01 },
137 	[OPAL_LOCKINGRANGE_ACE_RDLOCKED] =
138 		{ 0x00, 0x00, 0x00, 0x08, 0x00, 0x03, 0xE0, 0x01 },
139 	[OPAL_LOCKINGRANGE_ACE_WRLOCKED] =
140 		{ 0x00, 0x00, 0x00, 0x08, 0x00, 0x03, 0xE8, 0x01 },
141 	[OPAL_MBRCONTROL] =
142 		{ 0x00, 0x00, 0x08, 0x03, 0x00, 0x00, 0x00, 0x01 },
143 	[OPAL_MBR] =
144 		{ 0x00, 0x00, 0x08, 0x04, 0x00, 0x00, 0x00, 0x00 },
145 	[OPAL_AUTHORITY_TABLE] =
146 		{ 0x00, 0x00, 0x00, 0x09, 0x00, 0x00, 0x00, 0x00},
147 	[OPAL_C_PIN_TABLE] =
148 		{ 0x00, 0x00, 0x00, 0x0B, 0x00, 0x00, 0x00, 0x00},
149 	[OPAL_LOCKING_INFO_TABLE] =
150 		{ 0x00, 0x00, 0x08, 0x01, 0x00, 0x00, 0x00, 0x01 },
151 	[OPAL_ENTERPRISE_LOCKING_INFO_TABLE] =
152 		{ 0x00, 0x00, 0x08, 0x01, 0x00, 0x00, 0x00, 0x00 },
153 
154 	/* C_PIN_TABLE object ID's */
155 
156 	[OPAL_C_PIN_MSID] =
157 		{ 0x00, 0x00, 0x00, 0x0B, 0x00, 0x00, 0x84, 0x02},
158 	[OPAL_C_PIN_SID] =
159 		{ 0x00, 0x00, 0x00, 0x0B, 0x00, 0x00, 0x00, 0x01},
160 	[OPAL_C_PIN_ADMIN1] =
161 		{ 0x00, 0x00, 0x00, 0x0B, 0x00, 0x01, 0x00, 0x01},
162 
163 	/* half UID's (only first 4 bytes used) */
164 
165 	[OPAL_HALF_UID_AUTHORITY_OBJ_REF] =
166 		{ 0x00, 0x00, 0x0C, 0x05, 0xff, 0xff, 0xff, 0xff },
167 	[OPAL_HALF_UID_BOOLEAN_ACE] =
168 		{ 0x00, 0x00, 0x04, 0x0E, 0xff, 0xff, 0xff, 0xff },
169 
170 	/* special value for omitted optional parameter */
171 	[OPAL_UID_HEXFF] =
172 		{ 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff},
173 };
174 
175 /*
176  * TCG Storage SSC Methods.
177  * Derived from: TCG_Storage_Architecture_Core_Spec_v2.01_r1.00
178  * Section: 6.3 Assigned UIDs
179  */
180 static const u8 opalmethod[][OPAL_METHOD_LENGTH] = {
181 	[OPAL_PROPERTIES] =
182 		{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff, 0x01 },
183 	[OPAL_STARTSESSION] =
184 		{ 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xff, 0x02 },
185 	[OPAL_REVERT] =
186 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x02, 0x02 },
187 	[OPAL_ACTIVATE] =
188 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x02, 0x03 },
189 	[OPAL_EGET] =
190 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x00, 0x06 },
191 	[OPAL_ESET] =
192 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x00, 0x07 },
193 	[OPAL_NEXT] =
194 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x00, 0x08 },
195 	[OPAL_EAUTHENTICATE] =
196 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x00, 0x0c },
197 	[OPAL_GETACL] =
198 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x00, 0x0d },
199 	[OPAL_GENKEY] =
200 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x00, 0x10 },
201 	[OPAL_REVERTSP] =
202 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x00, 0x11 },
203 	[OPAL_GET] =
204 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x00, 0x16 },
205 	[OPAL_SET] =
206 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x00, 0x17 },
207 	[OPAL_AUTHENTICATE] =
208 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x00, 0x1c },
209 	[OPAL_RANDOM] =
210 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x06, 0x01 },
211 	[OPAL_ERASE] =
212 		{ 0x00, 0x00, 0x00, 0x06, 0x00, 0x00, 0x08, 0x03 },
213 };
214 
215 static int end_opal_session_error(struct opal_dev *dev);
216 static int opal_discovery0_step(struct opal_dev *dev);
217 
218 struct opal_suspend_data {
219 	struct opal_lock_unlock unlk;
220 	u8 lr;
221 	struct list_head node;
222 };
223 
224 /*
225  * Derived from:
226  * TCG_Storage_Architecture_Core_Spec_v2.01_r1.00
227  * Section: 5.1.5 Method Status Codes
228  */
229 static const char * const opal_errors[] = {
230 	"Success",
231 	"Not Authorized",
232 	"Unknown Error",
233 	"SP Busy",
234 	"SP Failed",
235 	"SP Disabled",
236 	"SP Frozen",
237 	"No Sessions Available",
238 	"Uniqueness Conflict",
239 	"Insufficient Space",
240 	"Insufficient Rows",
241 	"Invalid Function",
242 	"Invalid Parameter",
243 	"Invalid Reference",
244 	"Unknown Error",
245 	"TPER Malfunction",
246 	"Transaction Failure",
247 	"Response Overflow",
248 	"Authority Locked Out",
249 };
250 
251 static const char *opal_error_to_human(int error)
252 {
253 	if (error == 0x3f)
254 		return "Failed";
255 
256 	if (error >= ARRAY_SIZE(opal_errors) || error < 0)
257 		return "Unknown Error";
258 
259 	return opal_errors[error];
260 }
261 
262 static void print_buffer(const u8 *ptr, u32 length)
263 {
264 #ifdef DEBUG
265 	print_hex_dump_bytes("OPAL: ", DUMP_PREFIX_OFFSET, ptr, length);
266 	pr_debug("\n");
267 #endif
268 }
269 
270 static bool check_tper(const void *data)
271 {
272 	const struct d0_tper_features *tper = data;
273 	u8 flags = tper->supported_features;
274 
275 	if (!(flags & TPER_SYNC_SUPPORTED)) {
276 		pr_debug("TPer sync not supported. flags = %d\n",
277 			 tper->supported_features);
278 		return false;
279 	}
280 
281 	return true;
282 }
283 
284 static bool check_mbrenabled(const void *data)
285 {
286 	const struct d0_locking_features *lfeat = data;
287 	u8 sup_feat = lfeat->supported_features;
288 
289 	return !!(sup_feat & MBR_ENABLED_MASK);
290 }
291 
292 static bool check_sum(const void *data)
293 {
294 	const struct d0_single_user_mode *sum = data;
295 	u32 nlo = be32_to_cpu(sum->num_locking_objects);
296 
297 	if (nlo == 0) {
298 		pr_debug("Need at least one locking object.\n");
299 		return false;
300 	}
301 
302 	pr_debug("Number of locking objects: %d\n", nlo);
303 
304 	return true;
305 }
306 
307 static u16 get_comid_v100(const void *data)
308 {
309 	const struct d0_opal_v100 *v100 = data;
310 
311 	return be16_to_cpu(v100->baseComID);
312 }
313 
314 static u16 get_comid_v200(const void *data)
315 {
316 	const struct d0_opal_v200 *v200 = data;
317 
318 	return be16_to_cpu(v200->baseComID);
319 }
320 
321 static int opal_send_cmd(struct opal_dev *dev)
322 {
323 	return dev->send_recv(dev->data, dev->comid, TCG_SECP_01,
324 			      dev->cmd, IO_BUFFER_LENGTH,
325 			      true);
326 }
327 
328 static int opal_recv_cmd(struct opal_dev *dev)
329 {
330 	return dev->send_recv(dev->data, dev->comid, TCG_SECP_01,
331 			      dev->resp, IO_BUFFER_LENGTH,
332 			      false);
333 }
334 
335 static int opal_recv_check(struct opal_dev *dev)
336 {
337 	size_t buflen = IO_BUFFER_LENGTH;
338 	void *buffer = dev->resp;
339 	struct opal_header *hdr = buffer;
340 	int ret;
341 
342 	do {
343 		pr_debug("Sent OPAL command: outstanding=%d, minTransfer=%d\n",
344 			 hdr->cp.outstandingData,
345 			 hdr->cp.minTransfer);
346 
347 		if (hdr->cp.outstandingData == 0 ||
348 		    hdr->cp.minTransfer != 0)
349 			return 0;
350 
351 		memset(buffer, 0, buflen);
352 		ret = opal_recv_cmd(dev);
353 	} while (!ret);
354 
355 	return ret;
356 }
357 
358 static int opal_send_recv(struct opal_dev *dev, cont_fn *cont)
359 {
360 	int ret;
361 
362 	ret = opal_send_cmd(dev);
363 	if (ret)
364 		return ret;
365 	ret = opal_recv_cmd(dev);
366 	if (ret)
367 		return ret;
368 	ret = opal_recv_check(dev);
369 	if (ret)
370 		return ret;
371 	return cont(dev);
372 }
373 
374 static void check_geometry(struct opal_dev *dev, const void *data)
375 {
376 	const struct d0_geometry_features *geo = data;
377 
378 	dev->align = geo->alignment_granularity;
379 	dev->lowest_lba = geo->lowest_aligned_lba;
380 }
381 
382 static int execute_step(struct opal_dev *dev,
383 			const struct opal_step *step, size_t stepIndex)
384 {
385 	int error = step->fn(dev, step->data);
386 
387 	if (error) {
388 		pr_debug("Step %zu (%pS) failed with error %d: %s\n",
389 			 stepIndex, step->fn, error,
390 			 opal_error_to_human(error));
391 	}
392 
393 	return error;
394 }
395 
396 static int execute_steps(struct opal_dev *dev,
397 			 const struct opal_step *steps, size_t n_steps)
398 {
399 	size_t state = 0;
400 	int error;
401 
402 	/* first do a discovery0 */
403 	error = opal_discovery0_step(dev);
404 	if (error)
405 		return error;
406 
407 	for (state = 0; state < n_steps; state++) {
408 		error = execute_step(dev, &steps[state], state);
409 		if (error)
410 			goto out_error;
411 	}
412 
413 	return 0;
414 
415 out_error:
416 	/*
417 	 * For each OPAL command the first step in steps starts some sort of
418 	 * session. If an error occurred in the initial discovery0 or if an
419 	 * error occurred in the first step (and thus stopping the loop with
420 	 * state == 0) then there was an error before or during the attempt to
421 	 * start a session. Therefore we shouldn't attempt to terminate a
422 	 * session, as one has not yet been created.
423 	 */
424 	if (state > 0)
425 		end_opal_session_error(dev);
426 
427 	return error;
428 }
429 
430 static int opal_discovery0_end(struct opal_dev *dev)
431 {
432 	bool found_com_id = false, supported = true, single_user = false;
433 	const struct d0_header *hdr = (struct d0_header *)dev->resp;
434 	const u8 *epos = dev->resp, *cpos = dev->resp;
435 	u16 comid = 0;
436 	u32 hlen = be32_to_cpu(hdr->length);
437 
438 	print_buffer(dev->resp, hlen);
439 	dev->mbr_enabled = false;
440 
441 	if (hlen > IO_BUFFER_LENGTH - sizeof(*hdr)) {
442 		pr_debug("Discovery length overflows buffer (%zu+%u)/%u\n",
443 			 sizeof(*hdr), hlen, IO_BUFFER_LENGTH);
444 		return -EFAULT;
445 	}
446 
447 	epos += hlen; /* end of buffer */
448 	cpos += sizeof(*hdr); /* current position on buffer */
449 
450 	while (cpos < epos && supported) {
451 		const struct d0_features *body =
452 			(const struct d0_features *)cpos;
453 
454 		switch (be16_to_cpu(body->code)) {
455 		case FC_TPER:
456 			supported = check_tper(body->features);
457 			break;
458 		case FC_SINGLEUSER:
459 			single_user = check_sum(body->features);
460 			break;
461 		case FC_GEOMETRY:
462 			check_geometry(dev, body);
463 			break;
464 		case FC_LOCKING:
465 			dev->mbr_enabled = check_mbrenabled(body->features);
466 			break;
467 		case FC_ENTERPRISE:
468 		case FC_DATASTORE:
469 			/* some ignored properties */
470 			pr_debug("Found OPAL feature description: %d\n",
471 				 be16_to_cpu(body->code));
472 			break;
473 		case FC_OPALV100:
474 			comid = get_comid_v100(body->features);
475 			found_com_id = true;
476 			break;
477 		case FC_OPALV200:
478 			comid = get_comid_v200(body->features);
479 			found_com_id = true;
480 			break;
481 		case 0xbfff ... 0xffff:
482 			/* vendor specific, just ignore */
483 			break;
484 		default:
485 			pr_debug("OPAL Unknown feature: %d\n",
486 				 be16_to_cpu(body->code));
487 
488 		}
489 		cpos += body->length + 4;
490 	}
491 
492 	if (!supported) {
493 		pr_debug("This device is not Opal enabled. Not Supported!\n");
494 		return -EOPNOTSUPP;
495 	}
496 
497 	if (!single_user)
498 		pr_debug("Device doesn't support single user mode\n");
499 
500 
501 	if (!found_com_id) {
502 		pr_debug("Could not find OPAL comid for device. Returning early\n");
503 		return -EOPNOTSUPP;
504 	}
505 
506 	dev->comid = comid;
507 
508 	return 0;
509 }
510 
511 static int opal_discovery0(struct opal_dev *dev, void *data)
512 {
513 	int ret;
514 
515 	memset(dev->resp, 0, IO_BUFFER_LENGTH);
516 	dev->comid = OPAL_DISCOVERY_COMID;
517 	ret = opal_recv_cmd(dev);
518 	if (ret)
519 		return ret;
520 	return opal_discovery0_end(dev);
521 }
522 
523 static int opal_discovery0_step(struct opal_dev *dev)
524 {
525 	const struct opal_step discovery0_step = {
526 		opal_discovery0,
527 	};
528 	return execute_step(dev, &discovery0_step, 0);
529 }
530 
531 static size_t remaining_size(struct opal_dev *cmd)
532 {
533 	return IO_BUFFER_LENGTH - cmd->pos;
534 }
535 
536 static bool can_add(int *err, struct opal_dev *cmd, size_t len)
537 {
538 	if (*err)
539 		return false;
540 
541 	if (remaining_size(cmd) < len) {
542 		pr_debug("Error adding %zu bytes: end of buffer.\n", len);
543 		*err = -ERANGE;
544 		return false;
545 	}
546 
547 	return true;
548 }
549 
550 static void add_token_u8(int *err, struct opal_dev *cmd, u8 tok)
551 {
552 	if (!can_add(err, cmd, 1))
553 		return;
554 	cmd->cmd[cmd->pos++] = tok;
555 }
556 
557 static void add_short_atom_header(struct opal_dev *cmd, bool bytestring,
558 				  bool has_sign, int len)
559 {
560 	u8 atom;
561 	int err = 0;
562 
563 	atom = SHORT_ATOM_ID;
564 	atom |= bytestring ? SHORT_ATOM_BYTESTRING : 0;
565 	atom |= has_sign ? SHORT_ATOM_SIGNED : 0;
566 	atom |= len & SHORT_ATOM_LEN_MASK;
567 
568 	add_token_u8(&err, cmd, atom);
569 }
570 
571 static void add_medium_atom_header(struct opal_dev *cmd, bool bytestring,
572 				   bool has_sign, int len)
573 {
574 	u8 header0;
575 
576 	header0 = MEDIUM_ATOM_ID;
577 	header0 |= bytestring ? MEDIUM_ATOM_BYTESTRING : 0;
578 	header0 |= has_sign ? MEDIUM_ATOM_SIGNED : 0;
579 	header0 |= (len >> 8) & MEDIUM_ATOM_LEN_MASK;
580 	cmd->cmd[cmd->pos++] = header0;
581 	cmd->cmd[cmd->pos++] = len;
582 }
583 
584 static void add_token_u64(int *err, struct opal_dev *cmd, u64 number)
585 {
586 	size_t len;
587 	int msb;
588 
589 	if (!(number & ~TINY_ATOM_DATA_MASK)) {
590 		add_token_u8(err, cmd, number);
591 		return;
592 	}
593 
594 	msb = fls64(number);
595 	len = DIV_ROUND_UP(msb, 8);
596 
597 	if (!can_add(err, cmd, len + 1)) {
598 		pr_debug("Error adding u64: end of buffer.\n");
599 		return;
600 	}
601 	add_short_atom_header(cmd, false, false, len);
602 	while (len--)
603 		add_token_u8(err, cmd, number >> (len * 8));
604 }
605 
606 static u8 *add_bytestring_header(int *err, struct opal_dev *cmd, size_t len)
607 {
608 	size_t header_len = 1;
609 	bool is_short_atom = true;
610 
611 	if (len & ~SHORT_ATOM_LEN_MASK) {
612 		header_len = 2;
613 		is_short_atom = false;
614 	}
615 
616 	if (!can_add(err, cmd, header_len + len)) {
617 		pr_debug("Error adding bytestring: end of buffer.\n");
618 		return NULL;
619 	}
620 
621 	if (is_short_atom)
622 		add_short_atom_header(cmd, true, false, len);
623 	else
624 		add_medium_atom_header(cmd, true, false, len);
625 
626 	return &cmd->cmd[cmd->pos];
627 }
628 
629 static void add_token_bytestring(int *err, struct opal_dev *cmd,
630 				 const u8 *bytestring, size_t len)
631 {
632 	u8 *start;
633 
634 	start = add_bytestring_header(err, cmd, len);
635 	if (!start)
636 		return;
637 	memcpy(start, bytestring, len);
638 	cmd->pos += len;
639 }
640 
641 static int build_locking_range(u8 *buffer, size_t length, u8 lr)
642 {
643 	if (length > OPAL_UID_LENGTH) {
644 		pr_debug("Can't build locking range. Length OOB\n");
645 		return -ERANGE;
646 	}
647 
648 	memcpy(buffer, opaluid[OPAL_LOCKINGRANGE_GLOBAL], OPAL_UID_LENGTH);
649 
650 	if (lr == 0)
651 		return 0;
652 	buffer[5] = LOCKING_RANGE_NON_GLOBAL;
653 	buffer[7] = lr;
654 
655 	return 0;
656 }
657 
658 static int build_locking_user(u8 *buffer, size_t length, u8 lr)
659 {
660 	if (length > OPAL_UID_LENGTH) {
661 		pr_debug("Can't build locking range user. Length OOB\n");
662 		return -ERANGE;
663 	}
664 
665 	memcpy(buffer, opaluid[OPAL_USER1_UID], OPAL_UID_LENGTH);
666 
667 	buffer[7] = lr + 1;
668 
669 	return 0;
670 }
671 
672 static void set_comid(struct opal_dev *cmd, u16 comid)
673 {
674 	struct opal_header *hdr = (struct opal_header *)cmd->cmd;
675 
676 	hdr->cp.extendedComID[0] = comid >> 8;
677 	hdr->cp.extendedComID[1] = comid;
678 	hdr->cp.extendedComID[2] = 0;
679 	hdr->cp.extendedComID[3] = 0;
680 }
681 
682 static int cmd_finalize(struct opal_dev *cmd, u32 hsn, u32 tsn)
683 {
684 	struct opal_header *hdr;
685 	int err = 0;
686 
687 	/*
688 	 * Close the parameter list opened from cmd_start.
689 	 * The number of bytes added must be equal to
690 	 * CMD_FINALIZE_BYTES_NEEDED.
691 	 */
692 	add_token_u8(&err, cmd, OPAL_ENDLIST);
693 
694 	add_token_u8(&err, cmd, OPAL_ENDOFDATA);
695 	add_token_u8(&err, cmd, OPAL_STARTLIST);
696 	add_token_u8(&err, cmd, 0);
697 	add_token_u8(&err, cmd, 0);
698 	add_token_u8(&err, cmd, 0);
699 	add_token_u8(&err, cmd, OPAL_ENDLIST);
700 
701 	if (err) {
702 		pr_debug("Error finalizing command.\n");
703 		return -EFAULT;
704 	}
705 
706 	hdr = (struct opal_header *) cmd->cmd;
707 
708 	hdr->pkt.tsn = cpu_to_be32(tsn);
709 	hdr->pkt.hsn = cpu_to_be32(hsn);
710 
711 	hdr->subpkt.length = cpu_to_be32(cmd->pos - sizeof(*hdr));
712 	while (cmd->pos % 4) {
713 		if (cmd->pos >= IO_BUFFER_LENGTH) {
714 			pr_debug("Error: Buffer overrun\n");
715 			return -ERANGE;
716 		}
717 		cmd->cmd[cmd->pos++] = 0;
718 	}
719 	hdr->pkt.length = cpu_to_be32(cmd->pos - sizeof(hdr->cp) -
720 				      sizeof(hdr->pkt));
721 	hdr->cp.length = cpu_to_be32(cmd->pos - sizeof(hdr->cp));
722 
723 	return 0;
724 }
725 
726 static const struct opal_resp_tok *response_get_token(
727 				const struct parsed_resp *resp,
728 				int n)
729 {
730 	const struct opal_resp_tok *tok;
731 
732 	if (!resp) {
733 		pr_debug("Response is NULL\n");
734 		return ERR_PTR(-EINVAL);
735 	}
736 
737 	if (n >= resp->num) {
738 		pr_debug("Token number doesn't exist: %d, resp: %d\n",
739 			 n, resp->num);
740 		return ERR_PTR(-EINVAL);
741 	}
742 
743 	tok = &resp->toks[n];
744 	if (tok->len == 0) {
745 		pr_debug("Token length must be non-zero\n");
746 		return ERR_PTR(-EINVAL);
747 	}
748 
749 	return tok;
750 }
751 
752 static ssize_t response_parse_tiny(struct opal_resp_tok *tok,
753 				   const u8 *pos)
754 {
755 	tok->pos = pos;
756 	tok->len = 1;
757 	tok->width = OPAL_WIDTH_TINY;
758 
759 	if (pos[0] & TINY_ATOM_SIGNED) {
760 		tok->type = OPAL_DTA_TOKENID_SINT;
761 	} else {
762 		tok->type = OPAL_DTA_TOKENID_UINT;
763 		tok->stored.u = pos[0] & 0x3f;
764 	}
765 
766 	return tok->len;
767 }
768 
769 static ssize_t response_parse_short(struct opal_resp_tok *tok,
770 				    const u8 *pos)
771 {
772 	tok->pos = pos;
773 	tok->len = (pos[0] & SHORT_ATOM_LEN_MASK) + 1;
774 	tok->width = OPAL_WIDTH_SHORT;
775 
776 	if (pos[0] & SHORT_ATOM_BYTESTRING) {
777 		tok->type = OPAL_DTA_TOKENID_BYTESTRING;
778 	} else if (pos[0] & SHORT_ATOM_SIGNED) {
779 		tok->type = OPAL_DTA_TOKENID_SINT;
780 	} else {
781 		u64 u_integer = 0;
782 		ssize_t i, b = 0;
783 
784 		tok->type = OPAL_DTA_TOKENID_UINT;
785 		if (tok->len > 9) {
786 			pr_debug("uint64 with more than 8 bytes\n");
787 			return -EINVAL;
788 		}
789 		for (i = tok->len - 1; i > 0; i--) {
790 			u_integer |= ((u64)pos[i] << (8 * b));
791 			b++;
792 		}
793 		tok->stored.u = u_integer;
794 	}
795 
796 	return tok->len;
797 }
798 
799 static ssize_t response_parse_medium(struct opal_resp_tok *tok,
800 				     const u8 *pos)
801 {
802 	tok->pos = pos;
803 	tok->len = (((pos[0] & MEDIUM_ATOM_LEN_MASK) << 8) | pos[1]) + 2;
804 	tok->width = OPAL_WIDTH_MEDIUM;
805 
806 	if (pos[0] & MEDIUM_ATOM_BYTESTRING)
807 		tok->type = OPAL_DTA_TOKENID_BYTESTRING;
808 	else if (pos[0] & MEDIUM_ATOM_SIGNED)
809 		tok->type = OPAL_DTA_TOKENID_SINT;
810 	else
811 		tok->type = OPAL_DTA_TOKENID_UINT;
812 
813 	return tok->len;
814 }
815 
816 static ssize_t response_parse_long(struct opal_resp_tok *tok,
817 				   const u8 *pos)
818 {
819 	tok->pos = pos;
820 	tok->len = ((pos[1] << 16) | (pos[2] << 8) | pos[3]) + 4;
821 	tok->width = OPAL_WIDTH_LONG;
822 
823 	if (pos[0] & LONG_ATOM_BYTESTRING)
824 		tok->type = OPAL_DTA_TOKENID_BYTESTRING;
825 	else if (pos[0] & LONG_ATOM_SIGNED)
826 		tok->type = OPAL_DTA_TOKENID_SINT;
827 	else
828 		tok->type = OPAL_DTA_TOKENID_UINT;
829 
830 	return tok->len;
831 }
832 
833 static ssize_t response_parse_token(struct opal_resp_tok *tok,
834 				    const u8 *pos)
835 {
836 	tok->pos = pos;
837 	tok->len = 1;
838 	tok->type = OPAL_DTA_TOKENID_TOKEN;
839 	tok->width = OPAL_WIDTH_TOKEN;
840 
841 	return tok->len;
842 }
843 
844 static int response_parse(const u8 *buf, size_t length,
845 			  struct parsed_resp *resp)
846 {
847 	const struct opal_header *hdr;
848 	struct opal_resp_tok *iter;
849 	int num_entries = 0;
850 	int total;
851 	ssize_t token_length;
852 	const u8 *pos;
853 	u32 clen, plen, slen;
854 
855 	if (!buf)
856 		return -EFAULT;
857 
858 	if (!resp)
859 		return -EFAULT;
860 
861 	hdr = (struct opal_header *)buf;
862 	pos = buf;
863 	pos += sizeof(*hdr);
864 
865 	clen = be32_to_cpu(hdr->cp.length);
866 	plen = be32_to_cpu(hdr->pkt.length);
867 	slen = be32_to_cpu(hdr->subpkt.length);
868 	pr_debug("Response size: cp: %u, pkt: %u, subpkt: %u\n",
869 		 clen, plen, slen);
870 
871 	if (clen == 0 || plen == 0 || slen == 0 ||
872 	    slen > IO_BUFFER_LENGTH - sizeof(*hdr)) {
873 		pr_debug("Bad header length. cp: %u, pkt: %u, subpkt: %u\n",
874 			 clen, plen, slen);
875 		print_buffer(pos, sizeof(*hdr));
876 		return -EINVAL;
877 	}
878 
879 	if (pos > buf + length)
880 		return -EFAULT;
881 
882 	iter = resp->toks;
883 	total = slen;
884 	print_buffer(pos, total);
885 	while (total > 0) {
886 		if (pos[0] <= TINY_ATOM_BYTE) /* tiny atom */
887 			token_length = response_parse_tiny(iter, pos);
888 		else if (pos[0] <= SHORT_ATOM_BYTE) /* short atom */
889 			token_length = response_parse_short(iter, pos);
890 		else if (pos[0] <= MEDIUM_ATOM_BYTE) /* medium atom */
891 			token_length = response_parse_medium(iter, pos);
892 		else if (pos[0] <= LONG_ATOM_BYTE) /* long atom */
893 			token_length = response_parse_long(iter, pos);
894 		else /* TOKEN */
895 			token_length = response_parse_token(iter, pos);
896 
897 		if (token_length < 0)
898 			return token_length;
899 
900 		pos += token_length;
901 		total -= token_length;
902 		iter++;
903 		num_entries++;
904 	}
905 
906 	if (num_entries == 0) {
907 		pr_debug("Couldn't parse response.\n");
908 		return -EINVAL;
909 	}
910 	resp->num = num_entries;
911 
912 	return 0;
913 }
914 
915 static size_t response_get_string(const struct parsed_resp *resp, int n,
916 				  const char **store)
917 {
918 	u8 skip;
919 	const struct opal_resp_tok *tok;
920 
921 	*store = NULL;
922 	tok = response_get_token(resp, n);
923 	if (IS_ERR(tok))
924 		return 0;
925 
926 	if (tok->type != OPAL_DTA_TOKENID_BYTESTRING) {
927 		pr_debug("Token is not a byte string!\n");
928 		return 0;
929 	}
930 
931 	switch (tok->width) {
932 	case OPAL_WIDTH_TINY:
933 	case OPAL_WIDTH_SHORT:
934 		skip = 1;
935 		break;
936 	case OPAL_WIDTH_MEDIUM:
937 		skip = 2;
938 		break;
939 	case OPAL_WIDTH_LONG:
940 		skip = 4;
941 		break;
942 	default:
943 		pr_debug("Token has invalid width!\n");
944 		return 0;
945 	}
946 
947 	*store = tok->pos + skip;
948 	return tok->len - skip;
949 }
950 
951 static u64 response_get_u64(const struct parsed_resp *resp, int n)
952 {
953 	const struct opal_resp_tok *tok;
954 
955 	tok = response_get_token(resp, n);
956 	if (IS_ERR(tok))
957 		return 0;
958 
959 	if (tok->type != OPAL_DTA_TOKENID_UINT) {
960 		pr_debug("Token is not unsigned int: %d\n", tok->type);
961 		return 0;
962 	}
963 
964 	if (tok->width != OPAL_WIDTH_TINY && tok->width != OPAL_WIDTH_SHORT) {
965 		pr_debug("Atom is not short or tiny: %d\n", tok->width);
966 		return 0;
967 	}
968 
969 	return tok->stored.u;
970 }
971 
972 static bool response_token_matches(const struct opal_resp_tok *token, u8 match)
973 {
974 	if (IS_ERR(token) ||
975 	    token->type != OPAL_DTA_TOKENID_TOKEN ||
976 	    token->pos[0] != match)
977 		return false;
978 	return true;
979 }
980 
981 static u8 response_status(const struct parsed_resp *resp)
982 {
983 	const struct opal_resp_tok *tok;
984 
985 	tok = response_get_token(resp, 0);
986 	if (response_token_matches(tok, OPAL_ENDOFSESSION))
987 		return 0;
988 
989 	if (resp->num < 5)
990 		return DTAERROR_NO_METHOD_STATUS;
991 
992 	tok = response_get_token(resp, resp->num - 5);
993 	if (!response_token_matches(tok, OPAL_STARTLIST))
994 		return DTAERROR_NO_METHOD_STATUS;
995 
996 	tok = response_get_token(resp, resp->num - 1);
997 	if (!response_token_matches(tok, OPAL_ENDLIST))
998 		return DTAERROR_NO_METHOD_STATUS;
999 
1000 	return response_get_u64(resp, resp->num - 4);
1001 }
1002 
1003 /* Parses and checks for errors */
1004 static int parse_and_check_status(struct opal_dev *dev)
1005 {
1006 	int error;
1007 
1008 	print_buffer(dev->cmd, dev->pos);
1009 
1010 	error = response_parse(dev->resp, IO_BUFFER_LENGTH, &dev->parsed);
1011 	if (error) {
1012 		pr_debug("Couldn't parse response.\n");
1013 		return error;
1014 	}
1015 
1016 	return response_status(&dev->parsed);
1017 }
1018 
1019 static void clear_opal_cmd(struct opal_dev *dev)
1020 {
1021 	dev->pos = sizeof(struct opal_header);
1022 	memset(dev->cmd, 0, IO_BUFFER_LENGTH);
1023 }
1024 
1025 static int cmd_start(struct opal_dev *dev, const u8 *uid, const u8 *method)
1026 {
1027 	int err = 0;
1028 
1029 	clear_opal_cmd(dev);
1030 	set_comid(dev, dev->comid);
1031 
1032 	add_token_u8(&err, dev, OPAL_CALL);
1033 	add_token_bytestring(&err, dev, uid, OPAL_UID_LENGTH);
1034 	add_token_bytestring(&err, dev, method, OPAL_METHOD_LENGTH);
1035 
1036 	/*
1037 	 * Every method call is followed by its parameters enclosed within
1038 	 * OPAL_STARTLIST and OPAL_ENDLIST tokens. We automatically open the
1039 	 * parameter list here and close it later in cmd_finalize.
1040 	 */
1041 	add_token_u8(&err, dev, OPAL_STARTLIST);
1042 
1043 	return err;
1044 }
1045 
1046 static int start_opal_session_cont(struct opal_dev *dev)
1047 {
1048 	u32 hsn, tsn;
1049 	int error = 0;
1050 
1051 	error = parse_and_check_status(dev);
1052 	if (error)
1053 		return error;
1054 
1055 	hsn = response_get_u64(&dev->parsed, 4);
1056 	tsn = response_get_u64(&dev->parsed, 5);
1057 
1058 	if (hsn == 0 && tsn == 0) {
1059 		pr_debug("Couldn't authenticate session\n");
1060 		return -EPERM;
1061 	}
1062 
1063 	dev->hsn = hsn;
1064 	dev->tsn = tsn;
1065 	return 0;
1066 }
1067 
1068 static void add_suspend_info(struct opal_dev *dev,
1069 			     struct opal_suspend_data *sus)
1070 {
1071 	struct opal_suspend_data *iter;
1072 
1073 	list_for_each_entry(iter, &dev->unlk_lst, node) {
1074 		if (iter->lr == sus->lr) {
1075 			list_del(&iter->node);
1076 			kfree(iter);
1077 			break;
1078 		}
1079 	}
1080 	list_add_tail(&sus->node, &dev->unlk_lst);
1081 }
1082 
1083 static int end_session_cont(struct opal_dev *dev)
1084 {
1085 	dev->hsn = 0;
1086 	dev->tsn = 0;
1087 	return parse_and_check_status(dev);
1088 }
1089 
1090 static int finalize_and_send(struct opal_dev *dev, cont_fn cont)
1091 {
1092 	int ret;
1093 
1094 	ret = cmd_finalize(dev, dev->hsn, dev->tsn);
1095 	if (ret) {
1096 		pr_debug("Error finalizing command buffer: %d\n", ret);
1097 		return ret;
1098 	}
1099 
1100 	print_buffer(dev->cmd, dev->pos);
1101 
1102 	return opal_send_recv(dev, cont);
1103 }
1104 
1105 /*
1106  * request @column from table @table on device @dev. On success, the column
1107  * data will be available in dev->resp->tok[4]
1108  */
1109 static int generic_get_column(struct opal_dev *dev, const u8 *table,
1110 			      u64 column)
1111 {
1112 	int err;
1113 
1114 	err = cmd_start(dev, table, opalmethod[OPAL_GET]);
1115 
1116 	add_token_u8(&err, dev, OPAL_STARTLIST);
1117 
1118 	add_token_u8(&err, dev, OPAL_STARTNAME);
1119 	add_token_u8(&err, dev, OPAL_STARTCOLUMN);
1120 	add_token_u64(&err, dev, column);
1121 	add_token_u8(&err, dev, OPAL_ENDNAME);
1122 
1123 	add_token_u8(&err, dev, OPAL_STARTNAME);
1124 	add_token_u8(&err, dev, OPAL_ENDCOLUMN);
1125 	add_token_u64(&err, dev, column);
1126 	add_token_u8(&err, dev, OPAL_ENDNAME);
1127 
1128 	add_token_u8(&err, dev, OPAL_ENDLIST);
1129 
1130 	if (err)
1131 		return err;
1132 
1133 	return finalize_and_send(dev, parse_and_check_status);
1134 }
1135 
1136 /*
1137  * see TCG SAS 5.3.2.3 for a description of the available columns
1138  *
1139  * the result is provided in dev->resp->tok[4]
1140  */
1141 static int generic_get_table_info(struct opal_dev *dev, enum opal_uid table,
1142 				  u64 column)
1143 {
1144 	u8 uid[OPAL_UID_LENGTH];
1145 	const unsigned int half = OPAL_UID_LENGTH/2;
1146 
1147 	/* sed-opal UIDs can be split in two halves:
1148 	 *  first:  actual table index
1149 	 *  second: relative index in the table
1150 	 * so we have to get the first half of the OPAL_TABLE_TABLE and use the
1151 	 * first part of the target table as relative index into that table
1152 	 */
1153 	memcpy(uid, opaluid[OPAL_TABLE_TABLE], half);
1154 	memcpy(uid+half, opaluid[table], half);
1155 
1156 	return generic_get_column(dev, uid, column);
1157 }
1158 
1159 static int gen_key(struct opal_dev *dev, void *data)
1160 {
1161 	u8 uid[OPAL_UID_LENGTH];
1162 	int err;
1163 
1164 	memcpy(uid, dev->prev_data, min(sizeof(uid), dev->prev_d_len));
1165 	kfree(dev->prev_data);
1166 	dev->prev_data = NULL;
1167 
1168 	err = cmd_start(dev, uid, opalmethod[OPAL_GENKEY]);
1169 
1170 	if (err) {
1171 		pr_debug("Error building gen key command\n");
1172 		return err;
1173 
1174 	}
1175 	return finalize_and_send(dev, parse_and_check_status);
1176 }
1177 
1178 static int get_active_key_cont(struct opal_dev *dev)
1179 {
1180 	const char *activekey;
1181 	size_t keylen;
1182 	int error = 0;
1183 
1184 	error = parse_and_check_status(dev);
1185 	if (error)
1186 		return error;
1187 	keylen = response_get_string(&dev->parsed, 4, &activekey);
1188 	if (!activekey) {
1189 		pr_debug("%s: Couldn't extract the Activekey from the response\n",
1190 			 __func__);
1191 		return OPAL_INVAL_PARAM;
1192 	}
1193 	dev->prev_data = kmemdup(activekey, keylen, GFP_KERNEL);
1194 
1195 	if (!dev->prev_data)
1196 		return -ENOMEM;
1197 
1198 	dev->prev_d_len = keylen;
1199 
1200 	return 0;
1201 }
1202 
1203 static int get_active_key(struct opal_dev *dev, void *data)
1204 {
1205 	u8 uid[OPAL_UID_LENGTH];
1206 	int err;
1207 	u8 *lr = data;
1208 
1209 	err = build_locking_range(uid, sizeof(uid), *lr);
1210 	if (err)
1211 		return err;
1212 
1213 	err = generic_get_column(dev, uid, OPAL_ACTIVEKEY);
1214 	if (err)
1215 		return err;
1216 
1217 	return get_active_key_cont(dev);
1218 }
1219 
1220 static int generic_lr_enable_disable(struct opal_dev *dev,
1221 				     u8 *uid, bool rle, bool wle,
1222 				     bool rl, bool wl)
1223 {
1224 	int err;
1225 
1226 	err = cmd_start(dev, uid, opalmethod[OPAL_SET]);
1227 
1228 	add_token_u8(&err, dev, OPAL_STARTNAME);
1229 	add_token_u8(&err, dev, OPAL_VALUES);
1230 	add_token_u8(&err, dev, OPAL_STARTLIST);
1231 
1232 	add_token_u8(&err, dev, OPAL_STARTNAME);
1233 	add_token_u8(&err, dev, OPAL_READLOCKENABLED);
1234 	add_token_u8(&err, dev, rle);
1235 	add_token_u8(&err, dev, OPAL_ENDNAME);
1236 
1237 	add_token_u8(&err, dev, OPAL_STARTNAME);
1238 	add_token_u8(&err, dev, OPAL_WRITELOCKENABLED);
1239 	add_token_u8(&err, dev, wle);
1240 	add_token_u8(&err, dev, OPAL_ENDNAME);
1241 
1242 	add_token_u8(&err, dev, OPAL_STARTNAME);
1243 	add_token_u8(&err, dev, OPAL_READLOCKED);
1244 	add_token_u8(&err, dev, rl);
1245 	add_token_u8(&err, dev, OPAL_ENDNAME);
1246 
1247 	add_token_u8(&err, dev, OPAL_STARTNAME);
1248 	add_token_u8(&err, dev, OPAL_WRITELOCKED);
1249 	add_token_u8(&err, dev, wl);
1250 	add_token_u8(&err, dev, OPAL_ENDNAME);
1251 
1252 	add_token_u8(&err, dev, OPAL_ENDLIST);
1253 	add_token_u8(&err, dev, OPAL_ENDNAME);
1254 	return err;
1255 }
1256 
1257 static inline int enable_global_lr(struct opal_dev *dev, u8 *uid,
1258 				   struct opal_user_lr_setup *setup)
1259 {
1260 	int err;
1261 
1262 	err = generic_lr_enable_disable(dev, uid, !!setup->RLE, !!setup->WLE,
1263 					0, 0);
1264 	if (err)
1265 		pr_debug("Failed to create enable global lr command\n");
1266 	return err;
1267 }
1268 
1269 static int setup_locking_range(struct opal_dev *dev, void *data)
1270 {
1271 	u8 uid[OPAL_UID_LENGTH];
1272 	struct opal_user_lr_setup *setup = data;
1273 	u8 lr;
1274 	int err;
1275 
1276 	lr = setup->session.opal_key.lr;
1277 	err = build_locking_range(uid, sizeof(uid), lr);
1278 	if (err)
1279 		return err;
1280 
1281 	if (lr == 0)
1282 		err = enable_global_lr(dev, uid, setup);
1283 	else {
1284 		err = cmd_start(dev, uid, opalmethod[OPAL_SET]);
1285 
1286 		add_token_u8(&err, dev, OPAL_STARTNAME);
1287 		add_token_u8(&err, dev, OPAL_VALUES);
1288 		add_token_u8(&err, dev, OPAL_STARTLIST);
1289 
1290 		add_token_u8(&err, dev, OPAL_STARTNAME);
1291 		add_token_u8(&err, dev, OPAL_RANGESTART);
1292 		add_token_u64(&err, dev, setup->range_start);
1293 		add_token_u8(&err, dev, OPAL_ENDNAME);
1294 
1295 		add_token_u8(&err, dev, OPAL_STARTNAME);
1296 		add_token_u8(&err, dev, OPAL_RANGELENGTH);
1297 		add_token_u64(&err, dev, setup->range_length);
1298 		add_token_u8(&err, dev, OPAL_ENDNAME);
1299 
1300 		add_token_u8(&err, dev, OPAL_STARTNAME);
1301 		add_token_u8(&err, dev, OPAL_READLOCKENABLED);
1302 		add_token_u64(&err, dev, !!setup->RLE);
1303 		add_token_u8(&err, dev, OPAL_ENDNAME);
1304 
1305 		add_token_u8(&err, dev, OPAL_STARTNAME);
1306 		add_token_u8(&err, dev, OPAL_WRITELOCKENABLED);
1307 		add_token_u64(&err, dev, !!setup->WLE);
1308 		add_token_u8(&err, dev, OPAL_ENDNAME);
1309 
1310 		add_token_u8(&err, dev, OPAL_ENDLIST);
1311 		add_token_u8(&err, dev, OPAL_ENDNAME);
1312 	}
1313 	if (err) {
1314 		pr_debug("Error building Setup Locking range command.\n");
1315 		return err;
1316 
1317 	}
1318 
1319 	return finalize_and_send(dev, parse_and_check_status);
1320 }
1321 
1322 static int start_generic_opal_session(struct opal_dev *dev,
1323 				      enum opal_uid auth,
1324 				      enum opal_uid sp_type,
1325 				      const char *key,
1326 				      u8 key_len)
1327 {
1328 	u32 hsn;
1329 	int err;
1330 
1331 	if (key == NULL && auth != OPAL_ANYBODY_UID)
1332 		return OPAL_INVAL_PARAM;
1333 
1334 	hsn = GENERIC_HOST_SESSION_NUM;
1335 	err = cmd_start(dev, opaluid[OPAL_SMUID_UID],
1336 			opalmethod[OPAL_STARTSESSION]);
1337 
1338 	add_token_u64(&err, dev, hsn);
1339 	add_token_bytestring(&err, dev, opaluid[sp_type], OPAL_UID_LENGTH);
1340 	add_token_u8(&err, dev, 1);
1341 
1342 	switch (auth) {
1343 	case OPAL_ANYBODY_UID:
1344 		break;
1345 	case OPAL_ADMIN1_UID:
1346 	case OPAL_SID_UID:
1347 	case OPAL_PSID_UID:
1348 		add_token_u8(&err, dev, OPAL_STARTNAME);
1349 		add_token_u8(&err, dev, 0); /* HostChallenge */
1350 		add_token_bytestring(&err, dev, key, key_len);
1351 		add_token_u8(&err, dev, OPAL_ENDNAME);
1352 		add_token_u8(&err, dev, OPAL_STARTNAME);
1353 		add_token_u8(&err, dev, 3); /* HostSignAuth */
1354 		add_token_bytestring(&err, dev, opaluid[auth],
1355 				     OPAL_UID_LENGTH);
1356 		add_token_u8(&err, dev, OPAL_ENDNAME);
1357 		break;
1358 	default:
1359 		pr_debug("Cannot start Admin SP session with auth %d\n", auth);
1360 		return OPAL_INVAL_PARAM;
1361 	}
1362 
1363 	if (err) {
1364 		pr_debug("Error building start adminsp session command.\n");
1365 		return err;
1366 	}
1367 
1368 	return finalize_and_send(dev, start_opal_session_cont);
1369 }
1370 
1371 static int start_anybodyASP_opal_session(struct opal_dev *dev, void *data)
1372 {
1373 	return start_generic_opal_session(dev, OPAL_ANYBODY_UID,
1374 					  OPAL_ADMINSP_UID, NULL, 0);
1375 }
1376 
1377 static int start_SIDASP_opal_session(struct opal_dev *dev, void *data)
1378 {
1379 	int ret;
1380 	const u8 *key = dev->prev_data;
1381 
1382 	if (!key) {
1383 		const struct opal_key *okey = data;
1384 
1385 		ret = start_generic_opal_session(dev, OPAL_SID_UID,
1386 						 OPAL_ADMINSP_UID,
1387 						 okey->key,
1388 						 okey->key_len);
1389 	} else {
1390 		ret = start_generic_opal_session(dev, OPAL_SID_UID,
1391 						 OPAL_ADMINSP_UID,
1392 						 key, dev->prev_d_len);
1393 		kfree(key);
1394 		dev->prev_data = NULL;
1395 	}
1396 	return ret;
1397 }
1398 
1399 static int start_admin1LSP_opal_session(struct opal_dev *dev, void *data)
1400 {
1401 	struct opal_key *key = data;
1402 
1403 	return start_generic_opal_session(dev, OPAL_ADMIN1_UID,
1404 					  OPAL_LOCKINGSP_UID,
1405 					  key->key, key->key_len);
1406 }
1407 
1408 static int start_PSID_opal_session(struct opal_dev *dev, void *data)
1409 {
1410 	const struct opal_key *okey = data;
1411 
1412 	return start_generic_opal_session(dev, OPAL_PSID_UID,
1413 					  OPAL_ADMINSP_UID,
1414 					  okey->key,
1415 					  okey->key_len);
1416 }
1417 
1418 static int start_auth_opal_session(struct opal_dev *dev, void *data)
1419 {
1420 	struct opal_session_info *session = data;
1421 	u8 lk_ul_user[OPAL_UID_LENGTH];
1422 	size_t keylen = session->opal_key.key_len;
1423 	int err = 0;
1424 
1425 	u8 *key = session->opal_key.key;
1426 	u32 hsn = GENERIC_HOST_SESSION_NUM;
1427 
1428 	if (session->sum)
1429 		err = build_locking_user(lk_ul_user, sizeof(lk_ul_user),
1430 					 session->opal_key.lr);
1431 	else if (session->who != OPAL_ADMIN1 && !session->sum)
1432 		err = build_locking_user(lk_ul_user, sizeof(lk_ul_user),
1433 					 session->who - 1);
1434 	else
1435 		memcpy(lk_ul_user, opaluid[OPAL_ADMIN1_UID], OPAL_UID_LENGTH);
1436 
1437 	if (err)
1438 		return err;
1439 
1440 	err = cmd_start(dev, opaluid[OPAL_SMUID_UID],
1441 			opalmethod[OPAL_STARTSESSION]);
1442 
1443 	add_token_u64(&err, dev, hsn);
1444 	add_token_bytestring(&err, dev, opaluid[OPAL_LOCKINGSP_UID],
1445 			     OPAL_UID_LENGTH);
1446 	add_token_u8(&err, dev, 1);
1447 	add_token_u8(&err, dev, OPAL_STARTNAME);
1448 	add_token_u8(&err, dev, 0);
1449 	add_token_bytestring(&err, dev, key, keylen);
1450 	add_token_u8(&err, dev, OPAL_ENDNAME);
1451 	add_token_u8(&err, dev, OPAL_STARTNAME);
1452 	add_token_u8(&err, dev, 3);
1453 	add_token_bytestring(&err, dev, lk_ul_user, OPAL_UID_LENGTH);
1454 	add_token_u8(&err, dev, OPAL_ENDNAME);
1455 
1456 	if (err) {
1457 		pr_debug("Error building STARTSESSION command.\n");
1458 		return err;
1459 	}
1460 
1461 	return finalize_and_send(dev, start_opal_session_cont);
1462 }
1463 
1464 static int revert_tper(struct opal_dev *dev, void *data)
1465 {
1466 	int err;
1467 
1468 	err = cmd_start(dev, opaluid[OPAL_ADMINSP_UID],
1469 			opalmethod[OPAL_REVERT]);
1470 	if (err) {
1471 		pr_debug("Error building REVERT TPER command.\n");
1472 		return err;
1473 	}
1474 
1475 	return finalize_and_send(dev, parse_and_check_status);
1476 }
1477 
1478 static int internal_activate_user(struct opal_dev *dev, void *data)
1479 {
1480 	struct opal_session_info *session = data;
1481 	u8 uid[OPAL_UID_LENGTH];
1482 	int err;
1483 
1484 	memcpy(uid, opaluid[OPAL_USER1_UID], OPAL_UID_LENGTH);
1485 	uid[7] = session->who;
1486 
1487 	err = cmd_start(dev, uid, opalmethod[OPAL_SET]);
1488 	add_token_u8(&err, dev, OPAL_STARTNAME);
1489 	add_token_u8(&err, dev, OPAL_VALUES);
1490 	add_token_u8(&err, dev, OPAL_STARTLIST);
1491 	add_token_u8(&err, dev, OPAL_STARTNAME);
1492 	add_token_u8(&err, dev, 5); /* Enabled */
1493 	add_token_u8(&err, dev, OPAL_TRUE);
1494 	add_token_u8(&err, dev, OPAL_ENDNAME);
1495 	add_token_u8(&err, dev, OPAL_ENDLIST);
1496 	add_token_u8(&err, dev, OPAL_ENDNAME);
1497 
1498 	if (err) {
1499 		pr_debug("Error building Activate UserN command.\n");
1500 		return err;
1501 	}
1502 
1503 	return finalize_and_send(dev, parse_and_check_status);
1504 }
1505 
1506 static int erase_locking_range(struct opal_dev *dev, void *data)
1507 {
1508 	struct opal_session_info *session = data;
1509 	u8 uid[OPAL_UID_LENGTH];
1510 	int err;
1511 
1512 	if (build_locking_range(uid, sizeof(uid), session->opal_key.lr) < 0)
1513 		return -ERANGE;
1514 
1515 	err = cmd_start(dev, uid, opalmethod[OPAL_ERASE]);
1516 
1517 	if (err) {
1518 		pr_debug("Error building Erase Locking Range Command.\n");
1519 		return err;
1520 	}
1521 	return finalize_and_send(dev, parse_and_check_status);
1522 }
1523 
1524 static int set_mbr_done(struct opal_dev *dev, void *data)
1525 {
1526 	u8 *mbr_done_tf = data;
1527 	int err;
1528 
1529 	err = cmd_start(dev, opaluid[OPAL_MBRCONTROL],
1530 			opalmethod[OPAL_SET]);
1531 
1532 	add_token_u8(&err, dev, OPAL_STARTNAME);
1533 	add_token_u8(&err, dev, OPAL_VALUES);
1534 	add_token_u8(&err, dev, OPAL_STARTLIST);
1535 	add_token_u8(&err, dev, OPAL_STARTNAME);
1536 	add_token_u8(&err, dev, OPAL_MBRDONE);
1537 	add_token_u8(&err, dev, *mbr_done_tf); /* Done T or F */
1538 	add_token_u8(&err, dev, OPAL_ENDNAME);
1539 	add_token_u8(&err, dev, OPAL_ENDLIST);
1540 	add_token_u8(&err, dev, OPAL_ENDNAME);
1541 
1542 	if (err) {
1543 		pr_debug("Error Building set MBR Done command\n");
1544 		return err;
1545 	}
1546 
1547 	return finalize_and_send(dev, parse_and_check_status);
1548 }
1549 
1550 static int set_mbr_enable_disable(struct opal_dev *dev, void *data)
1551 {
1552 	u8 *mbr_en_dis = data;
1553 	int err;
1554 
1555 	err = cmd_start(dev, opaluid[OPAL_MBRCONTROL],
1556 			opalmethod[OPAL_SET]);
1557 
1558 	add_token_u8(&err, dev, OPAL_STARTNAME);
1559 	add_token_u8(&err, dev, OPAL_VALUES);
1560 	add_token_u8(&err, dev, OPAL_STARTLIST);
1561 	add_token_u8(&err, dev, OPAL_STARTNAME);
1562 	add_token_u8(&err, dev, OPAL_MBRENABLE);
1563 	add_token_u8(&err, dev, *mbr_en_dis);
1564 	add_token_u8(&err, dev, OPAL_ENDNAME);
1565 	add_token_u8(&err, dev, OPAL_ENDLIST);
1566 	add_token_u8(&err, dev, OPAL_ENDNAME);
1567 
1568 	if (err) {
1569 		pr_debug("Error Building set MBR done command\n");
1570 		return err;
1571 	}
1572 
1573 	return finalize_and_send(dev, parse_and_check_status);
1574 }
1575 
1576 static int write_shadow_mbr(struct opal_dev *dev, void *data)
1577 {
1578 	struct opal_shadow_mbr *shadow = data;
1579 	const u8 __user *src;
1580 	u8 *dst;
1581 	size_t off = 0;
1582 	u64 len;
1583 	int err = 0;
1584 
1585 	/* do we fit in the available shadow mbr space? */
1586 	err = generic_get_table_info(dev, OPAL_MBR, OPAL_TABLE_ROWS);
1587 	if (err) {
1588 		pr_debug("MBR: could not get shadow size\n");
1589 		return err;
1590 	}
1591 
1592 	len = response_get_u64(&dev->parsed, 4);
1593 	if (shadow->size > len || shadow->offset > len - shadow->size) {
1594 		pr_debug("MBR: does not fit in shadow (%llu vs. %llu)\n",
1595 			 shadow->offset + shadow->size, len);
1596 		return -ENOSPC;
1597 	}
1598 
1599 	/* do the actual transmission(s) */
1600 	src = (u8 __user *)(uintptr_t)shadow->data;
1601 	while (off < shadow->size) {
1602 		err = cmd_start(dev, opaluid[OPAL_MBR], opalmethod[OPAL_SET]);
1603 		add_token_u8(&err, dev, OPAL_STARTNAME);
1604 		add_token_u8(&err, dev, OPAL_WHERE);
1605 		add_token_u64(&err, dev, shadow->offset + off);
1606 		add_token_u8(&err, dev, OPAL_ENDNAME);
1607 
1608 		add_token_u8(&err, dev, OPAL_STARTNAME);
1609 		add_token_u8(&err, dev, OPAL_VALUES);
1610 
1611 		/*
1612 		 * The bytestring header is either 1 or 2 bytes, so assume 2.
1613 		 * There also needs to be enough space to accommodate the
1614 		 * trailing OPAL_ENDNAME (1 byte) and tokens added by
1615 		 * cmd_finalize.
1616 		 */
1617 		len = min(remaining_size(dev) - (2+1+CMD_FINALIZE_BYTES_NEEDED),
1618 			  (size_t)(shadow->size - off));
1619 		pr_debug("MBR: write bytes %zu+%llu/%llu\n",
1620 			 off, len, shadow->size);
1621 
1622 		dst = add_bytestring_header(&err, dev, len);
1623 		if (!dst)
1624 			break;
1625 		if (copy_from_user(dst, src + off, len))
1626 			err = -EFAULT;
1627 		dev->pos += len;
1628 
1629 		add_token_u8(&err, dev, OPAL_ENDNAME);
1630 		if (err)
1631 			break;
1632 
1633 		err = finalize_and_send(dev, parse_and_check_status);
1634 		if (err)
1635 			break;
1636 
1637 		off += len;
1638 	}
1639 	return err;
1640 }
1641 
1642 static int generic_pw_cmd(u8 *key, size_t key_len, u8 *cpin_uid,
1643 			  struct opal_dev *dev)
1644 {
1645 	int err;
1646 
1647 	err = cmd_start(dev, cpin_uid, opalmethod[OPAL_SET]);
1648 
1649 	add_token_u8(&err, dev, OPAL_STARTNAME);
1650 	add_token_u8(&err, dev, OPAL_VALUES);
1651 	add_token_u8(&err, dev, OPAL_STARTLIST);
1652 	add_token_u8(&err, dev, OPAL_STARTNAME);
1653 	add_token_u8(&err, dev, OPAL_PIN);
1654 	add_token_bytestring(&err, dev, key, key_len);
1655 	add_token_u8(&err, dev, OPAL_ENDNAME);
1656 	add_token_u8(&err, dev, OPAL_ENDLIST);
1657 	add_token_u8(&err, dev, OPAL_ENDNAME);
1658 
1659 	return err;
1660 }
1661 
1662 static int set_new_pw(struct opal_dev *dev, void *data)
1663 {
1664 	u8 cpin_uid[OPAL_UID_LENGTH];
1665 	struct opal_session_info *usr = data;
1666 
1667 	memcpy(cpin_uid, opaluid[OPAL_C_PIN_ADMIN1], OPAL_UID_LENGTH);
1668 
1669 	if (usr->who != OPAL_ADMIN1) {
1670 		cpin_uid[5] = 0x03;
1671 		if (usr->sum)
1672 			cpin_uid[7] = usr->opal_key.lr + 1;
1673 		else
1674 			cpin_uid[7] = usr->who;
1675 	}
1676 
1677 	if (generic_pw_cmd(usr->opal_key.key, usr->opal_key.key_len,
1678 			   cpin_uid, dev)) {
1679 		pr_debug("Error building set password command.\n");
1680 		return -ERANGE;
1681 	}
1682 
1683 	return finalize_and_send(dev, parse_and_check_status);
1684 }
1685 
1686 static int set_sid_cpin_pin(struct opal_dev *dev, void *data)
1687 {
1688 	u8 cpin_uid[OPAL_UID_LENGTH];
1689 	struct opal_key *key = data;
1690 
1691 	memcpy(cpin_uid, opaluid[OPAL_C_PIN_SID], OPAL_UID_LENGTH);
1692 
1693 	if (generic_pw_cmd(key->key, key->key_len, cpin_uid, dev)) {
1694 		pr_debug("Error building Set SID cpin\n");
1695 		return -ERANGE;
1696 	}
1697 	return finalize_and_send(dev, parse_and_check_status);
1698 }
1699 
1700 static int add_user_to_lr(struct opal_dev *dev, void *data)
1701 {
1702 	u8 lr_buffer[OPAL_UID_LENGTH];
1703 	u8 user_uid[OPAL_UID_LENGTH];
1704 	struct opal_lock_unlock *lkul = data;
1705 	int err;
1706 
1707 	memcpy(lr_buffer, opaluid[OPAL_LOCKINGRANGE_ACE_RDLOCKED],
1708 	       OPAL_UID_LENGTH);
1709 
1710 	if (lkul->l_state == OPAL_RW)
1711 		memcpy(lr_buffer, opaluid[OPAL_LOCKINGRANGE_ACE_WRLOCKED],
1712 		       OPAL_UID_LENGTH);
1713 
1714 	lr_buffer[7] = lkul->session.opal_key.lr;
1715 
1716 	memcpy(user_uid, opaluid[OPAL_USER1_UID], OPAL_UID_LENGTH);
1717 
1718 	user_uid[7] = lkul->session.who;
1719 
1720 	err = cmd_start(dev, lr_buffer, opalmethod[OPAL_SET]);
1721 
1722 	add_token_u8(&err, dev, OPAL_STARTNAME);
1723 	add_token_u8(&err, dev, OPAL_VALUES);
1724 
1725 	add_token_u8(&err, dev, OPAL_STARTLIST);
1726 	add_token_u8(&err, dev, OPAL_STARTNAME);
1727 	add_token_u8(&err, dev, 3);
1728 
1729 	add_token_u8(&err, dev, OPAL_STARTLIST);
1730 
1731 
1732 	add_token_u8(&err, dev, OPAL_STARTNAME);
1733 	add_token_bytestring(&err, dev,
1734 			     opaluid[OPAL_HALF_UID_AUTHORITY_OBJ_REF],
1735 			     OPAL_UID_LENGTH/2);
1736 	add_token_bytestring(&err, dev, user_uid, OPAL_UID_LENGTH);
1737 	add_token_u8(&err, dev, OPAL_ENDNAME);
1738 
1739 
1740 	add_token_u8(&err, dev, OPAL_STARTNAME);
1741 	add_token_bytestring(&err, dev,
1742 			     opaluid[OPAL_HALF_UID_AUTHORITY_OBJ_REF],
1743 			     OPAL_UID_LENGTH/2);
1744 	add_token_bytestring(&err, dev, user_uid, OPAL_UID_LENGTH);
1745 	add_token_u8(&err, dev, OPAL_ENDNAME);
1746 
1747 
1748 	add_token_u8(&err, dev, OPAL_STARTNAME);
1749 	add_token_bytestring(&err, dev, opaluid[OPAL_HALF_UID_BOOLEAN_ACE],
1750 			     OPAL_UID_LENGTH/2);
1751 	add_token_u8(&err, dev, 1);
1752 	add_token_u8(&err, dev, OPAL_ENDNAME);
1753 
1754 
1755 	add_token_u8(&err, dev, OPAL_ENDLIST);
1756 	add_token_u8(&err, dev, OPAL_ENDNAME);
1757 	add_token_u8(&err, dev, OPAL_ENDLIST);
1758 	add_token_u8(&err, dev, OPAL_ENDNAME);
1759 
1760 	if (err) {
1761 		pr_debug("Error building add user to locking range command.\n");
1762 		return err;
1763 	}
1764 
1765 	return finalize_and_send(dev, parse_and_check_status);
1766 }
1767 
1768 static int lock_unlock_locking_range(struct opal_dev *dev, void *data)
1769 {
1770 	u8 lr_buffer[OPAL_UID_LENGTH];
1771 	struct opal_lock_unlock *lkul = data;
1772 	u8 read_locked = 1, write_locked = 1;
1773 	int err = 0;
1774 
1775 	if (build_locking_range(lr_buffer, sizeof(lr_buffer),
1776 				lkul->session.opal_key.lr) < 0)
1777 		return -ERANGE;
1778 
1779 	switch (lkul->l_state) {
1780 	case OPAL_RO:
1781 		read_locked = 0;
1782 		write_locked = 1;
1783 		break;
1784 	case OPAL_RW:
1785 		read_locked = 0;
1786 		write_locked = 0;
1787 		break;
1788 	case OPAL_LK:
1789 		/* vars are initialized to locked */
1790 		break;
1791 	default:
1792 		pr_debug("Tried to set an invalid locking state... returning to uland\n");
1793 		return OPAL_INVAL_PARAM;
1794 	}
1795 
1796 	err = cmd_start(dev, lr_buffer, opalmethod[OPAL_SET]);
1797 
1798 	add_token_u8(&err, dev, OPAL_STARTNAME);
1799 	add_token_u8(&err, dev, OPAL_VALUES);
1800 	add_token_u8(&err, dev, OPAL_STARTLIST);
1801 
1802 	add_token_u8(&err, dev, OPAL_STARTNAME);
1803 	add_token_u8(&err, dev, OPAL_READLOCKED);
1804 	add_token_u8(&err, dev, read_locked);
1805 	add_token_u8(&err, dev, OPAL_ENDNAME);
1806 
1807 	add_token_u8(&err, dev, OPAL_STARTNAME);
1808 	add_token_u8(&err, dev, OPAL_WRITELOCKED);
1809 	add_token_u8(&err, dev, write_locked);
1810 	add_token_u8(&err, dev, OPAL_ENDNAME);
1811 
1812 	add_token_u8(&err, dev, OPAL_ENDLIST);
1813 	add_token_u8(&err, dev, OPAL_ENDNAME);
1814 
1815 	if (err) {
1816 		pr_debug("Error building SET command.\n");
1817 		return err;
1818 	}
1819 	return finalize_and_send(dev, parse_and_check_status);
1820 }
1821 
1822 
1823 static int lock_unlock_locking_range_sum(struct opal_dev *dev, void *data)
1824 {
1825 	u8 lr_buffer[OPAL_UID_LENGTH];
1826 	u8 read_locked = 1, write_locked = 1;
1827 	struct opal_lock_unlock *lkul = data;
1828 	int ret;
1829 
1830 	clear_opal_cmd(dev);
1831 	set_comid(dev, dev->comid);
1832 
1833 	if (build_locking_range(lr_buffer, sizeof(lr_buffer),
1834 				lkul->session.opal_key.lr) < 0)
1835 		return -ERANGE;
1836 
1837 	switch (lkul->l_state) {
1838 	case OPAL_RO:
1839 		read_locked = 0;
1840 		write_locked = 1;
1841 		break;
1842 	case OPAL_RW:
1843 		read_locked = 0;
1844 		write_locked = 0;
1845 		break;
1846 	case OPAL_LK:
1847 		/* vars are initialized to locked */
1848 		break;
1849 	default:
1850 		pr_debug("Tried to set an invalid locking state.\n");
1851 		return OPAL_INVAL_PARAM;
1852 	}
1853 	ret = generic_lr_enable_disable(dev, lr_buffer, 1, 1,
1854 					read_locked, write_locked);
1855 
1856 	if (ret < 0) {
1857 		pr_debug("Error building SET command.\n");
1858 		return ret;
1859 	}
1860 	return finalize_and_send(dev, parse_and_check_status);
1861 }
1862 
1863 static int activate_lsp(struct opal_dev *dev, void *data)
1864 {
1865 	struct opal_lr_act *opal_act = data;
1866 	u8 user_lr[OPAL_UID_LENGTH];
1867 	u8 uint_3 = 0x83;
1868 	int err, i;
1869 
1870 	err = cmd_start(dev, opaluid[OPAL_LOCKINGSP_UID],
1871 			opalmethod[OPAL_ACTIVATE]);
1872 
1873 	if (opal_act->sum) {
1874 		err = build_locking_range(user_lr, sizeof(user_lr),
1875 					  opal_act->lr[0]);
1876 		if (err)
1877 			return err;
1878 
1879 		add_token_u8(&err, dev, OPAL_STARTNAME);
1880 		add_token_u8(&err, dev, uint_3);
1881 		add_token_u8(&err, dev, 6);
1882 		add_token_u8(&err, dev, 0);
1883 		add_token_u8(&err, dev, 0);
1884 
1885 		add_token_u8(&err, dev, OPAL_STARTLIST);
1886 		add_token_bytestring(&err, dev, user_lr, OPAL_UID_LENGTH);
1887 		for (i = 1; i < opal_act->num_lrs; i++) {
1888 			user_lr[7] = opal_act->lr[i];
1889 			add_token_bytestring(&err, dev, user_lr, OPAL_UID_LENGTH);
1890 		}
1891 		add_token_u8(&err, dev, OPAL_ENDLIST);
1892 		add_token_u8(&err, dev, OPAL_ENDNAME);
1893 	}
1894 
1895 	if (err) {
1896 		pr_debug("Error building Activate LockingSP command.\n");
1897 		return err;
1898 	}
1899 
1900 	return finalize_and_send(dev, parse_and_check_status);
1901 }
1902 
1903 /* Determine if we're in the Manufactured Inactive or Active state */
1904 static int get_lsp_lifecycle(struct opal_dev *dev, void *data)
1905 {
1906 	u8 lc_status;
1907 	int err;
1908 
1909 	err = generic_get_column(dev, opaluid[OPAL_LOCKINGSP_UID],
1910 				 OPAL_LIFECYCLE);
1911 	if (err)
1912 		return err;
1913 
1914 	lc_status = response_get_u64(&dev->parsed, 4);
1915 	/* 0x08 is Manufactured Inactive */
1916 	/* 0x09 is Manufactured */
1917 	if (lc_status != OPAL_MANUFACTURED_INACTIVE) {
1918 		pr_debug("Couldn't determine the status of the Lifecycle state\n");
1919 		return -ENODEV;
1920 	}
1921 
1922 	return 0;
1923 }
1924 
1925 static int get_msid_cpin_pin(struct opal_dev *dev, void *data)
1926 {
1927 	const char *msid_pin;
1928 	size_t strlen;
1929 	int err;
1930 
1931 	err = generic_get_column(dev, opaluid[OPAL_C_PIN_MSID], OPAL_PIN);
1932 	if (err)
1933 		return err;
1934 
1935 	strlen = response_get_string(&dev->parsed, 4, &msid_pin);
1936 	if (!msid_pin) {
1937 		pr_debug("Couldn't extract MSID_CPIN from response\n");
1938 		return OPAL_INVAL_PARAM;
1939 	}
1940 
1941 	dev->prev_data = kmemdup(msid_pin, strlen, GFP_KERNEL);
1942 	if (!dev->prev_data)
1943 		return -ENOMEM;
1944 
1945 	dev->prev_d_len = strlen;
1946 
1947 	return 0;
1948 }
1949 
1950 static int end_opal_session(struct opal_dev *dev, void *data)
1951 {
1952 	int err = 0;
1953 
1954 	clear_opal_cmd(dev);
1955 	set_comid(dev, dev->comid);
1956 	add_token_u8(&err, dev, OPAL_ENDOFSESSION);
1957 
1958 	if (err < 0)
1959 		return err;
1960 	return finalize_and_send(dev, end_session_cont);
1961 }
1962 
1963 static int end_opal_session_error(struct opal_dev *dev)
1964 {
1965 	const struct opal_step error_end_session = {
1966 		end_opal_session,
1967 	};
1968 	return execute_step(dev, &error_end_session, 0);
1969 }
1970 
1971 static inline void setup_opal_dev(struct opal_dev *dev)
1972 {
1973 	dev->tsn = 0;
1974 	dev->hsn = 0;
1975 	dev->prev_data = NULL;
1976 }
1977 
1978 static int check_opal_support(struct opal_dev *dev)
1979 {
1980 	int ret;
1981 
1982 	mutex_lock(&dev->dev_lock);
1983 	setup_opal_dev(dev);
1984 	ret = opal_discovery0_step(dev);
1985 	dev->supported = !ret;
1986 	mutex_unlock(&dev->dev_lock);
1987 	return ret;
1988 }
1989 
1990 static void clean_opal_dev(struct opal_dev *dev)
1991 {
1992 
1993 	struct opal_suspend_data *suspend, *next;
1994 
1995 	mutex_lock(&dev->dev_lock);
1996 	list_for_each_entry_safe(suspend, next, &dev->unlk_lst, node) {
1997 		list_del(&suspend->node);
1998 		kfree(suspend);
1999 	}
2000 	mutex_unlock(&dev->dev_lock);
2001 }
2002 
2003 void free_opal_dev(struct opal_dev *dev)
2004 {
2005 	if (!dev)
2006 		return;
2007 	clean_opal_dev(dev);
2008 	kfree(dev);
2009 }
2010 EXPORT_SYMBOL(free_opal_dev);
2011 
2012 struct opal_dev *init_opal_dev(void *data, sec_send_recv *send_recv)
2013 {
2014 	struct opal_dev *dev;
2015 
2016 	dev = kmalloc(sizeof(*dev), GFP_KERNEL);
2017 	if (!dev)
2018 		return NULL;
2019 
2020 	INIT_LIST_HEAD(&dev->unlk_lst);
2021 	mutex_init(&dev->dev_lock);
2022 	dev->data = data;
2023 	dev->send_recv = send_recv;
2024 	if (check_opal_support(dev) != 0) {
2025 		pr_debug("Opal is not supported on this device\n");
2026 		kfree(dev);
2027 		return NULL;
2028 	}
2029 	return dev;
2030 }
2031 EXPORT_SYMBOL(init_opal_dev);
2032 
2033 static int opal_secure_erase_locking_range(struct opal_dev *dev,
2034 					   struct opal_session_info *opal_session)
2035 {
2036 	const struct opal_step erase_steps[] = {
2037 		{ start_auth_opal_session, opal_session },
2038 		{ get_active_key, &opal_session->opal_key.lr },
2039 		{ gen_key, },
2040 		{ end_opal_session, }
2041 	};
2042 	int ret;
2043 
2044 	mutex_lock(&dev->dev_lock);
2045 	setup_opal_dev(dev);
2046 	ret = execute_steps(dev, erase_steps, ARRAY_SIZE(erase_steps));
2047 	mutex_unlock(&dev->dev_lock);
2048 	return ret;
2049 }
2050 
2051 static int opal_erase_locking_range(struct opal_dev *dev,
2052 				    struct opal_session_info *opal_session)
2053 {
2054 	const struct opal_step erase_steps[] = {
2055 		{ start_auth_opal_session, opal_session },
2056 		{ erase_locking_range, opal_session },
2057 		{ end_opal_session, }
2058 	};
2059 	int ret;
2060 
2061 	mutex_lock(&dev->dev_lock);
2062 	setup_opal_dev(dev);
2063 	ret = execute_steps(dev, erase_steps, ARRAY_SIZE(erase_steps));
2064 	mutex_unlock(&dev->dev_lock);
2065 	return ret;
2066 }
2067 
2068 static int opal_enable_disable_shadow_mbr(struct opal_dev *dev,
2069 					  struct opal_mbr_data *opal_mbr)
2070 {
2071 	u8 enable_disable = opal_mbr->enable_disable == OPAL_MBR_ENABLE ?
2072 		OPAL_TRUE : OPAL_FALSE;
2073 
2074 	const struct opal_step mbr_steps[] = {
2075 		{ start_admin1LSP_opal_session, &opal_mbr->key },
2076 		{ set_mbr_done, &enable_disable },
2077 		{ end_opal_session, },
2078 		{ start_admin1LSP_opal_session, &opal_mbr->key },
2079 		{ set_mbr_enable_disable, &enable_disable },
2080 		{ end_opal_session, }
2081 	};
2082 	int ret;
2083 
2084 	if (opal_mbr->enable_disable != OPAL_MBR_ENABLE &&
2085 	    opal_mbr->enable_disable != OPAL_MBR_DISABLE)
2086 		return -EINVAL;
2087 
2088 	mutex_lock(&dev->dev_lock);
2089 	setup_opal_dev(dev);
2090 	ret = execute_steps(dev, mbr_steps, ARRAY_SIZE(mbr_steps));
2091 	mutex_unlock(&dev->dev_lock);
2092 	return ret;
2093 }
2094 
2095 static int opal_set_mbr_done(struct opal_dev *dev,
2096 			     struct opal_mbr_done *mbr_done)
2097 {
2098 	u8 mbr_done_tf = mbr_done->done_flag == OPAL_MBR_DONE ?
2099 		OPAL_TRUE : OPAL_FALSE;
2100 
2101 	const struct opal_step mbr_steps[] = {
2102 		{ start_admin1LSP_opal_session, &mbr_done->key },
2103 		{ set_mbr_done, &mbr_done_tf },
2104 		{ end_opal_session, }
2105 	};
2106 	int ret;
2107 
2108 	if (mbr_done->done_flag != OPAL_MBR_DONE &&
2109 	    mbr_done->done_flag != OPAL_MBR_NOT_DONE)
2110 		return -EINVAL;
2111 
2112 	mutex_lock(&dev->dev_lock);
2113 	setup_opal_dev(dev);
2114 	ret = execute_steps(dev, mbr_steps, ARRAY_SIZE(mbr_steps));
2115 	mutex_unlock(&dev->dev_lock);
2116 	return ret;
2117 }
2118 
2119 static int opal_write_shadow_mbr(struct opal_dev *dev,
2120 				 struct opal_shadow_mbr *info)
2121 {
2122 	const struct opal_step mbr_steps[] = {
2123 		{ start_admin1LSP_opal_session, &info->key },
2124 		{ write_shadow_mbr, info },
2125 		{ end_opal_session, }
2126 	};
2127 	int ret;
2128 
2129 	if (info->size == 0)
2130 		return 0;
2131 
2132 	mutex_lock(&dev->dev_lock);
2133 	setup_opal_dev(dev);
2134 	ret = execute_steps(dev, mbr_steps, ARRAY_SIZE(mbr_steps));
2135 	mutex_unlock(&dev->dev_lock);
2136 	return ret;
2137 }
2138 
2139 static int opal_save(struct opal_dev *dev, struct opal_lock_unlock *lk_unlk)
2140 {
2141 	struct opal_suspend_data *suspend;
2142 
2143 	suspend = kzalloc(sizeof(*suspend), GFP_KERNEL);
2144 	if (!suspend)
2145 		return -ENOMEM;
2146 
2147 	suspend->unlk = *lk_unlk;
2148 	suspend->lr = lk_unlk->session.opal_key.lr;
2149 
2150 	mutex_lock(&dev->dev_lock);
2151 	setup_opal_dev(dev);
2152 	add_suspend_info(dev, suspend);
2153 	mutex_unlock(&dev->dev_lock);
2154 	return 0;
2155 }
2156 
2157 static int opal_add_user_to_lr(struct opal_dev *dev,
2158 			       struct opal_lock_unlock *lk_unlk)
2159 {
2160 	const struct opal_step steps[] = {
2161 		{ start_admin1LSP_opal_session, &lk_unlk->session.opal_key },
2162 		{ add_user_to_lr, lk_unlk },
2163 		{ end_opal_session, }
2164 	};
2165 	int ret;
2166 
2167 	if (lk_unlk->l_state != OPAL_RO &&
2168 	    lk_unlk->l_state != OPAL_RW) {
2169 		pr_debug("Locking state was not RO or RW\n");
2170 		return -EINVAL;
2171 	}
2172 	if (lk_unlk->session.who < OPAL_USER1 ||
2173 	    lk_unlk->session.who > OPAL_USER9) {
2174 		pr_debug("Authority was not within the range of users: %d\n",
2175 			 lk_unlk->session.who);
2176 		return -EINVAL;
2177 	}
2178 	if (lk_unlk->session.sum) {
2179 		pr_debug("%s not supported in sum. Use setup locking range\n",
2180 			 __func__);
2181 		return -EINVAL;
2182 	}
2183 
2184 	mutex_lock(&dev->dev_lock);
2185 	setup_opal_dev(dev);
2186 	ret = execute_steps(dev, steps, ARRAY_SIZE(steps));
2187 	mutex_unlock(&dev->dev_lock);
2188 	return ret;
2189 }
2190 
2191 static int opal_reverttper(struct opal_dev *dev, struct opal_key *opal, bool psid)
2192 {
2193 	/* controller will terminate session */
2194 	const struct opal_step revert_steps[] = {
2195 		{ start_SIDASP_opal_session, opal },
2196 		{ revert_tper, }
2197 	};
2198 	const struct opal_step psid_revert_steps[] = {
2199 		{ start_PSID_opal_session, opal },
2200 		{ revert_tper, }
2201 	};
2202 
2203 	int ret;
2204 
2205 	mutex_lock(&dev->dev_lock);
2206 	setup_opal_dev(dev);
2207 	if (psid)
2208 		ret = execute_steps(dev, psid_revert_steps,
2209 				    ARRAY_SIZE(psid_revert_steps));
2210 	else
2211 		ret = execute_steps(dev, revert_steps,
2212 				    ARRAY_SIZE(revert_steps));
2213 	mutex_unlock(&dev->dev_lock);
2214 
2215 	/*
2216 	 * If we successfully reverted lets clean
2217 	 * any saved locking ranges.
2218 	 */
2219 	if (!ret)
2220 		clean_opal_dev(dev);
2221 
2222 	return ret;
2223 }
2224 
2225 static int __opal_lock_unlock(struct opal_dev *dev,
2226 			      struct opal_lock_unlock *lk_unlk)
2227 {
2228 	const struct opal_step unlock_steps[] = {
2229 		{ start_auth_opal_session, &lk_unlk->session },
2230 		{ lock_unlock_locking_range, lk_unlk },
2231 		{ end_opal_session, }
2232 	};
2233 	const struct opal_step unlock_sum_steps[] = {
2234 		{ start_auth_opal_session, &lk_unlk->session },
2235 		{ lock_unlock_locking_range_sum, lk_unlk },
2236 		{ end_opal_session, }
2237 	};
2238 
2239 	if (lk_unlk->session.sum)
2240 		return execute_steps(dev, unlock_sum_steps,
2241 				     ARRAY_SIZE(unlock_sum_steps));
2242 	else
2243 		return execute_steps(dev, unlock_steps,
2244 				     ARRAY_SIZE(unlock_steps));
2245 }
2246 
2247 static int __opal_set_mbr_done(struct opal_dev *dev, struct opal_key *key)
2248 {
2249 	u8 mbr_done_tf = OPAL_TRUE;
2250 	const struct opal_step mbrdone_step[] = {
2251 		{ start_admin1LSP_opal_session, key },
2252 		{ set_mbr_done, &mbr_done_tf },
2253 		{ end_opal_session, }
2254 	};
2255 
2256 	return execute_steps(dev, mbrdone_step, ARRAY_SIZE(mbrdone_step));
2257 }
2258 
2259 static int opal_lock_unlock(struct opal_dev *dev,
2260 			    struct opal_lock_unlock *lk_unlk)
2261 {
2262 	int ret;
2263 
2264 	if (lk_unlk->session.who > OPAL_USER9)
2265 		return -EINVAL;
2266 
2267 	mutex_lock(&dev->dev_lock);
2268 	ret = __opal_lock_unlock(dev, lk_unlk);
2269 	mutex_unlock(&dev->dev_lock);
2270 	return ret;
2271 }
2272 
2273 static int opal_take_ownership(struct opal_dev *dev, struct opal_key *opal)
2274 {
2275 	const struct opal_step owner_steps[] = {
2276 		{ start_anybodyASP_opal_session, },
2277 		{ get_msid_cpin_pin, },
2278 		{ end_opal_session, },
2279 		{ start_SIDASP_opal_session, opal },
2280 		{ set_sid_cpin_pin, opal },
2281 		{ end_opal_session, }
2282 	};
2283 	int ret;
2284 
2285 	if (!dev)
2286 		return -ENODEV;
2287 
2288 	mutex_lock(&dev->dev_lock);
2289 	setup_opal_dev(dev);
2290 	ret = execute_steps(dev, owner_steps, ARRAY_SIZE(owner_steps));
2291 	mutex_unlock(&dev->dev_lock);
2292 	return ret;
2293 }
2294 
2295 static int opal_activate_lsp(struct opal_dev *dev,
2296 			     struct opal_lr_act *opal_lr_act)
2297 {
2298 	const struct opal_step active_steps[] = {
2299 		{ start_SIDASP_opal_session, &opal_lr_act->key },
2300 		{ get_lsp_lifecycle, },
2301 		{ activate_lsp, opal_lr_act },
2302 		{ end_opal_session, }
2303 	};
2304 	int ret;
2305 
2306 	if (!opal_lr_act->num_lrs || opal_lr_act->num_lrs > OPAL_MAX_LRS)
2307 		return -EINVAL;
2308 
2309 	mutex_lock(&dev->dev_lock);
2310 	setup_opal_dev(dev);
2311 	ret = execute_steps(dev, active_steps, ARRAY_SIZE(active_steps));
2312 	mutex_unlock(&dev->dev_lock);
2313 	return ret;
2314 }
2315 
2316 static int opal_setup_locking_range(struct opal_dev *dev,
2317 				    struct opal_user_lr_setup *opal_lrs)
2318 {
2319 	const struct opal_step lr_steps[] = {
2320 		{ start_auth_opal_session, &opal_lrs->session },
2321 		{ setup_locking_range, opal_lrs },
2322 		{ end_opal_session, }
2323 	};
2324 	int ret;
2325 
2326 	mutex_lock(&dev->dev_lock);
2327 	setup_opal_dev(dev);
2328 	ret = execute_steps(dev, lr_steps, ARRAY_SIZE(lr_steps));
2329 	mutex_unlock(&dev->dev_lock);
2330 	return ret;
2331 }
2332 
2333 static int opal_set_new_pw(struct opal_dev *dev, struct opal_new_pw *opal_pw)
2334 {
2335 	const struct opal_step pw_steps[] = {
2336 		{ start_auth_opal_session, &opal_pw->session },
2337 		{ set_new_pw, &opal_pw->new_user_pw },
2338 		{ end_opal_session, }
2339 	};
2340 	int ret;
2341 
2342 	if (opal_pw->session.who > OPAL_USER9  ||
2343 	    opal_pw->new_user_pw.who > OPAL_USER9)
2344 		return -EINVAL;
2345 
2346 	mutex_lock(&dev->dev_lock);
2347 	setup_opal_dev(dev);
2348 	ret = execute_steps(dev, pw_steps, ARRAY_SIZE(pw_steps));
2349 	mutex_unlock(&dev->dev_lock);
2350 	return ret;
2351 }
2352 
2353 static int opal_activate_user(struct opal_dev *dev,
2354 			      struct opal_session_info *opal_session)
2355 {
2356 	const struct opal_step act_steps[] = {
2357 		{ start_admin1LSP_opal_session, &opal_session->opal_key },
2358 		{ internal_activate_user, opal_session },
2359 		{ end_opal_session, }
2360 	};
2361 	int ret;
2362 
2363 	/* We can't activate Admin1 it's active as manufactured */
2364 	if (opal_session->who < OPAL_USER1 ||
2365 	    opal_session->who > OPAL_USER9) {
2366 		pr_debug("Who was not a valid user: %d\n", opal_session->who);
2367 		return -EINVAL;
2368 	}
2369 
2370 	mutex_lock(&dev->dev_lock);
2371 	setup_opal_dev(dev);
2372 	ret = execute_steps(dev, act_steps, ARRAY_SIZE(act_steps));
2373 	mutex_unlock(&dev->dev_lock);
2374 	return ret;
2375 }
2376 
2377 bool opal_unlock_from_suspend(struct opal_dev *dev)
2378 {
2379 	struct opal_suspend_data *suspend;
2380 	bool was_failure = false;
2381 	int ret = 0;
2382 
2383 	if (!dev)
2384 		return false;
2385 	if (!dev->supported)
2386 		return false;
2387 
2388 	mutex_lock(&dev->dev_lock);
2389 	setup_opal_dev(dev);
2390 
2391 	list_for_each_entry(suspend, &dev->unlk_lst, node) {
2392 		dev->tsn = 0;
2393 		dev->hsn = 0;
2394 
2395 		ret = __opal_lock_unlock(dev, &suspend->unlk);
2396 		if (ret) {
2397 			pr_debug("Failed to unlock LR %hhu with sum %d\n",
2398 				 suspend->unlk.session.opal_key.lr,
2399 				 suspend->unlk.session.sum);
2400 			was_failure = true;
2401 		}
2402 		if (dev->mbr_enabled) {
2403 			ret = __opal_set_mbr_done(dev, &suspend->unlk.session.opal_key);
2404 			if (ret)
2405 				pr_debug("Failed to set MBR Done in S3 resume\n");
2406 		}
2407 	}
2408 	mutex_unlock(&dev->dev_lock);
2409 	return was_failure;
2410 }
2411 EXPORT_SYMBOL(opal_unlock_from_suspend);
2412 
2413 int sed_ioctl(struct opal_dev *dev, unsigned int cmd, void __user *arg)
2414 {
2415 	void *p;
2416 	int ret = -ENOTTY;
2417 
2418 	if (!capable(CAP_SYS_ADMIN))
2419 		return -EACCES;
2420 	if (!dev)
2421 		return -ENOTSUPP;
2422 	if (!dev->supported)
2423 		return -ENOTSUPP;
2424 
2425 	p = memdup_user(arg, _IOC_SIZE(cmd));
2426 	if (IS_ERR(p))
2427 		return PTR_ERR(p);
2428 
2429 	switch (cmd) {
2430 	case IOC_OPAL_SAVE:
2431 		ret = opal_save(dev, p);
2432 		break;
2433 	case IOC_OPAL_LOCK_UNLOCK:
2434 		ret = opal_lock_unlock(dev, p);
2435 		break;
2436 	case IOC_OPAL_TAKE_OWNERSHIP:
2437 		ret = opal_take_ownership(dev, p);
2438 		break;
2439 	case IOC_OPAL_ACTIVATE_LSP:
2440 		ret = opal_activate_lsp(dev, p);
2441 		break;
2442 	case IOC_OPAL_SET_PW:
2443 		ret = opal_set_new_pw(dev, p);
2444 		break;
2445 	case IOC_OPAL_ACTIVATE_USR:
2446 		ret = opal_activate_user(dev, p);
2447 		break;
2448 	case IOC_OPAL_REVERT_TPR:
2449 		ret = opal_reverttper(dev, p, false);
2450 		break;
2451 	case IOC_OPAL_LR_SETUP:
2452 		ret = opal_setup_locking_range(dev, p);
2453 		break;
2454 	case IOC_OPAL_ADD_USR_TO_LR:
2455 		ret = opal_add_user_to_lr(dev, p);
2456 		break;
2457 	case IOC_OPAL_ENABLE_DISABLE_MBR:
2458 		ret = opal_enable_disable_shadow_mbr(dev, p);
2459 		break;
2460 	case IOC_OPAL_MBR_DONE:
2461 		ret = opal_set_mbr_done(dev, p);
2462 		break;
2463 	case IOC_OPAL_WRITE_SHADOW_MBR:
2464 		ret = opal_write_shadow_mbr(dev, p);
2465 		break;
2466 	case IOC_OPAL_ERASE_LR:
2467 		ret = opal_erase_locking_range(dev, p);
2468 		break;
2469 	case IOC_OPAL_SECURE_ERASE_LR:
2470 		ret = opal_secure_erase_locking_range(dev, p);
2471 		break;
2472 	case IOC_OPAL_PSID_REVERT_TPR:
2473 		ret = opal_reverttper(dev, p, true);
2474 		break;
2475 	default:
2476 		break;
2477 	}
2478 
2479 	kfree(p);
2480 	return ret;
2481 }
2482 EXPORT_SYMBOL_GPL(sed_ioctl);
2483