From 8fb9cd4e23d294d4e03f4d36da2bec21e47997f3 Mon Sep 17 00:00:00 2001
From: Ayke van Laethem <aykevanlaethem@gmail.com>
Date: Wed, 22 Aug 2018 04:35:25 +0200
Subject: [PATCH] Implement shifting with a differently-sized integer

---
 compiler.go | 26 +++++++++++++++++++-------
 1 file changed, 19 insertions(+), 7 deletions(-)

diff --git a/compiler.go b/compiler.go
index 5300fc09..2f9f3165 100644
--- a/compiler.go
+++ b/compiler.go
@@ -1530,13 +1530,25 @@ func (c *Compiler) parseBinOp(frame *Frame, binop *ssa.BinOp) (llvm.Value, error
 		return c.builder.CreateOr(x, y, ""), nil
 	case token.XOR: // ^
 		return c.builder.CreateXor(x, y, ""), nil
-	case token.SHL: // <<
-		return c.builder.CreateShl(x, y, ""), nil
-	case token.SHR: // >>
-		if signed {
-			return c.builder.CreateAShr(x, y, ""), nil
-		} else {
-			return c.builder.CreateLShr(x, y, ""), nil
+	case token.SHL, token.SHR:
+		sizeX := c.targetData.TypeAllocSize(x.Type())
+		sizeY := c.targetData.TypeAllocSize(y.Type())
+		if sizeX > sizeY {
+			// x and y must have equal sizes, make Y bigger in this case.
+			// y is unsigned, this has been checked by the Go type checker.
+			y = c.builder.CreateZExt(y, x.Type(), "")
+		}
+		switch binop.Op {
+		case token.SHL: // <<
+			return c.builder.CreateShl(x, y, ""), nil
+		case token.SHR: // >>
+			if signed {
+				return c.builder.CreateAShr(x, y, ""), nil
+			} else {
+				return c.builder.CreateLShr(x, y, ""), nil
+			}
+		default:
+			panic("unreachable")
 		}
 	case token.AND_NOT: // &^
 		// Go specific. Calculate "and not" with x & (~y)