NIR: asm statement support, Wip

compiler/nir/ast2ir.nim

@@ -10,7 +10,7 @@
 import std / [assertions, tables, sets]
 import ".." / [ast, astalgo, types, options, lineinfos, msgs, magicsys,
   modulegraphs, renderer, transf, bitsets, trees, nimsets,
-  expanddefaults]
+  expanddefaults, wordrecg]
 from ".." / lowerings import lowerSwap, lowerTupleUnpacking
 from ".." / pathutils import customPath
 import .. / ic / bitabs
@@ -2454,6 +2454,7 @@ proc genProc(cOuter: var ProcCon; prc: PSym) =
         c.code.addStrVal c.lit.strings, info, irModule(c, ast.originatingModule(prc))
         c.code.addImmediateVal info, prc.itemId.item.int
     addCallConv c, info, prc.typ.callConv
+    if sfPure in prc.flags: c.code.addPragmaId info, AsmNoStackFrame
     if sfCompilerProc in prc.flags:
       build c.code, info, PragmaPair:
         c.code.addPragmaId info, CoreName
@@ -2463,7 +2464,7 @@ proc genProc(cOuter: var ProcCon; prc: PSym) =
         c.code.addPragmaId info, ExternName
         c.code.addStrVal c.lit.strings, info, prc.loc.r
       if sfImportc in prc.flags:
-        if lfHeader in prc. loc.flags:
+        if lfHeader in prc.loc.flags:
           assert(prc. annex != nil)
           let str = getStr(prc. annex.path)
           build c.code, info, PragmaPair:
@@ -2517,6 +2518,53 @@ proc genComplexCall(c: var ProcCon; n: PNode; d: var Value) =
   else:
     genCall c, n, d
 
+template genEmitCode(c: var ProcCon; n: PNode; isAsmStmt: bool) =
+  let offset =
+    if isAsmStmt: 1 # first son is pragmas
+    else: 0
+
+  build c.code, info, EmitCode:
+    for i in offset..<n.len:
+      let it = n[i]
+      case it.kind:
+        of nkStrLit..nkTripleStrLit:
+          c.code.addVerbatim c.lit.strings, info, it.strVal
+        of nkSym:
+          c.code.addSymUse info, toSymId(c, it.sym)
+        else:
+          gen(c, it)
+
+import target_props
+proc genAsm(c: var ProcCon; n: PNode) =
+  let info = toLineInfo(c, n.info)
+
+  var inlineAsmSyntax = ""
+  if (let p = n[0]; p).kind == nkPragma:
+    for i in p:
+      if whichPragma(i) == wInlineAsmSyntax:
+        inlineAsmSyntax = i[1].strVal
+
+  build c.code, info, Emit:
+    c.code.addEmitTarget info, Asm
+
+    c.code.addBoolInfo info, IsGlobal, c.prc == nil
+    c.code.addBoolInfo info, InPure, (
+      if c.prc != nil:
+        sfPure in c.prc.flags
+      else: false
+    )
+
+    if inlineAsmSyntax != "":
+      c.code.addEnumInfo info, InlineAsmSyntax, (
+        case inlineAsmSyntax:
+          of "gcc": GCCExtendedAsm
+          of "vcc": VisualCPP
+          else: raiseAssert "never"
+      )
+
+    genEmitCode c, n, true
+
+
 proc gen(c: var ProcCon; n: PNode; d: var Value; flags: GenFlags = {}) =
   when defined(nimCompilerStacktraceHints):
     setFrameMsg c.config$n.info & " " & $n.kind & " " & $flags
@@ -2560,6 +2608,7 @@ proc gen(c: var ProcCon; n: PNode; d: var Value; flags: GenFlags = {}) =
     unused(c, n, d)
     genWhile(c, n)
   of nkBlockExpr, nkBlockStmt: genBlock(c, n, d)
+  of nkAsmStmt: genAsm(c, n)
   of nkReturnStmt: genReturn(c, n)
   of nkRaiseStmt: genRaise(c, n)
   of nkBreakStmt: genBreak(c, n)

compiler/nir/cir.nim

@@ -14,6 +14,7 @@ from std / strutils import toOctal
 import .. / ic / [bitabs, rodfiles]
 import nirtypes, nirinsts, nirfiles
 import ../../dist/checksums/src/checksums/md5
+import target_props
 
 type
   Token = LitId # indexing into the tokens BiTable[string]
@@ -31,6 +32,7 @@ type
     Semicolon = ";"
     Comma = ", "
     Space = " "
+    Quote = $'"'
     Colon = ": "
     Dot = "."
     Arrow = "->"
@@ -63,6 +65,7 @@ proc fillTokenTable(tab: var BiTable[string]) =
 type
   GeneratedCode* = object
     m: NirModule
+    props: TargetProps
     includes: seq[LitId]
     includedHeaders: IntSet
     data: seq[LitId]
@@ -75,8 +78,11 @@ type
     generatedTypes: IntSet
     mangledModules: Table[LitId, LitId]
 
-proc initGeneratedCode*(m: sink NirModule): GeneratedCode =
-  result = GeneratedCode(m: m, code: @[], tokens: initBiTable[string]())
+proc initGeneratedCode*(m: sink NirModule; props: sink TargetProps): GeneratedCode =
+  result = GeneratedCode(
+    m: m, code: @[], tokens: initBiTable[string](), 
+    props: props
+  )
   fillTokenTable(result.tokens)
 
 proc add*(g: var GeneratedCode; t: PredefinedToken) {.inline.} =
@@ -444,7 +450,9 @@ proc genProcDecl(c: var GeneratedCode; t: Tree; n: NodePos; isExtern: bool) =
         let lit = t[y].litId
         raiseAssert "cannot eval: " & c.m.lit.strings[lit]
       else: discard
-    of PragmaId: discard
+    of PragmaId:
+      let key = cast[PragmaKey](t[prc].rawOperand)
+      if key == AsmNoStackFrame: c.add "NIM_NAKED "
     else: break
     next t, prc
 
@@ -554,6 +562,54 @@ template moveToDataSection(body: untyped) =
     c.data.add c.code[i]
   setLen c.code, oldLen
 
+proc genGccAsm(c: var GeneratedCode; t: Tree; n: NodePos) =
+  c.add "__asm__ "
+  c.add ParLe
+  c.add NewLine
+
+  var asmTemplate = true
+  var left = 0
+  var s = ""
+
+  template maybeAddQuote =
+    if asmTemplate: c.add Quote
+
+  template findSecUpdate(s: string): bool =
+    var result = false
+    for i in s:
+      if i == ':': result = true
+      if i notin {' ', '\t', '\n'}: break #found char
+    result
+
+  maybeAddQuote
+  for ch in sons(t, n):
+    if t[ch].kind == Verbatim:
+      s = c.m.lit.strings[t[ch].litId]
+      left = 0
+      for i in 0..s.high:
+        if s[i] == '\n':
+          c.add s[left..i - 1]
+          if asmTemplate: c.add r"\n"
+          maybeAddQuote
+          left = i + 1
+          c.add NewLine
+          if not findSecUpdate(s[i+1..^1]): maybeAddQuote # next '"'
+        elif s[i] == ':': asmTemplate = false
+    else:
+      c.add s[left..^1]
+      c.gen(t, ch)
+  if not findSecUpdate(s):
+    c.add s[left..^1]
+    maybeAddQuote
+  c.add ParRi
+  c.add Semicolon
+
+proc genVisualCPPAsm(c: var GeneratedCode; t: Tree; n: NodePos) =
+  c.add "__asm "
+  c.add CurlyLe
+  c.gen(t, n) # inline asm
+  c.add CurlyRi
+
 proc gen(c: var GeneratedCode; t: Tree; n: NodePos) =
   case t[n].kind
   of Nop:
@@ -807,11 +863,65 @@ proc gen(c: var GeneratedCode; t: Tree; n: NodePos) =
   of NumberConv: genNumberConv c, t, n
   of CheckedObjConv: binaryop ""
   of ObjConv: binaryop ""
-  of Emit: raiseAssert "cannot interpret: Emit"
   of ProcDecl: genProcDecl c, t, n, false
   of ForeignProcDecl: genProcDecl c, t, n, true
   of PragmaPair, PragmaId, TestOf, Yld, SetExc, TestExc:
     c.add "cannot interpret: " & $t[n].kind
+  of Verbatim:
+    c.add c.m.lit.strings[t[n].litId]
+  of Emit:
+    requireInfo t, n, IsGlobal
+    let
+      target = cast[EmitTargetKind](t[n.firstSon].rawOperand)
+      code = lastSon(t, n)
+
+    case target:
+      of Asm:
+        requireInfo t, n, InPure
+
+        # Resolve asm overloads (as example for icc inlineAsmSyntax pragma must be specified)
+        var inlineAsmSyntax = c.props.inlineAsmSyntax
+        if haveInfo(t, n, InlineAsmSyntax):
+          inlineAsmSyntax = inlineAsmSyntax * {
+            fetchInfo(t, n, InlineAsmSyntax).infoVal(
+              t, InlineAsmSyntaxKind
+            )
+          }
+
+        if inlineAsmSyntax.len > 1:
+          raiseAssert "Ambiguous asm syntax, please specify via inlineAsmSyntax pragma"
+
+        let isBasicAsm = (
+          fetchInfo(t, n, IsGlobal).infoVal(t, bool) or
+          fetchInfo(t, n, InPure).infoVal(t, bool)
+        ) and inlineAsmSyntax != {VisualCPP}
+
+        if inlineAsmSyntax.len == 0 and not isBasicAsm:
+          raiseAssert "Your compiler does not support the specified inline assembler"
+
+        if not isBasicAsm:
+          if inlineAsmSyntax == {GCCExtendedAsm}: genGccAsm(c, t, code)
+          elif inlineAsmSyntax == {VisualCPP}: genVisualCPPAsm(c, t, code)
+          else: raiseAssert "Not implemented inline asm syntax"
+        else:
+          assert (
+            isLastSon(t, code, code.firstSon) and
+            t[code.firstSon].kind == Verbatim
+          ), "Invalid basic asm. Basic asm should be only one verbatim"
+          genGccAsm(c, t, code)
+
+      of Code:
+        raiseAssert"not supported"
+
+  of EmitTarget:
+    discard
+
+  of EmitCode:
+    for ch in sons(t, n):
+      gen c, t, ch
+
+  of Info, InfoId:
+    discard "Info can't generate code"
 
 const
   Prelude = """
@@ -940,6 +1050,11 @@ typedef NU8 NU;
 #define nimCheckRange(x, a, b, L) ({if (x < a || x > b) goto L; x})
 #define nimCheckIndex(x, a, L) ({if (x >= a) goto L; x})
 
+#ifdef _MSC_VER
+#define NIM_NAKED __declspec(naked)
+#else
+#define NIM_NAKED __attribute__((naked))
+#endif
 """
 
 proc traverseCode(c: var GeneratedCode) =
@@ -958,8 +1073,8 @@ proc traverseCode(c: var GeneratedCode) =
         c.init.add c.code[i]
       setLen c.code, oldLen
 
-proc generateCode*(inp, outp: string) =
-  var c = initGeneratedCode(load(inp))
+proc generateCode*(inp, outp: string; props: sink TargetProps) =
+  var c = initGeneratedCode(load(inp), props)
 
   var co = TypeOrder()
   traverseTypes(c.m.types, c.m.lit, co)