ARC: disasm: fix kprobe get/set functions

Layout of pt_regs structure differs between ARCompact/ARCv2/ARCv3.
Fix set_reg/get_reg functions to handle proper registers.

Signed-off-by: Sergey Matyukevich <sergey.matyukevich@synopsys.com>

Author: Sergey Matyukevich

arch/arc/kernel/disasm.c

@@ -434,18 +434,52 @@ long __kprobes get_reg(int reg, struct pt_regs *regs,
 {
 	long *p;
 
+#if defined(CONFIG_ISA_ARCOMPACT)
 	if (reg <= 12) {
 		p = &regs->r0;
 		return p[-reg];
 	}
 
 	if (cregs && (reg <= 25)) {
 		p = &cregs->r13;
-		return p[13-reg];
+		return p[13 - reg];
 	}
 
 	if (reg == 26)
 		return regs->gp;
+
+#elif defined(CONFIG_ISA_ARCV2)
+	if (reg <= 11) {
+		p = &regs->r0;
+		return p[reg];
+	}
+
+	if (reg == 12)
+		return regs->r12;
+
+	if (cregs && (reg <= 25)) {
+		p = &cregs->r13;
+		return p[13 - reg];
+	}
+
+	if (reg == 26)
+		return regs->gp;
+
+#else /* CONFIG_ISA_ARCV3 */
+	if (reg <= 13) {
+		p = &regs->r0;
+		return p[reg];
+	}
+
+	if (cregs && (reg <= 26)) {
+		p = &cregs->r14;
+		return p[reg - 14];
+	}
+
+	if (reg == 30)
+		return regs->gp;
+#endif
+
 	if (reg == 27)
 		return regs->fp;
 	if (reg == 28)
@@ -461,6 +495,7 @@ void __kprobes set_reg(int reg, long val, struct pt_regs *regs,
 {
 	long *p;
 
+#if defined(CONFIG_ISA_ARCOMPACT)
 	switch (reg) {
 	case 0 ... 12:
 		p = &regs->r0;
@@ -469,7 +504,37 @@ void __kprobes set_reg(int reg, long val, struct pt_regs *regs,
 	case 13 ... 25:
 		if (cregs) {
 			p = &cregs->r13;
-			p[13-reg] = val;
+			p[13 - reg] = val;
+		}
+		break;
+	case 26:
+		regs->gp = val;
+		break;
+	case 27:
+		regs->fp = val;
+		break;
+	case 28:
+		regs->sp = val;
+		break;
+	case 31:
+		regs->blink = val;
+		break;
+	default:
+		break;
+	}
+#elif defined(CONFIG_ISA_ARCV2)
+	switch (reg) {
+	case 0 ... 11:
+		p = &regs->r0;
+		p[reg] = val;
+		break;
+	case 12:
+		regs->r12 = val;
+		break;
+	case 13 ... 25:
+		if (cregs) {
+			p = &cregs->r13;
+			p[13 - reg] = val;
 		}
 		break;
 	case 26:
@@ -487,6 +552,35 @@ void __kprobes set_reg(int reg, long val, struct pt_regs *regs,
 	default:
 		break;
 	}
+#else /* CONFIG_ISA_ARCV3 */
+	switch (reg) {
+	case 0 ... 13:
+		p = &regs->r0;
+		p[reg] = val;
+		break;
+	case 14 ... 26:
+		if (cregs) {
+			p = &cregs->r14;
+			p[reg - 14] = val;
+		}
+		break;
+	case 27:
+		regs->fp = val;
+		break;
+	case 28:
+		regs->sp = val;
+		break;
+	case 30:
+		regs->gp = val;
+		break;
+	case 31:
+		regs->blink = val;
+		break;
+	default:
+		break;
+	}
+
+#endif
 }
 
 /*