X-Git-Url: http://git.joshuawise.com/snipe.git/blobdiff_plain/12aa4087bee3e70f170d7457794921de4e385227..1144856ba9d6018d9922c6ede7e97779a0fe6373:/codegen/codegen.sml

diff --git a/codegen/codegen.sml b/codegen/codegen.sml
index 0297b9f..6ee8c2f 100644
--- a/codegen/codegen.sml
+++ b/codegen/codegen.sml
@@ -1,10 +1,7 @@
-(* L1 Compiler
- * Assembly Code Generator for FAKE assembly
- * Author: Alex Vaynberg <alv@andrew.cmu.edu>
- * Based on code by: Kaustuv Chaudhuri <kaustuv+@cs.cmu.edu>
- * Modified: Frank Pfenning <fp@cs.cmu.edu>
- *
- * Implements a "convenient munch" algorithm
+(* L3 Compiler
+ * Assembly code generator for fake x86 assembly
+ * Author: Joshua Wise <jwise@andrew.cmu.edu>
+ * Author: Chris Lu <czl@andrew.cmu.edu>
  *)
 
 signature CODEGEN =
@@ -17,40 +14,449 @@ struct
   structure T = Tree
   structure X = x86
 
+  (* effect : T.exp -> bool
+   * true iff the given expression has an effect.
+   *)
+  fun effect (T.BINOP(T.DIV, _, _)) = true
+    | effect (T.BINOP(T.MOD, _, _)) = true
+    | effect (T.CALL _) = true
+    | effect (T.BINOP(_, a, b)) = (effect a) orelse (effect b)
+    | effect (T.UNOP (_, a)) = effect a
+    | effect (T.MEMORY m) = true
+    | effect (T.ALLOC(_)) = true
+    | effect _ = false
+
+  (* hasfixed : T.exp -> bool
+   * true iff the given expression has an hasfixed. Somewhat like effect, hmm?
+   *)
+  fun hasfixed (T.BINOP(T.DIV, _, _)) = true
+    | hasfixed (T.BINOP(T.MOD, _, _)) = true
+    | hasfixed (T.BINOP(T.LSH, _, _)) = true
+    | hasfixed (T.BINOP(T.RSH, _, _)) = true
+    | hasfixed (T.CALL _) = true
+    | hasfixed (T.BINOP(_, a, b)) = (hasfixed a) orelse (hasfixed b)
+    | hasfixed (T.UNOP (_, a)) = hasfixed a
+    | hasfixed (T.ALLOC(_)) = true
+    | hasfixed (T.MEMORY m) = hasfixed m
+    | hasfixed _ = false
+
   (* munch_exp : prex86oper -> T.exp -> prex86insn list *)
   (* munch_exp d e
    * generates instructions to achieve d <- e
    * d must be TEMP(t) or REG(r)
    *)
-  fun munch_exp d (T.CONST(n)) = [X.MOVL(d, X.CONST n)]
-    | munch_exp d (T.TEMP(t)) = [X.MOVL(d, X.TEMP t)]
-    | munch_exp d (T.BINOP(T.ADD, e1, T.CONST n)) = (munch_exp d e1) @ [X.ADDL(d, X.CONST n)]
-    | munch_exp d (T.BINOP(T.ADD, T.CONST n, e1)) = (munch_exp d e1) @ [X.ADDL(d, X.CONST n)]
-    | munch_exp d (T.BINOP(T.ADD, e1, e2)) = let val t1 = Temp.new () in (munch_exp d e1) @ (munch_exp (X.TEMP t1) e2) @ [X.ADDL(d, X.TEMP t1)] end
-    | munch_exp d (T.BINOP(T.SUB, T.CONST 0w0, e1)) = (munch_exp d e1) @ [X.NEG d]
-    | munch_exp d (T.BINOP(T.SUB, e1, T.CONST(n))) = (munch_exp d e1) @ [X.SUBL(d, X.CONST n)]
-    | munch_exp d (T.BINOP(T.SUB, e1, e2)) = let val t1 = Temp.new () in (munch_exp d e1) @ (munch_exp (X.TEMP t1) e2) @ [X.SUBL(d, X.TEMP t1)] end
+  and munch_exp d (T.CONST(n)) = [X.MOV(d, X.CONST n)]
+    | munch_exp d (T.TEMP(t)) = [X.MOV(d, X.TEMP t)]
+    | munch_exp d (T.ARG(0, sz)) = [X.MOV(d, X.OSIZE (X.sts sz, X.REG X.EDI))]
+    | munch_exp d (T.ARG(1, sz)) = [X.MOV(d, X.OSIZE (X.sts sz, X.REG X.ESI))]
+    | munch_exp d (T.ARG(2, sz)) = [X.MOV(d, X.OSIZE (X.sts sz, X.REG X.EDX))]
+    | munch_exp d (T.ARG(3, sz)) = [X.MOV(d, X.OSIZE (X.sts sz, X.REG X.ECX))]
+    | munch_exp d (T.ARG(4, sz)) = [X.MOV(d, X.OSIZE (X.sts sz, X.REG X.R8D))]
+    | munch_exp d (T.ARG(5, sz)) = [X.MOV(d, X.OSIZE (X.sts sz, X.REG X.R9D))]
+    | munch_exp d (T.ARG(t, sz)) = [X.MOV(d, X.OSIZE (X.sts sz, X.STACKARG (t - 6)))]
+    | munch_exp d (T.CALL(name, l, rsz)) =	(* Scary demons live here. *)
+        let
+          val nargs = length l
+          val nstack = if (nargs <= 6)
+                       then 0
+                       else nargs - 6
+          val stackb = nstack * 8
+          fun argdest 1 = X.REG X.EDI
+            | argdest 2 = X.REG X.ESI
+            | argdest 3 = X.REG X.EDX
+            | argdest 4 = X.REG X.ECX
+            | argdest 5 = X.REG X.R8D
+            | argdest 6 = X.REG X.R9D
+            | argdest n = X.REL (X.REG X.RSP, X.CONST (Word32.fromInt (~(stackb - 8 * (n - 7)))) )
+
+          val dests = List.tabulate (nargs, fn x => argdest (x+1))
+          val (exps,_) = ListPair.unzip l
+          val hf = List.map hasfixed exps
+          val (d_hf, l_hf) = ListPair.unzip (ListPair.foldr
+            (fn (a,b,c) => if b then a::c else c)
+            nil
+            (ListPair.zip (dests,l), hf)
+          )
+          val (d_nohf, l_nohf) = ListPair.unzip (ListPair.foldr
+            (fn (a,b,c) => if b then c else a::c)
+            nil
+            (ListPair.zip (dests,l), hf)
+          )
+          val temps = List.map (fn (_, sz) => Temp.new ("arg") sz (* xxx? *)) l_hf
+          val argevals_hf = List.map
+            (fn (t,(exp,_)) => munch_exp (X.TEMP t) exp)
+            (ListPair.zip (temps, l_hf))
+          val argpushes = List.map
+            (fn (dest, t) => [(X.MOV (X.OSIZE(X.sts (Temp.size t), dest), X.TEMP t))])
+            (ListPair.zip (d_hf, temps))
+          val argevals_nohf = List.map
+            (fn (d,(exp,sz)) => munch_exp (X.OSIZE (X.sts sz, d)) exp)
+            (ListPair.zip (d_nohf, l_nohf))
+        in
+          List.concat argevals_hf @ 
+          List.concat argpushes @
+          List.concat argevals_nohf @
+          [ X.SUB (X.OSIZE (X.Qword, X.REG X.RSP), X.CONST (Word32.fromInt stackb)),
+            X.CALL (name, nargs),
+            X.ADD (X.OSIZE (X.Qword, X.REG X.RSP), X.CONST (Word32.fromInt stackb)),
+            X.MOV (d, X.OSIZE (X.sts rsz, X.REG X.EAX)) ]	(* Finally! *)
+        end
+(*    | munch_exp d (T.BINOP(T.ADD, e1, T.CONST 0w0)) = munch_exp d e1
+    | munch_exp d (T.BINOP(T.ADD, T.CONST 0w0, e1)) = munch_exp d e1
+    | munch_exp d (T.BINOP(T.ADD, e1, T.CONST n)) = (munch_exp d e1) @ [X.ADD(d, X.CONST n)]
+    | munch_exp d (T.BINOP(T.ADD, T.CONST n, e1)) = (munch_exp d e1) @ [X.ADD(d, X.CONST n)]
+    | munch_exp d (T.BINOP(T.ADD, e1, T.TEMP t)) = (munch_exp d e1) @ [X.ADD(d, X.TEMP t)]
+    | munch_exp d (T.BINOP(T.ADD, T.TEMP t, e2)) = (munch_exp d e2) @ [X.ADD(d, X.TEMP t)] *)
+    | munch_exp d (T.BINOP(T.ADD, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("add") 4)
+        in
+          (munch_exp d e1) @ (munch_exp t1 e2) @ [X.ADD(d, t1)]
+        end
+(*    | munch_exp d (T.BINOP(T.SUB, T.CONST 0w0, e1)) = (munch_exp d e1) @ [X.NEG d]
+    | munch_exp d (T.BINOP(T.SUB, e1, T.CONST 0w0)) = munch_exp d e1
+    | munch_exp d (T.BINOP(T.SUB, e1, T.CONST(n))) = (munch_exp d e1) @ [X.SUB(d, X.CONST n)]
+    | munch_exp d (T.BINOP(T.SUB, e1, T.TEMP t)) = (munch_exp d e1) @ [X.SUB(d, X.TEMP t)] *)
+    | munch_exp d (T.BINOP(T.SUB, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("sub") 4)
+        in
+          (munch_exp d e1) @ (munch_exp t1 e2) @ [X.SUB(d, t1)]
+        end
     | munch_exp d (T.BINOP(T.MUL, T.TEMP t, T.CONST n)) = [X.IMUL3(d, X.TEMP t, n)]
     | munch_exp d (T.BINOP(T.MUL, T.CONST n, T.TEMP t)) = [X.IMUL3(d, X.TEMP t, n)]
+(*
+    | munch_exp d (T.BINOP(T.MUL, e1, T.CONST 0w1)) = munch_exp d e1
+    | munch_exp d (T.BINOP(T.MUL, T.CONST 0w1, e1)) = munch_exp d e1
     | munch_exp d (T.BINOP(T.MUL, e1, T.CONST n)) = (munch_exp d e1) @ [X.IMUL(d, X.CONST n)]
-    | munch_exp d (T.BINOP(T.MUL, T.CONST n, e1)) = (munch_exp d e1) @ [X.IMUL(d, X.CONST n)]
-    | munch_exp d (T.BINOP(T.MUL, e1, e2)) = let val t1 = Temp.new () in (munch_exp d e1) @ (munch_exp (X.TEMP t1) e2) @ [X.IMUL(d, X.TEMP t1)] end
-    | munch_exp d (T.BINOP(T.DIV, e1, e2)) = let val t1 = Temp.new () in (munch_exp (X.TEMP t1) e1) @ (munch_exp d e2) @ [X.MOVL (X.REG X.EAX, X.TEMP t1), X.CLTD, X.IDIVL d, X.MOVL (d, X.REG X.EAX)] end
-    | munch_exp d (T.BINOP(T.MOD, e1, e2)) = let val t1 = Temp.new () in (munch_exp (X.TEMP t1) e1) @ (munch_exp d e2) @ [X.MOVL (X.REG X.EAX, X.TEMP t1), X.CLTD, X.IDIVL d, X.MOVL (d, X.REG X.EDX)] end
+    | munch_exp d (T.BINOP(T.MUL, T.CONST n, e1)) = (munch_exp d e1) @ [X.IMUL(d, X.CONST n)] *)
+    | munch_exp d (T.BINOP(T.MUL, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("mul") 4)
+        in
+          (munch_exp d e1) @ (munch_exp t1 e2) @ [X.IMUL(d, t1)]
+        end
+    | munch_exp d (T.BINOP(T.DIV, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("div") 4)
+        in
+          (munch_exp t1 e1) @ (munch_exp d e2) @
+          [X.MOV (X.REG X.EAX, t1), X.CLTD, X.IDIV d, X.MOV (d, X.REG X.EAX)]
+        end
+    | munch_exp d (T.BINOP(T.MOD, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("mod") 4)
+        in
+          (munch_exp t1 e1) @ (munch_exp d e2) @
+          [X.MOV (X.REG X.EAX, t1), X.CLTD, X.IDIV d, X.MOV (d, X.REG X.EDX)]
+        end
+    | munch_exp d (T.BINOP(T.LSH, e1, T.CONST n)) = (munch_exp d e1) @ [X.SAL (d, X.CONST (n mod 0w32))]
+    | munch_exp d (T.BINOP(T.LSH, e1, T.TEMP t)) = (munch_exp d e1) @ [X.MOV (X.REG X.ECX, X.TEMP t), X.SAL (d, X.REG X.ECX)]
+    | munch_exp d (T.BINOP(T.LSH, e1, e2)) =
+        let
+          val t = X.TEMP (Temp.new ("lsh") 4)
+        in
+          (munch_exp d e1) @ (munch_exp t e2) @ [X.MOV (X.REG X.ECX, t), X.SAL (d, X.REG X.ECX)]
+        end
+    | munch_exp d (T.BINOP(T.RSH, e1, T.CONST n)) = (munch_exp d e1) @ [X.SAR (d, X.CONST (n mod 0w32))]
+    | munch_exp d (T.BINOP(T.RSH, e1, T.TEMP t)) = (munch_exp d e1) @ [X.MOV (X.REG X.ECX, X.TEMP t), X.SAR (d, X.REG X.ECX)]
+    | munch_exp d (T.BINOP(T.RSH, e1, e2)) =
+        let
+          val t = X.TEMP (Temp.new ("rsh") 4)
+        in
+          (munch_exp d e1) @ (munch_exp t e2) @ [X.MOV (X.REG X.ECX, t), X.SAR (d, X.REG X.ECX)]
+        end
+    | munch_exp d (T.BINOP(T.BITAND, T.CONST n, e1)) = (munch_exp d e1) @ [X.AND (d, X.CONST n)]
+    | munch_exp d (T.BINOP(T.BITAND, e1, T.CONST n)) = (munch_exp d e1) @ [X.AND (d, X.CONST n)] 
+    | munch_exp d (T.BINOP(T.BITAND, T.TEMP t, e1)) = (munch_exp d e1) @ [X.AND (d, X.TEMP t)]
+    | munch_exp d (T.BINOP(T.BITAND, e1, T.TEMP t)) = (munch_exp d e1) @ [X.AND (d, X.TEMP t)] 
+    | munch_exp d (T.BINOP(T.BITAND, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("bitand") 4)
+        in
+          (munch_exp d e1) @ (munch_exp t1 e2) @ [X.AND(d, t1)]
+        end
+    | munch_exp d (T.BINOP(T.BITOR, T.CONST n, e1)) = (munch_exp d e1) @ [X.OR (d, X.CONST n)]
+    | munch_exp d (T.BINOP(T.BITOR, e1, T.CONST n)) = (munch_exp d e1) @ [X.OR (d, X.CONST n)] 
+    | munch_exp d (T.BINOP(T.BITOR, T.TEMP t, e1)) = (munch_exp d e1) @ [X.OR (d, X.TEMP t)]
+    | munch_exp d (T.BINOP(T.BITOR, e1, T.TEMP t)) = (munch_exp d e1) @ [X.OR (d, X.TEMP t)] 
+    | munch_exp d (T.BINOP(T.BITOR, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("bitor") 4)
+        in
+          (munch_exp d e1) @ (munch_exp t1 e2) @ [X.OR(d, t1)]
+        end
+    | munch_exp d (T.BINOP(T.BITXOR, T.CONST n, e1)) = (munch_exp d e1) @ [X.XOR (d, X.CONST n)]
+    | munch_exp d (T.BINOP(T.BITXOR, e1, T.CONST n)) = (munch_exp d e1) @ [X.XOR (d, X.CONST n)]
+    | munch_exp d (T.BINOP(T.BITXOR, T.TEMP t, e1)) = (munch_exp d e1) @ [X.XOR (d, X.TEMP t)]
+    | munch_exp d (T.BINOP(T.BITXOR, e1, T.TEMP t)) = (munch_exp d e1) @ [X.XOR (d, X.TEMP t)]
+    | munch_exp d (T.BINOP(T.BITXOR, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("bitxor") 4)
+        in
+          (munch_exp d e1) @ (munch_exp t1 e2) @ [X.XOR(d, t1)]
+        end
+    | munch_exp d (a as T.BINOP(T.LOGAND, e1, e2)) =
+        let
+          val (insn1, pos1, neg1) = munch_cond e1
+          val (insn2, pos2, neg2) = munch_cond e2
+          val t1 = X.TEMP (Temp.new("logand 1") 4)
+          val t2 = X.TEMP (Temp.new("logand 2") 4)
+          val l = Label.new ()
+        in
+          if (effect e2 orelse (length insn2 > 10))
+          then (insn1) @
+               [X.SETcc(pos1, t1), X.Jcc (neg1, l)] @
+               (insn2) @
+               [X.SETcc(pos2, t1), X.LABEL l, X.MOVZB(d, t1)]
+          else insn1 @ [X.SETcc (pos1, t1)] @ insn2 @ [X.SETcc (pos2, t2), X.AND(X.OSIZE (X.Byte, t1), X.OSIZE (X.Byte, t2)), X.MOVZB(d, t1)]
+        end
+    | munch_exp d (a as T.BINOP(T.LOGOR, e1, e2)) =
+        let
+          val (insn1, pos1, neg1) = munch_cond e1
+          val (insn2, pos2, neg2) = munch_cond e2
+          val t1 = X.TEMP (Temp.new("logor 1") 4)
+          val t2 = X.TEMP (Temp.new("logor 2") 4)
+          val l = Label.new ()
+        in
+          if (effect e2 orelse (length insn2 > 10))
+          then (insn1) @
+               [X.SETcc(pos1, t1), X.Jcc (pos1, l)] @
+               (insn2) @
+               [X.SETcc(pos2, t1), X.LABEL l, X.MOVZB(d, t1)]
+          else insn1 @ [X.SETcc (pos1, t1)] @ insn2 @ [X.SETcc (pos2, t2), X.OR(X.OSIZE (X.Byte, t1), X.OSIZE (X.Byte, t2)), X.MOVZB(d, t1)]
+        end
+    | munch_exp d (a as T.BINOP(T.EQ, _, _)) =
+        let val (insns, pos, neg) = munch_cond a in insns @ [X.SETcc (pos, d), X.MOVZB(d, d)] end
+    | munch_exp d (a as T.BINOP(T.NEQ, _, _)) =
+        let val (insns, pos, neg) = munch_cond a in insns @ [X.SETcc (pos, d), X.MOVZB(d, d)] end
+    | munch_exp d (a as T.BINOP(T.LE, _, _)) =
+        let val (insns, pos, neg) = munch_cond a in insns @ [X.SETcc (pos, d), X.MOVZB(d, d)] end
+    | munch_exp d (a as T.BINOP(T.LT, _, _)) =
+        let val (insns, pos, neg) = munch_cond a in insns @ [X.SETcc (pos, d), X.MOVZB(d, d)] end
+    | munch_exp d (a as T.BINOP(T.GE, _, _)) =
+        let val (insns, pos, neg) = munch_cond a in insns @ [X.SETcc (pos, d), X.MOVZB(d, d)] end
+    | munch_exp d (a as T.BINOP(T.GT, _, _)) =
+        let val (insns, pos, neg) = munch_cond a in insns @ [X.SETcc (pos, d), X.MOVZB(d, d)] end
+    | munch_exp d (T.UNOP(T.NEG, T.CONST n)) = [X.MOV (d, X.CONST (~n))]
+    | munch_exp d (T.UNOP(T.NEG, e1)) = (munch_exp d e1) @ [X.NEG d]
+    | munch_exp d (T.UNOP(T.BITNOT, T.CONST n)) = [X.MOV (d, X.CONST (Word32.notb n))]
+    | munch_exp d (T.UNOP(T.BITNOT, e1)) = (munch_exp d e1) @ [X.NOT d]
+    | munch_exp d (T.UNOP(T.BANG, T.CONST n)) = if (n = 0w0) then [X.MOV (d, X.CONST 0w1)] else [X.MOV (d, X.CONST 0w0)]
+    | munch_exp d (T.UNOP(T.BANG, e)) = 
+        let
+          val (insns, pos, neg) = munch_cond e
+        in
+          insns @ [X.SETcc (neg, d), X.MOVZB(d, d)]
+        end
+    | munch_exp d (T.MEMORY e1) =
+        let
+          val a = X.TEMP (Temp.new "addr" 8)
+        in
+          munch_exp a e1 @ [X.MOV (d, X.REL (a, X.CONST 0w0))]
+        end
+    | munch_exp d (T.ALLOC(exp)) = (munch_exp d (T.CALL (Symbol.symbol "calloc", [(exp, 4), (T.CONST 0w1, 4)], 8)))
+                                   @ [X.MOV (X.REL (d, X.CONST 0w0), X.CONST 0w0)]
+
+  (* munch_cond : T.exp -> X.insn list * X.cond * X.cond
+   * munch_cond stm generates code to set flags, and then returns a conditional
+   * to test if the expression was true and for if it was false.
+   *)
+  and munch_cond (T.UNOP (T.BANG, e)) =
+        let
+          val (insns, pos, neg) = munch_cond e
+        in
+          (insns, neg, pos)
+        end
+    | munch_cond (T.BINOP(T.NEQ, T.TEMP t, T.CONST n)) = ([X.CMP(X.TEMP t, X.CONST n)], X.NE, X.E)
+    | munch_cond (T.BINOP(T.NEQ, T.CONST n, T.TEMP t)) = ([X.CMP(X.TEMP t, X.CONST n)], X.NE, X.E)
+    | munch_cond (T.BINOP(T.NEQ, T.CONST n, e1)) =
+        let val t = X.TEMP (Temp.new ("const neq") 4) in (munch_exp t e1 @ [X.CMP(t, X.CONST n)], X.NE, X.E) end
+    | munch_cond (T.BINOP(T.NEQ, e1, T.CONST n)) =
+        let val t = X.TEMP (Temp.new ("const neq") 4) in (munch_exp t e1 @ [X.CMP(t, X.CONST n)], X.NE, X.E) end
+    | munch_cond (T.BINOP(T.NEQ, T.TEMP t, e1)) =
+        let val t1 = X.TEMP (Temp.new ("const neq") 4) in (munch_exp t1 e1 @ [X.CMP(t1, X.TEMP t)], X.NE, X.E) end
+    | munch_cond (T.BINOP(T.NEQ, e1, T.TEMP t)) =
+        let val t1 = X.TEMP (Temp.new ("const neq") 4) in (munch_exp t1 e1 @ [X.CMP(t1, X.TEMP t)], X.NE, X.E) end
+    | munch_cond (T.BINOP(T.NEQ, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("var neq 1") 4)
+          val t2 = X.TEMP (Temp.new ("var neq 2") 4)
+        in
+          (munch_exp t1 e1 @ munch_exp t2 e2 @
+           [X.CMP(t1, t2)], X.NE, X.E)
+        end
+    | munch_cond (T.BINOP(T.EQ, T.TEMP t, T.CONST n)) = ([X.CMP(X.TEMP t, X.CONST n)], X.E, X.NE)
+    | munch_cond (T.BINOP(T.EQ, T.CONST n, T.TEMP t)) = ([X.CMP(X.TEMP t, X.CONST n)], X.E, X.NE)
+    | munch_cond (T.BINOP(T.EQ, T.CONST n, e1)) =
+        let val t = X.TEMP (Temp.new ("const eq") 4) in (munch_exp t e1 @ [X.CMP(t, X.CONST n)], X.E, X.NE) end
+    | munch_cond (T.BINOP(T.EQ, e1, T.CONST n)) =
+        let val t = X.TEMP (Temp.new ("const eq") 4) in (munch_exp t e1 @ [X.CMP(t, X.CONST n)], X.E, X.NE) end
+    | munch_cond (T.BINOP(T.EQ, T.TEMP t, e1)) =
+        let val t1 = X.TEMP (Temp.new ("const eq") 4) in (munch_exp t1 e1 @ [X.CMP(t1, X.TEMP t)], X.E, X.NE) end
+    | munch_cond (T.BINOP(T.EQ, e1, T.TEMP t)) =
+        let val t1 = X.TEMP (Temp.new ("const eq") 4) in (munch_exp t1 e1 @ [X.CMP(t1, X.TEMP t)], X.E, X.NE) end
+    | munch_cond (T.BINOP(T.EQ, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("var eq 1") 4)
+          val t2 = X.TEMP (Temp.new ("var eq 2") 4)
+        in
+          (munch_exp t1 e1 @ munch_exp t2 e2 @
+           [X.CMP(t1, t2)], X.E, X.NE)
+        end
+    | munch_cond (T.BINOP(T.LE, T.TEMP t, T.CONST n)) = ([X.CMP(X.TEMP t, X.CONST n)], X.LE, X.G)
+    | munch_cond (T.BINOP(T.LE, T.CONST n, T.TEMP t)) = ([X.CMP(X.TEMP t, X.CONST n)], X.GE, X.L)
+    | munch_cond (T.BINOP(T.LE, T.CONST n, e1)) =
+        let val t = X.TEMP (Temp.new ("const le") 4) in (munch_exp t e1 @ [X.CMP(t, X.CONST n)], X.GE, X.L) end
+    | munch_cond (T.BINOP(T.LE, e1, T.CONST n)) =
+        let val t = X.TEMP (Temp.new ("const le") 4) in (munch_exp t e1 @ [X.CMP(t, X.CONST n)], X.LE, X.G) end
+    | munch_cond (T.BINOP(T.LE, T.TEMP t, e1)) =
+        let val t1 = X.TEMP (Temp.new ("const le") 4) in (munch_exp t1 e1 @ [X.CMP(t1, X.TEMP t)], X.GE, X.L) end
+    | munch_cond (T.BINOP(T.LE, e1, T.TEMP t)) =
+        let val t1 = X.TEMP (Temp.new ("const le") 4) in (munch_exp t1 e1 @ [X.CMP(t1, X.TEMP t)], X.LE, X.G) end
+    | munch_cond (T.BINOP(T.LE, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("var le 1") 4)
+          val t2 = X.TEMP (Temp.new ("var le 2") 4)
+        in
+          (munch_exp t1 e1 @ munch_exp t2 e2 @
+           [X.CMP(t1, t2)], X.LE, X.G)
+        end
+    | munch_cond (T.BINOP(T.LT, T.TEMP t, T.CONST n)) = ([X.CMP(X.TEMP t, X.CONST n)], X.L, X.GE)
+    | munch_cond (T.BINOP(T.LT, T.CONST n, T.TEMP t)) = ([X.CMP(X.TEMP t, X.CONST n)], X.G, X.LE)
+    | munch_cond (T.BINOP(T.LT, T.CONST n, e1)) =
+        let val t = X.TEMP (Temp.new ("const lt") 4) in (munch_exp t e1 @ [X.CMP(t, X.CONST n)], X.G, X.LE) end
+    | munch_cond (T.BINOP(T.LT, e1, T.CONST n)) =
+        let val t = X.TEMP (Temp.new ("const lt") 4) in (munch_exp t e1 @ [X.CMP(t, X.CONST n)], X.L, X.GE) end
+    | munch_cond (T.BINOP(T.LT, T.TEMP t, e1)) =
+        let val t1 = X.TEMP (Temp.new ("const lt") 4) in (munch_exp t1 e1 @ [X.CMP(t1, X.TEMP t)], X.G, X.LE) end
+    | munch_cond (T.BINOP(T.LT, e1, T.TEMP t)) =
+        let val t1 = X.TEMP (Temp.new ("const lt") 4) in (munch_exp t1 e1 @ [X.CMP(t1, X.TEMP t)], X.L, X.GE) end
+    | munch_cond (T.BINOP(T.LT, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("var lt 1") 4)
+          val t2 = X.TEMP (Temp.new ("var lt 2") 4)
+        in
+          (munch_exp t1 e1 @ munch_exp t2 e2 @
+           [X.CMP(t1, t2)], X.L, X.GE)
+        end
+    | munch_cond (T.BINOP(T.GT, T.TEMP t, T.CONST n)) = ([X.CMP(X.TEMP t, X.CONST n)], X.G, X.LE)
+    | munch_cond (T.BINOP(T.GT, T.CONST n, T.TEMP t)) = ([X.CMP(X.TEMP t, X.CONST n)], X.L, X.GE)
+    | munch_cond (T.BINOP(T.GT, e1, T.CONST n)) =
+        let val t = X.TEMP (Temp.new ("const gt") 4) in (munch_exp t e1 @ [X.CMP(t, X.CONST n)], X.G, X.LE) end
+    | munch_cond (T.BINOP(T.GT, T.CONST n, e1)) =
+        let val t = X.TEMP (Temp.new ("const gt") 4) in (munch_exp t e1 @ [X.CMP(t, X.CONST n)], X.L, X.GE) end
+    | munch_cond (T.BINOP(T.GT, e1, T.TEMP t)) =
+        let val t1 = X.TEMP (Temp.new ("const gt") 4) in (munch_exp t1 e1 @ [X.CMP(t1, X.TEMP t)], X.G, X.LE) end
+    | munch_cond (T.BINOP(T.GT, T.TEMP t, e1)) =
+        let val t1 = X.TEMP (Temp.new ("const gt") 4) in (munch_exp t1 e1 @ [X.CMP(t1, X.TEMP t)], X.L, X.GE) end
+    | munch_cond (T.BINOP(T.GT, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("var gt 1") 4)
+          val t2 = X.TEMP (Temp.new ("var gt 2") 4)
+        in
+          (munch_exp t1 e1 @ munch_exp t2 e2 @
+           [X.CMP(t1, t2)], X.G, X.LE)
+        end
+    | munch_cond (T.BINOP(T.GE, T.TEMP t, T.CONST n)) = ([X.CMP(X.TEMP t, X.CONST n)], X.GE, X.L)
+    | munch_cond (T.BINOP(T.GE, T.CONST n, T.TEMP t)) = ([X.CMP(X.TEMP t, X.CONST n)], X.LE, X.G)
+    | munch_cond (T.BINOP(T.GE, e1, T.CONST n)) =
+        let val t = X.TEMP (Temp.new ("const ge") 4) in (munch_exp t e1 @ [X.CMP(t, X.CONST n)], X.GE, X.L) end
+    | munch_cond (T.BINOP(T.GE, T.CONST n, e1)) =
+        let val t = X.TEMP (Temp.new ("const ge") 4) in (munch_exp t e1 @ [X.CMP(t, X.CONST n)], X.LE, X.G) end
+    | munch_cond (T.BINOP(T.GE, e1, T.TEMP t)) =
+        let val t1 = X.TEMP (Temp.new ("const ge") 4) in (munch_exp t1 e1 @ [X.CMP(t1, X.TEMP t)], X.GE, X.L) end
+    | munch_cond (T.BINOP(T.GE, T.TEMP t, e1)) =
+        let val t1 = X.TEMP (Temp.new ("const ge") 4) in (munch_exp t1 e1 @ [X.CMP(t1, X.TEMP t)], X.LE, X.G) end
+    | munch_cond (T.BINOP(T.GE, e1, e2)) =
+        let
+          val t1 = X.TEMP (Temp.new ("var ge 1") 4)
+          val t2 = X.TEMP (Temp.new ("var ge 2") 4)
+        in
+          (munch_exp t1 e1 @ munch_exp t2 e2 @
+           [X.CMP(t1, t2)], X.GE, X.L)
+        end
+    | munch_cond (T.BINOP(T.LOGOR, e1, e2)) =
+        let
+          val (insn1, pos1, neg1) = munch_cond e1
+          val (insn2, pos2, neg2) = munch_cond e2
+          val t1 = X.TEMP (Temp.new("logor c 1") 4)
+          val t2 = X.TEMP (Temp.new("logor c 2") 4)
+          val l = Label.new ()
+        in
+          if (effect e2 orelse (length insn2 > 10))
+          then ((insn1) @
+                [X.SETcc (pos1, t1), X.Jcc (pos1, l)] @
+                (insn2) @
+                [X.SETcc (pos2, t1), X.LABEL l, X.TEST(X.OSIZE (X.Byte, t1), X.OSIZE (X.Byte, t1))],
+                X.NE, X.E)
+          else (insn1 @ [X.SETcc (pos1, t1)] @ insn2 @ [X.SETcc (pos2, t2), X.OR(X.OSIZE (X.Byte, t1), X.OSIZE (X.Byte, t2))], X.NE, X.E)
+        end
+    | munch_cond (T.BINOP(T.LOGAND, e1, e2)) =
+        let
+          val (insn1, pos1, neg1) = munch_cond e1
+          val (insn2, pos2, neg2) = munch_cond e2
+          val t1 = X.TEMP (Temp.new("logand c 1") 4)
+          val t2 = X.TEMP (Temp.new("logand c 2") 4)
+          val l = Label.new ()
+        in
+          if (effect e2 orelse (length insn2 > 10))
+          then ((insn1) @
+                [X.SETcc (pos1, t1), X.Jcc (neg1, l)] @
+                (insn2) @
+                [X.SETcc (pos2, t1), X.LABEL l, X.TEST(X.OSIZE (X.Byte, t1), X.OSIZE (X.Byte, t1))],
+                X.NE, X.E)
+          else (insn1 @ [X.SETcc (pos1, t1)] @ insn2 @ [X.SETcc (pos2, t2), X.AND(X.OSIZE (X.Byte, t1), X.OSIZE (X.Byte, t2))], X.NE, X.E)
+        end
+    | munch_cond e =
+      let
+        val t = X.TEMP (Temp.new ("munch c") 4)
+      in
+        (munch_exp t e @ [ X.TEST (t,t) ], X.NE, X.E)
+      end
 
-  (* munch_stm : T.stm -> AS.instr list *)
+  (* munch_lval : T.exp -> (X.insn list * X.operand)
+   * Takes an expression that has been typechecked as being a valid lvalue, and then returns an instruction list and an operand to store your shit in.
+   *)
+  fun munch_lval (T.TEMP t) = ([], X.TEMP t)
+    | munch_lval (T.MEMORY m) = 
+      let
+        val t = Temp.new "lv addr" 8
+      in
+        (munch_exp (X.TEMP t) m, X.REL (X.TEMP t, X.CONST 0w0))
+      end
+    | munch_lval _ = raise ErrorMsg.InternalError "That wasn't really a valid lvalue..."
+
+  (* munch_stm : T.stm -> X.insn list *)
   (* munch_stm stm generates code to execute stm *)
-  fun munch_stm (T.MOVE(T.TEMP(t1), e2)) =
-        munch_exp (X.TEMP t1) e2
-    | munch_stm (T.MOVE(_, _)) =
-        raise ErrorMsg.InternalError "Incorrect first operand for T.MOVE?"
-    | munch_stm (T.RETURN(e)) =
+  fun munch_stm (T.MOVE (T.TEMP t, a as T.TEMP _, _)) = munch_exp (X.TEMP t) a
+    | munch_stm (T.MOVE (T.TEMP t, a as T.CONST _, _)) = munch_exp (X.TEMP t) a
+    | munch_stm (T.MOVE (T.TEMP t, a as T.ARG (an, sz), _)) = munch_exp (X.TEMP t) a
+    | munch_stm (T.MOVE (T.TEMP t, a as T.CALL _, _)) = munch_exp (X.TEMP t) a
+    | munch_stm (T.MOVE (a, e2, sz)) =
+        let
+          val t = Temp.new ("assign") sz
+          val (m, r) = munch_lval a
+        in
+          m @ munch_exp (X.TEMP t) e2
+          @ [X.MOV(X.OSIZE (X.sts sz, r), X.TEMP t)]
+        end
+    | munch_stm (T.RETURN(e, sz)) =
         let
-          val t = Temp.new ()
+          val t = Temp.new ("retval") sz
         in
           munch_exp (X.TEMP t) e
-          @ [X.MOVL(X.REG X.EAX, X.TEMP t), X.RET]
+          @ [X.MOV(X.OSIZE (X.sts sz, X.REG X.EAX), X.TEMP t), X.RET]
         end
+    | munch_stm (T.LABEL(l)) = [X.LABEL l]
+    | munch_stm (T.JUMP(l)) = [X.JMP l]
+    | munch_stm (T.JUMPIFN(e, l)) =
+       let
+         val (insns, pos, neg) = munch_cond e 
+       in
+         insns @ [X.Jcc (neg, l)]
+       end
+    | munch_stm (T.EFFECT(exp, sz)) = let val t = X.TEMP (Temp.new "throwaway" sz) in munch_exp t exp end
 
   fun codegen nil = nil
     | codegen (stm::stms) = munch_stm stm @ codegen stms