# # # The Nim Compiler # (c) Copyright 2015 Andreas Rumpf # # See the file "copying.txt", included in this # distribution, for details about the copyright. # ## This module implements the C code generator. import ast, astalgo, trees, platform, magicsys, extccomp, options, nversion, nimsets, msgs, bitsets, idents, types, ccgutils, ropes, wordrecg, treetab, cgmeth, rodutils, renderer, cgendata, aliases, lowerings, lineinfos, pathutils, transf, injectdestructors, astmsgs, modulepaths, pushpoppragmas, mangleutils, cbuilderbase, modulegraphs from expanddefaults import caseObjDefaultBranch import pipelineutils when defined(nimPreviewSlimSystem): import std/assertions when not defined(leanCompiler): import spawn, semparallel import std/strutils except `%`, addf # collides with ropes.`%` import std/[dynlib, math, tables, sets, os, intsets, hashes] const # we use some ASCII control characters to insert directives that will be converted to real code in a postprocessing pass postprocessDirStart = '\1' postprocessDirSep = '\31' postprocessDirEnd = '\23' when not declared(dynlib.libCandidates): proc libCandidates(s: string, dest: var seq[string]) = ## given a library name pattern `s` write possible library names to `dest`. var le = strutils.find(s, '(') var ri = strutils.find(s, ')', le+1) if le >= 0 and ri > le: var prefix = substr(s, 0, le - 1) var suffix = substr(s, ri + 1) for middle in split(substr(s, le + 1, ri - 1), '|'): libCandidates(prefix & middle & suffix, dest) else: dest.add(s) when options.hasTinyCBackend: import tccgen proc hcrOn(m: BModule): bool = m.config.hcrOn proc hcrOn(p: BProc): bool = p.module.config.hcrOn proc addForwardedProc(m: BModule, prc: PSym) = m.g.forwardedProcs.add(prc) proc newModule*(g: BModuleList; module: PSym; conf: ConfigRef; idgen: IdGenerator): BModule proc findPendingModule(m: BModule, s: PSym): BModule = # TODO fixme if m.config.symbolFiles == v2Sf or optCompress in m.config.globalOptions: let ms = s.itemId.module #getModule(s) result = m.g.mods[ms] elif m.config.cmd in {cmdNifC, cmdM}: var ms = getModule(s) registerModule m.g.graph, ms if ms.position >= m.g.mods.len: result = newModule(m.g, ms, m.config, idGeneratorFromModule(ms)) else: result = m.g.mods[ms.position] if result == nil: result = newModule(m.g, ms, m.config, idGeneratorFromModule(ms)) else: var ms = getModule(s) result = m.g.mods[ms.position] proc initLoc(k: TLocKind, lode: PNode, s: TStorageLoc, flags: TLocFlags = {}): TLoc = result = TLoc(k: k, storage: s, lode: lode, snippet: "", flags: flags) proc fillLoc(a: var TLoc, k: TLocKind, lode: PNode, r: Rope, s: TStorageLoc) {.inline.} = # fills the loc if it is not already initialized if a.k == locNone: a.k = k a.lode = lode a.storage = s if a.snippet == "": a.snippet = r proc fillLoc(a: var TLoc, k: TLocKind, lode: PNode, s: TStorageLoc) {.inline.} = # fills the loc if it is not already initialized if a.k == locNone: a.k = k a.lode = lode a.storage = s proc t(a: TLoc): PType {.inline.} = if a.lode.kind == nkSym: result = a.lode.sym.typ else: result = a.lode.typ proc lodeTyp(t: PType): PNode = result = newNode(nkEmpty) result.typ = t proc isSimpleConst(typ: PType): bool = let t = skipTypes(typ, abstractVar) result = t.kind notin {tyTuple, tyObject, tyArray, tySet, tySequence} and not (t.kind == tyProc and t.callConv == ccClosure) proc useHeader(m: BModule, sym: PSym) = if lfHeader in sym.loc.flags: assert(sym.annex != nil) let str = getStr(sym.annex.path) m.includeHeader(str) proc cgsym(m: BModule, name: string) proc cgsymValue(m: BModule, name: string): Rope proc getCFile(m: BModule): AbsoluteFile proc getModuleDllPath(m: BModule): Rope = let (dir, name, ext) = splitFile(getCFile(m)) let filename = strutils.`%`(platform.OS[m.g.config.target.targetOS].dllFrmt, [name & ext]) result = makeCString(dir.string & "/" & filename) proc getModuleDllPath(m: BModule, module: int): Rope = result = getModuleDllPath(m.g.mods[module]) proc getModuleDllPath(m: BModule, s: PSym): Rope = result = getModuleDllPath(m.g.mods[s.itemId.module]) import std/macros proc cgFormatValue(result: var string; value: string) = result.add value proc cgFormatValue(result: var string; value: BiggestInt) = result.addInt value proc cgFormatValue(result: var string; value: Int128) = result.addInt128 value template addf(result: var Builder, args: varargs[untyped]) = result.buf.addf(args) # TODO: please document macro ropecg(m: BModule, frmt: static[FormatStr], args: untyped): Rope = args.expectKind nnkBracket # echo "ropecg ", newLit(frmt).repr, ", ", args.repr var i = 0 result = nnkStmtListExpr.newTree() result.add quote do: assert `m` != nil let resVar = genSym(nskVar, "res") # during `koch boot` the median of all generates strings from this # macro is around 40 bytes in length. result.add newVarStmt(resVar, newCall(bindSym"newStringOfCap", newLit(80))) let formatValue = bindSym"cgFormatValue" var num = 0 var strLit = "" template flushStrLit() = if strLit != "": result.add newCall(ident "add", resVar, newLit(strLit)) strLit.setLen 0 while i < frmt.len: if frmt[i] == '$': inc(i) # skip '$' case frmt[i] of '$': strLit.add '$' inc(i) of '#': flushStrLit() inc(i) result.add newCall(formatValue, resVar, args[num]) inc(num) of '^': flushStrLit() inc(i) result.add newCall(formatValue, resVar, args[^1]) inc(num) of '0'..'9': var j = 0 while true: j = (j * 10) + ord(frmt[i]) - ord('0') inc(i) if i >= frmt.len or not (frmt[i] in {'0'..'9'}): break num = j if j > args.len: error("ropes: invalid format string " & newLit(frmt).repr & " args.len: " & $args.len) flushStrLit() result.add newCall(formatValue, resVar, args[j-1]) of 'n': flushStrLit() result.add quote do: if optLineDir notin `m`.config.options: `resVar`.add("\L") inc(i) of 'N': strLit.add "\L" inc(i) else: error("ropes: invalid format string $" & frmt[i]) elif frmt[i] == '#' and frmt[i+1] in IdentStartChars: inc(i) var j = i while frmt[j] in IdentChars: inc(j) var ident = newLit(substr(frmt, i, j-1)) i = j flushStrLit() result.add newCall(formatValue, resVar, newCall(ident"cgsymValue", m, ident)) elif frmt[i] == '#' and frmt[i+1] == '$': inc(i, 2) var j = 0 while frmt[i] in Digits: j = (j * 10) + ord(frmt[i]) - ord('0') inc(i) let ident = args[j-1] flushStrLit() result.add newCall(formatValue, resVar, newCall(ident"cgsymValue", m, ident)) elif frmt[i] == '#' and frmt[i+1] == '#': inc(i, 2) strLit.add("#") else: strLit.add(frmt[i]) inc(i) flushStrLit() result.add newCall(ident"rope", resVar) proc addIndent(p: BProc; result: var Rope) = var i = result.len let newLen = i + p.blocks.len result.setLen newLen while i < newLen: result[i] = '\t' inc i proc addIndent(p: BProc; result: var Builder) = var i = result.buf.len let newLen = i + p.blocks.len result.buf.setLen newLen while i < newLen: result.buf[i] = '\t' inc i template appcg(m: BModule, c: var (Rope | Builder), frmt: FormatStr, args: untyped) = c.add(ropecg(m, frmt, args)) template appcg(m: BModule, sec: TCFileSection, frmt: FormatStr, args: untyped) = m.s[sec].add(ropecg(m, frmt, args)) template appcg(p: BProc, sec: TCProcSection, frmt: FormatStr, args: untyped) = p.s(sec).add(ropecg(p.module, frmt, args)) template line(p: BProc, sec: TCProcSection, r: string) = addIndent p, p.s(sec) p.s(sec).add(r) template lineF(p: BProc, sec: TCProcSection, frmt: FormatStr, args: untyped) = addIndent p, p.s(sec) p.s(sec).add(frmt % args) template lineCg(p: BProc, sec: TCProcSection, frmt: FormatStr, args: untyped) = addIndent p, p.s(sec) p.s(sec).add(ropecg(p.module, frmt, args)) template linefmt(p: BProc, sec: TCProcSection, frmt: FormatStr, args: untyped) = addIndent p, p.s(sec) p.s(sec).add(ropecg(p.module, frmt, args)) proc safeLineNm(info: TLineInfo): int = result = toLinenumber(info) if result < 0: result = 0 # negative numbers are not allowed in #line proc genPostprocessDir(field1, field2, field3: string): string = result = postprocessDirStart & field1 & postprocessDirSep & field2 & postprocessDirSep & field3 & postprocessDirEnd proc genCLineDir(r: var Builder, fileIdx: FileIndex, line: int; conf: ConfigRef) = assert line >= 0 if optLineDir in conf.options and line > 0: if fileIdx == InvalidFileIdx: r.add(rope("\n#line " & $line & " \"generated_not_to_break_here\"\n")) else: r.add(rope("\n#line " & $line & " FX_" & $fileIdx.int32 & "\n")) proc genCLineDir(r: var Builder, fileIdx: FileIndex, line: int; p: BProc; info: TLineInfo; lastFileIndex: FileIndex) = assert line >= 0 if optLineDir in p.config.options and line > 0: if fileIdx == InvalidFileIdx: r.add(rope("\n#line " & $line & " \"generated_not_to_break_here\"\n")) else: r.add(rope("\n#line " & $line & " FX_" & $fileIdx.int32 & "\n")) proc genCLineDir(r: var Builder, info: TLineInfo; conf: ConfigRef) = if optLineDir in conf.options: genCLineDir(r, info.fileIndex, info.safeLineNm, conf) proc freshLineInfo(p: BProc; info: TLineInfo): bool = if p.lastLineInfo.line != info.line or p.lastLineInfo.fileIndex != info.fileIndex: p.lastLineInfo.line = info.line p.lastLineInfo.fileIndex = info.fileIndex result = true else: result = false proc genCLineDir(r: var Builder, p: BProc, info: TLineInfo; conf: ConfigRef) = if optLineDir in conf.options: let lastFileIndex = p.lastLineInfo.fileIndex if freshLineInfo(p, info): genCLineDir(r, info.fileIndex, info.safeLineNm, p, info, lastFileIndex) proc genLineDir(p: BProc, t: PNode) = if p == p.module.preInitProc: return let line = t.info.safeLineNm if optEmbedOrigSrc in p.config.globalOptions: var code = sourceLine(p.config, t.info) if code.endsWith('\\'): code.add "#" p.s(cpsStmts).add("// " & code & "\L") let lastFileIndex = p.lastLineInfo.fileIndex let freshLine = freshLineInfo(p, t.info) if freshLine: genCLineDir(p.s(cpsStmts), t.info.fileIndex, line, p, t.info, lastFileIndex) if ({optLineTrace, optStackTrace} * p.options == {optLineTrace, optStackTrace}) and (p.prc == nil or sfPure notin p.prc.flags) and t.info.fileIndex != InvalidFileIdx: if freshLine: line(p, cpsStmts, genPostprocessDir("nimln", $line, $t.info.fileIndex.int32)) proc accessThreadLocalVar(p: BProc, s: PSym) proc emulatedThreadVars(conf: ConfigRef): bool {.inline.} proc genProc(m: BModule, prc: PSym) proc raiseInstr(p: BProc; result: var Builder) template compileToCpp(m: BModule): untyped = m.config.backend == backendCpp or sfCompileToCpp in m.module.flags proc getTempName(m: BModule): Rope = result = m.tmpBase & rope(m.labels) inc m.labels proc isNoReturn(m: BModule; s: PSym): bool {.inline.} = sfNoReturn in s.flags and m.config.exc != excGoto include cbuilderexprs include cbuilderdecls include cbuilderstmts proc rdLoc(a: TLoc): Rope = # 'read' location (deref if indirect) if lfIndirect in a.flags: result = cDeref(a.snippet) else: result = a.snippet proc addRdLoc(a: TLoc; result: var Builder) = if lfIndirect in a.flags: result.add cDeref(a.snippet) else: result.add a.snippet proc lenField(p: BProc, val: Rope): Rope {.inline.} = if p.module.compileToCpp: result = derefField(val, "len") else: result = dotField(derefField(val, "Sup"), "len") proc lenExpr(p: BProc; a: TLoc): Rope = if optSeqDestructors in p.config.globalOptions: result = dotField(rdLoc(a), "len") else: let ra = rdLoc(a) result = cIfExpr(ra, lenField(p, ra), cIntValue(0)) proc dataFieldAccessor(p: BProc, sym: Rope): Rope = if optSeqDestructors in p.config.globalOptions: result = dotField(wrapPar(sym), "p") else: result = sym proc dataField(p: BProc, val: Rope): Rope {.inline.} = result = derefField(dataFieldAccessor(p, val), "data") proc genProcPrototype(m: BModule, sym: PSym) include ccgliterals include ccgtypes # ------------------------------ Manager of temporaries ------------------ template mapTypeChooser(n: PNode): TSymKind = (if n.kind == nkSym: n.sym.kind else: skVar) template mapTypeChooser(a: TLoc): TSymKind = mapTypeChooser(a.lode) proc addAddrLoc(conf: ConfigRef; a: TLoc; result: var Builder) = if lfIndirect notin a.flags and mapType(conf, a.t, mapTypeChooser(a) == skParam) != ctArray: result.add wrapPar(cAddr(a.snippet)) else: result.add a.snippet proc addrLoc(conf: ConfigRef; a: TLoc): Rope = if lfIndirect notin a.flags and mapType(conf, a.t, mapTypeChooser(a) == skParam) != ctArray: result = wrapPar(cAddr(a.snippet)) else: result = a.snippet proc byRefLoc(p: BProc; a: TLoc): Rope = if lfIndirect notin a.flags and mapType(p.config, a.t, mapTypeChooser(a) == skParam) != ctArray and not p.module.compileToCpp: result = wrapPar(cAddr(a.snippet)) else: result = a.snippet proc rdCharLoc(a: TLoc): Rope = # read a location that may need a char-cast: result = rdLoc(a) if skipTypes(a.t, abstractRange).kind == tyChar: result = cCast(NimUint8, result) type TAssignmentFlag = enum needToCopy needTempForOpenArray needAssignCall TAssignmentFlags = set[TAssignmentFlag] proc genObjConstr(p: BProc, e: PNode, d: var TLoc) proc rawConstExpr(p: BProc, n: PNode; d: var TLoc) proc genAssignment(p: BProc, dest, src: TLoc, flags: TAssignmentFlags) type ObjConstrMode = enum constructObj, constructRefObj proc genObjectInit(p: BProc, section: TCProcSection, t: PType, a: var TLoc, mode: ObjConstrMode) = #if optNimV2 in p.config.globalOptions: return case analyseObjectWithTypeField(t) of frNone: discard of frHeader: var r = rdLoc(a) if mode == constructRefObj: r = cDeref(r) var s = skipTypes(t, abstractInst) if not p.module.compileToCpp: while s.kind == tyObject and s[0] != nil: r = dotField(r, "Sup") s = skipTypes(s[0], skipPtrs) if optTinyRtti in p.config.globalOptions: p.s(section).addFieldAssignment(r, "m_type", genTypeInfoV2(p.module, t, a.lode.info)) else: p.s(section).addFieldAssignment(r, "m_type", genTypeInfoV1(p.module, t, a.lode.info)) of frEmbedded: if optTinyRtti in p.config.globalOptions: var tmp: TLoc = default(TLoc) if mode == constructRefObj: let objType = t.skipTypes(abstractInst+{tyRef}) rawConstExpr(p, newNodeIT(nkType, a.lode.info, objType), tmp) let ra = rdLoc(a) let rtmp = rdLoc(tmp) let rt = getTypeDesc(p.module, objType, descKindFromSymKind mapTypeChooser(a)) p.s(cpsStmts).addCallStmt(cgsymValue(p.module, "nimCopyMem"), cCast(CPointer, ra), cCast(CConstPointer, cAddr(rtmp)), cSizeof(rt)) else: rawConstExpr(p, newNodeIT(nkType, a.lode.info, t), tmp) genAssignment(p, a, tmp, {}) else: # worst case for performance: var r = if mode == constructObj: addrLoc(p.config, a) else: rdLoc(a) p.s(section).addCallStmt(cgsymValue(p.module, "objectInit"), r, genTypeInfoV1(p.module, t, a.lode.info)) if isException(t): var r = rdLoc(a) if mode == constructRefObj: r = cDeref(r) var s = skipTypes(t, abstractInst) if not p.module.compileToCpp: while s.kind == tyObject and s[0] != nil and s.sym.magic != mException: r = dotField(r, "Sup") s = skipTypes(s[0], skipPtrs) p.s(section).addFieldAssignment(r, "name", makeCString(t.skipTypes(abstractInst).sym.name.s)) proc genRefAssign(p: BProc, dest, src: TLoc) proc isComplexValueType(t: PType): bool {.inline.} = let t = t.skipTypes(abstractInst + tyUserTypeClasses) result = t.kind in {tyArray, tySet, tyTuple, tyObject, tyOpenArray} or (t.kind == tyProc and t.callConv == ccClosure) include ccgreset proc resetLoc(p: BProc, loc: var TLoc) = let containsGcRef = optSeqDestructors notin p.config.globalOptions and containsGarbageCollectedRef(loc.t) let typ = skipTypes(loc.t, abstractVarRange) if isImportedCppType(typ): var didGenTemp = false let rl = rdLoc(loc) let init = genCppInitializer(p.module, p, typ, didGenTemp) p.s(cpsStmts).addAssignment(rl, init) return if optSeqDestructors in p.config.globalOptions and typ.kind in {tyString, tySequence}: assert loc.snippet != "" let atyp = skipTypes(loc.t, abstractInst) let rl = rdLoc(loc) if atyp.kind in {tyVar, tyLent}: p.s(cpsStmts).addAssignment(derefField(rl, "len"), cIntValue(0)) p.s(cpsStmts).addAssignment(derefField(rl, "p"), NimNil) else: p.s(cpsStmts).addAssignment(dotField(rl, "len"), cIntValue(0)) p.s(cpsStmts).addAssignment(dotField(rl, "p"), NimNil) elif not isComplexValueType(typ): if containsGcRef: var nilLoc: TLoc = initLoc(locTemp, loc.lode, OnStack) nilLoc.snippet = NimNil genRefAssign(p, loc, nilLoc) else: p.s(cpsStmts).addAssignment(rdLoc(loc), cIntValue(0)) else: if loc.storage != OnStack and containsGcRef: specializeReset(p, loc) when false: linefmt(p, cpsStmts, "#genericReset((void*)$1, $2);$n", [addrLoc(p.config, loc), genTypeInfoV1(p.module, loc.t, loc.lode.info)]) # XXX: generated reset procs should not touch the m_type # field, so disabling this should be safe: genObjectInit(p, cpsStmts, loc.t, loc, constructObj) else: # array passed as argument decayed into pointer, bug #7332 # so we use getTypeDesc here rather than rdLoc(loc) let tyDesc = getTypeDesc(p.module, loc.t, descKindFromSymKind mapTypeChooser(loc)) if p.module.compileToCpp and isOrHasImportedCppType(typ): if lfIndirect in loc.flags: #C++ cant be just zeroed. We need to call the ctors var tmp = getTemp(p, loc.t) let ral = addrLoc(p.config, loc) let ratmp = addrLoc(p.config, tmp) p.s(cpsStmts).addCallStmt(cgsymValue(p.module, "nimCopyMem"), cCast(CPointer, ral), cCast(CConstPointer, ratmp), cSizeof(tyDesc)) else: let ral = addrLoc(p.config, loc) p.s(cpsStmts).addCallStmt(cgsymValue(p.module, "nimZeroMem"), cCast(CPointer, ral), cSizeof(tyDesc)) # XXX: We can be extra clever here and call memset only # on the bytes following the m_type field? genObjectInit(p, cpsStmts, loc.t, loc, constructObj) proc constructLoc(p: BProc, loc: var TLoc, isTemp = false) = let typ = loc.t if optSeqDestructors in p.config.globalOptions and skipTypes(typ, abstractInst + {tyStatic}).kind in {tyString, tySequence}: let rl = rdLoc(loc) p.s(cpsStmts).addFieldAssignment(rl, "len", cIntValue(0)) p.s(cpsStmts).addFieldAssignment(rl, "p", NimNil) elif not isComplexValueType(typ): if containsGarbageCollectedRef(loc.t): var nilLoc: TLoc = initLoc(locTemp, loc.lode, OnStack) nilLoc.snippet = NimNil genRefAssign(p, loc, nilLoc) else: let rl = rdLoc(loc) let rt = getTypeDesc(p.module, typ, descKindFromSymKind mapTypeChooser(loc)) p.s(cpsStmts).addAssignment(rl, cCast(rt, cIntValue(0))) else: if (not isTemp or containsGarbageCollectedRef(loc.t)) and not hasNoInit(loc.t): # don't use nimZeroMem for temporary values for performance if we can # avoid it: if not isOrHasImportedCppType(typ): let ral = addrLoc(p.config, loc) let rt = getTypeDesc(p.module, typ, descKindFromSymKind mapTypeChooser(loc)) p.s(cpsStmts).addCallStmt(cgsymValue(p.module, "nimZeroMem"), cCast(CPointer, ral), cSizeof(rt)) genObjectInit(p, cpsStmts, loc.t, loc, constructObj) proc initLocalVar(p: BProc, v: PSym, immediateAsgn: bool) = if sfNoInit notin v.flags: # we know it is a local variable and thus on the stack! # If ``not immediateAsgn`` it is not initialized in a binding like # ``var v = X`` and thus we need to init it. # If ``v`` contains a GC-ref we may pass it to ``unsureAsgnRef`` somehow # which requires initialization. However this can really only happen if # ``var v = X()`` gets transformed into ``X(&v)``. # Nowadays the logic in ccgcalls deals with this case however. if not immediateAsgn: backendEnsureMutable v constructLoc(p, v.locImpl) proc getTemp(p: BProc, t: PType, needsInit=false): TLoc = inc(p.labels) result = TLoc(snippet: "T" & rope(p.labels) & "_", k: locTemp, lode: lodeTyp t, storage: OnStack, flags: {}) if p.module.compileToCpp and isOrHasImportedCppType(t): var didGenTemp = false linefmt(p, cpsLocals, "$1 $2$3;$n", [getTypeDesc(p.module, t, dkVar), result.snippet, genCppInitializer(p.module, p, t, didGenTemp)]) else: p.s(cpsLocals).addVar(kind = Local, name = result.snippet, typ = getTypeDesc(p.module, t, dkVar)) constructLoc(p, result, not needsInit) when false: # XXX Introduce a compiler switch in order to detect these easily. if getSize(p.config, t) > 1024 * 1024: if p.prc != nil: echo "ENORMOUS TEMPORARY! ", p.config $ p.prc.info else: echo "ENORMOUS TEMPORARY! ", p.config $ p.lastLineInfo writeStackTrace() proc getTempCpp(p: BProc, t: PType, value: Rope): TLoc = inc(p.labels) result = TLoc(snippet: "T" & rope(p.labels) & "_", k: locTemp, lode: lodeTyp t, storage: OnStack, flags: {}) p.s(cpsStmts).addVar(kind = Local, name = result.snippet, typ = "auto", initializer = value) proc getIntTemp(p: BProc): TLoc = inc(p.labels) result = TLoc(snippet: "T" & rope(p.labels) & "_", k: locTemp, storage: OnStack, lode: lodeTyp getSysType(p.module.g.graph, unknownLineInfo, tyInt), flags: {}) p.s(cpsLocals).addVar(kind = Local, name = result.snippet, typ = NimInt) proc localVarDecl(res: var Builder, p: BProc; n: PNode, initializer: Snippet = "", initializerKind: VarInitializerKind = Assignment) = let s = n.sym if s.loc.k == locNone: fillLocalName(p, s) backendEnsureMutable s fillLoc(s.locImpl, locLocalVar, n, OnStack) if s.kind == skLet: incl(s, lfNoDeepCopy) genCLineDir(res, p, n.info, p.config) res.addVar(p.module, s, name = s.loc.snippet, typ = getTypeDesc(p.module, s.typ, dkVar), initializer = initializer, initializerKind = initializerKind) proc assignLocalVar(p: BProc, n: PNode) = #assert(s.loc.k == locNone) # not yet assigned # this need not be fulfilled for inline procs; they are regenerated # for each module that uses them! var initializer: Snippet = "" var initializerKind: VarInitializerKind = Assignment if p.module.compileToCpp and isOrHasImportedCppType(n.typ): var didGenTemp = false initializer = genCppInitializer(p.module, p, n.typ, didGenTemp) initializerKind = CppConstructor localVarDecl(p.s(cpsLocals), p, n, initializer, initializerKind) if optLineDir in p.config.options: p.s(cpsLocals).add("\n") include ccgthreadvars proc varInDynamicLib(m: BModule, sym: PSym) proc treatGlobalDifferentlyForHCR(m: BModule, s: PSym): bool = return m.hcrOn and {sfThread, sfGlobal} * s.flags == {sfGlobal} and ({lfNoDecl, lfHeader} * s.loc.flags == {}) # and s.owner.kind == skModule # owner isn't always a module (global pragma on local var) # and s.loc.k == locGlobalVar # loc isn't always initialized when this proc is used proc genGlobalVarDecl(res: var Builder, p: BProc, n: PNode; td: Snippet; initializer: Snippet = "", initializerKind: VarInitializerKind = Assignment, allowConst = true) = let s = n.sym let vis = if p.hcrOn: StaticProc elif sfImportc in s.flags: Extern elif lfExportLib in s.loc.flags: ExportLibVar else: Private var typ = td if allowConst and s.kind == skLet and initializer.len != 0: typ = constType(typ) if p.hcrOn: typ = ptrType(typ) res.addVar(p.module, s, name = s.loc.snippet, typ = typ, visibility = vis, initializer = initializer, initializerKind = initializerKind) proc assignGlobalVar(p: BProc, n: PNode; value: Rope) = let s = n.sym if s.loc.k == locNone: fillBackendName(p.module, s) backendEnsureMutable s fillLoc(s.locImpl, locGlobalVar, n, OnHeap) if treatGlobalDifferentlyForHCR(p.module, s): incl(s, lfIndirect) if lfDynamicLib in s.loc.flags: var q = findPendingModule(p.module, s) if q != nil and not containsOrIncl(q.declaredThings, s.id): varInDynamicLib(q, s) else: backendEnsureMutable s s.locImpl.snippet = mangleDynLibProc(s) if value != "": internalError(p.config, n.info, ".dynlib variables cannot have a value") return useHeader(p.module, s) if lfNoDecl in s.loc.flags: return if not containsOrIncl(p.module.declaredThings, s.id): if sfThread in s.flags: declareThreadVar(p.module, s, sfImportc in s.flags) if value != "": internalError(p.config, n.info, ".threadvar variables cannot have a value") else: let td = getTypeDesc(p.module, s.loc.t, dkVar) var initializer: Snippet = "" if s.constraint.isNil: if value != "": if p.module.compileToCpp and value.startsWith "{{}": # TODO: taking this branch, re"\{\{\}(,\s\{\})*\}" might be emitted, resulting in # either warnings (GCC 12.2+) or errors (Clang 15, MSVC 19.3+) of C++11+ compilers **when # explicit constructors are around** due to overload resolution rules in place [^0][^1][^2] # *Workaround* here: have C++'s static initialization mechanism do the default init work, # for us lacking a deeper knowledge of an imported object's constructors' ex-/implicitness # (so far) *and yet* trying to achieve default initialization. # Still, generating {}s in genConstObjConstr() just to omit them here is faaaar from ideal; # need to figure out a better way, possibly by keeping around more data about the # imported objects' contructors? # # [^0]: https://en.cppreference.com/w/cpp/language/aggregate_initialization # [^1]: https://cplusplus.github.io/CWG/issues/1518.html # [^2]: https://eel.is/c++draft/over.match.ctor discard else: initializer = value else: discard else: initializer = value genGlobalVarDecl(p.module.s[cfsVars], p, n, td, initializer = initializer) if p.withinLoop > 0 and value == "": # fixes tests/run/tzeroarray: backendEnsureMutable s resetLoc(p, s.locImpl) proc callGlobalVarCppCtor(p: BProc; v: PSym; vn, value: PNode; didGenTemp: var bool) = let s = vn.sym fillBackendName(p.module, s) backendEnsureMutable s fillLoc(s.locImpl, locGlobalVar, vn, OnHeap) let td = getTypeDesc(p.module, vn.sym.typ, dkVar) var val = genCppParamsForCtor(p, value, didGenTemp) if didGenTemp: return # generated in the caller if val.len != 0: val = "(" & val & ")" genGlobalVarDecl(p.module.s[cfsVars], p, vn, td, initializer = val, initializerKind = CppConstructor, allowConst = false) proc assignParam(p: BProc, s: PSym, retType: PType) = assert(s.loc.snippet != "") scopeMangledParam(p, s) proc fillProcLoc(m: BModule; n: PNode) = let sym = n.sym if sym.loc.k == locNone: fillBackendName(m, sym) backendEnsureMutable sym fillLoc(sym.locImpl, locProc, n, OnStack) proc getLabel(p: BProc): TLabel = inc(p.labels) result = "LA" & rope(p.labels) & "_" proc fixLabel(p: BProc, labl: TLabel) = p.s(cpsStmts).addLabel(labl) proc genVarPrototype(m: BModule, n: PNode) proc requestConstImpl(p: BProc, sym: PSym) proc genStmts(p: BProc, t: PNode) proc expr(p: BProc, n: PNode, d: var TLoc) proc putLocIntoDest(p: BProc, d: var TLoc, s: TLoc) proc genLiteral(p: BProc, n: PNode; result: var Builder) proc genOtherArg(p: BProc; ri: PNode; i: int; typ: PType; result: var Builder; argBuilder: var CallBuilder) proc raiseExit(p: BProc) proc raiseExitCleanup(p: BProc, destroy: string) proc initLocExpr(p: BProc, e: PNode, flags: TLocFlags = {}): TLoc = result = initLoc(locNone, e, OnUnknown, flags) expr(p, e, result) proc initLocExprSingleUse(p: BProc, e: PNode): TLoc = result = initLoc(locNone, e, OnUnknown) if e.kind in nkCallKinds and (e[0].kind != nkSym or e[0].sym.magic == mNone): # We cannot check for tfNoSideEffect here because of mutable parameters. discard "bug #8202; enforce evaluation order for nested calls for C++ too" # We may need to consider that 'f(g())' cannot be rewritten to 'tmp = g(); f(tmp)' # if 'tmp' lacks a move/assignment operator. if e[0].kind == nkSym and sfCompileToCpp in e[0].sym.flags: result.flags.incl lfSingleUse else: result.flags.incl lfSingleUse expr(p, e, result) include ccgcalls, "ccgstmts.nim" proc initFrame(p: BProc, procname, filename: Rope): Rope = # XXX cbuilder const frameDefines = """ $1define nimfr_(proc, file) \ TFrame FR_; \ FR_.procname = proc; FR_.filename = file; FR_.line = 0; FR_.len = 0; #nimFrame(&FR_); $1define nimln_(n) \ FR_.line = n; $1define nimlf_(n, file) \ FR_.line = n; FR_.filename = file; """ if p.module.s[cfsFrameDefines].buf.len == 0: appcg(p.module, p.module.s[cfsFrameDefines], frameDefines, ["#"]) cgsym(p.module, "nimFrame") result = ropecg(p.module, "\tnimfr_($1, $2);$n", [procname, filename]) proc initFrameNoDebug(p: BProc; frame, procname, filename: Snippet; line: int): Snippet = cgsym(p.module, "nimFrame") p.blocks[0].sections[cpsLocals].addVar(name = frame, typ = "TFrame") var res = newBuilder("") res.add('\t') res.addFieldAssignment(frame, "procname", procname) res.add('\t') res.addFieldAssignment(frame, "filename", filename) res.add('\t') res.addFieldAssignment(frame, "line", cIntValue(line)) res.add('\t') res.addFieldAssignment(frame, "len", cIntValue(-1)) res.add('\t') res.addCallStmt("nimFrame", cAddr(frame)) result = extract(res) proc deinitFrameNoDebug(p: BProc; frame: Snippet): Snippet = var res = newBuilder("") res.add('\t') res.addCallStmt(cgsymValue(p.module, "popFrameOfAddr"), cAddr(frame)) result = extract(res) proc deinitFrame(p: BProc): Snippet = var res = newBuilder("") res.add('\t') res.addCallStmt(cgsymValue(p.module, "popFrame")) result = extract(res) include ccgexprs # ----------------------------- dynamic library handling ----------------- # We don't finalize dynamic libs as the OS does this for us. proc isGetProcAddr(lib: PLib): bool = let n = lib.path result = n.kind in nkCallKinds and n.typ != nil and n.typ.kind in {tyPointer, tyProc} proc loadDynamicLib(m: BModule, lib: PLib) = assert(lib != nil) if not lib.generated: lib.generated = true var tmp = getTempName(m) assert(lib.name == "") lib.name = tmp # BUGFIX: cgsym has awful side-effects let loadFn = cgsymValue(m, "nimLoadLibrary") let loadErrorFn = cgsymValue(m, "nimLoadLibraryError") m.s[cfsVars].addVar(Global, name = tmp, typ = CPointer) if lib.path.kind in {nkStrLit..nkTripleStrLit}: var s: TStringSeq = @[] libCandidates(lib.path.strVal, s) rawMessage(m.config, hintDependency, lib.path.strVal) let last = high(s) for i in 0..last: inc(m.labels) template doLoad(j: int) = let n = newStrNode(nkStrLit, s[j]) n.info = lib.path.info m.s[cfsDynLibInit].addAssignmentWithValue(tmp): var call: CallBuilder m.s[cfsDynLibInit].addCall(call, loadFn): m.s[cfsDynLibInit].addArgument(call): genStringLiteral(m, n, m.s[cfsDynLibInit]) if i == 0: doLoad(i) m.s[cfsDynLibInit].addSingleIfStmt(cOp(Not, tmp)): if i == last: m.s[cfsDynLibInit].addStmt(): var call: CallBuilder m.s[cfsDynLibInit].addCall(call, loadErrorFn): m.s[cfsDynLibInit].addArgument(call): genStringLiteral(m, lib.path, m.s[cfsDynLibInit]) else: doLoad(i + 1) else: var p = newProc(nil, m) p.options.excl optStackTrace p.flags.incl nimErrorFlagDisabled var dest: TLoc = initLoc(locTemp, lib.path, OnStack) dest.snippet = getTempName(m) m.s[cfsDynLibInit].addVar(name = rdLoc(dest), typ = getTypeDesc(m, lib.path.typ, dkVar)) expr(p, lib.path, dest) m.s[cfsVars].add(extract(p.s(cpsLocals))) m.s[cfsDynLibInit].add(extract(p.s(cpsInit))) m.s[cfsDynLibInit].add(extract(p.s(cpsStmts))) let rd = rdLoc(dest) m.s[cfsDynLibInit].addAssignment(tmp, cCall(loadFn, rd)) m.s[cfsDynLibInit].addSingleIfStmt(cOp(Not, tmp)): m.s[cfsDynLibInit].addCallStmt(loadErrorFn, rd) if lib.name == "": internalError(m.config, "loadDynamicLib") proc mangleDynLibProc(sym: PSym): Rope = # we have to build this as a single rope in order not to trip the # optimization in genInfixCall, see test tests/cpp/t8241.nim if sfCompilerProc in sym.flags: # NOTE: sym.loc.snippet is the external name! result = rope(sym.name.s) else: result = rope(strutils.`%`("Dl_$1_", $sym.id)) proc symInDynamicLib(m: BModule, sym: PSym) = var lib = sym.annex let isCall = isGetProcAddr(lib) var extname = sym.loc.snippet if not isCall: loadDynamicLib(m, lib) var tmp = mangleDynLibProc(sym) backendEnsureMutable sym sym.locImpl.snippet = tmp # from now on we only need the internal name sym.typ.sym = nil # generate a new name inc(m.labels, 2) if isCall: let n = lib.path var a: TLoc = initLocExpr(m.initProc, n[0]) let callee = rdLoc(a) var params: seq[Snippet] = @[] for i in 1.. 0 and n[0].kind == nkEmpty of nkSym: if n.sym.kind == skResult: result = true else: for i in 0.. 0: result = easyResultAsgn(n[0]) if result != nil: incl n.flags, nfPreventCg else: discard type InitResultEnum = enum Unknown, InitSkippable, InitRequired proc allPathsAsgnResult(p: BProc; n: PNode): InitResultEnum = # Exceptions coming from calls don't have not be considered here: # # proc bar(): string = raise newException(...) # # proc foo(): string = # # optimized out: 'reset(result)' # result = bar() # # try: # a = foo() # except: # echo "a was not written to" # template allPathsInBranch(it) = let a = allPathsAsgnResult(p, it) case a of InitRequired: return InitRequired of InitSkippable: discard of Unknown: # sticky, but can be overwritten by InitRequired: result = Unknown result = Unknown case n.kind of nkStmtList, nkStmtListExpr: for it in n: result = allPathsAsgnResult(p, it) if result != Unknown: return result of nkAsgn, nkFastAsgn, nkSinkAsgn: if n[0].kind == nkSym and n[0].sym.kind == skResult: if not containsResult(n[1]): if allPathsAsgnResult(p, n[1]) == InitRequired: result = InitRequired else: result = InitSkippable else: result = InitRequired elif containsResult(n): result = InitRequired else: result = allPathsAsgnResult(p, n[1]) of nkReturnStmt: if n.len > 0: if n[0].kind == nkEmpty and result != InitSkippable: # This is a bare `return` statement, if `result` was not initialized # anywhere else (or if we're not sure about this) let's require it to be # initialized. This avoids cases like #9286 where this heuristic lead to # wrong code being generated. result = InitRequired else: result = allPathsAsgnResult(p, n[0]) of nkIfStmt, nkIfExpr: var exhaustive = false result = InitSkippable for it in n: # Every condition must not use 'result': if it.len == 2 and containsResult(it[0]): return InitRequired if it.len == 1: exhaustive = true allPathsInBranch(it.lastSon) # if the 'if' statement is not exhaustive and yet it touched 'result' # in some way, say Unknown. if not exhaustive: result = Unknown of nkCaseStmt: if containsResult(n[0]): return InitRequired result = InitSkippable var exhaustive = skipTypes(n[0].typ, abstractVarRange-{tyTypeDesc}).kind notin {tyFloat..tyFloat128, tyString, tyCstring} for i in 1..= prc.ast.len: internalError(m.config, prc.info, "proc has no result symbol") let resNode = prc.ast[resultPos] let res = resNode.sym # get result symbol if not isInvalidReturnType(m.config, prc.typ) and sfConstructor notin prc.flags: if sfNoInit in prc.flags: incl(res, sfNoInit) if sfNoInit in prc.flags and p.module.compileToCpp and (let val = easyResultAsgn(procBody); val != nil): var a: TLoc = initLocExprSingleUse(p, val) let ra = rdLoc(a) localVarDecl(p.s(cpsStmts), p, resNode, initializer = ra) else: # declare the result symbol: assignLocalVar(p, resNode) assert(res.loc.snippet != "") if p.config.selectedGC in {gcArc, gcAtomicArc, gcOrc} and allPathsAsgnResult(p, procBody) == InitSkippable: # In an ideal world the codegen could rely on injectdestructors doing its job properly # and then the analysis step would not be required. discard "result init optimized out" else: initLocalVar(p, res, immediateAsgn=false) var returnBuilder = newBuilder("\t") let rres = rdLoc(res.loc) returnBuilder.addReturn(rres) returnStmt = extract(returnBuilder) elif sfConstructor in prc.flags: resNode.sym.incl lfIndirect backendEnsureMutable resNode.sym fillLoc(resNode.sym.locImpl, locParam, resNode, "this", OnHeap) backendEnsureMutable prc prc.locImpl.snippet = getTypeDesc(m, resNode.sym.locImpl.t, dkVar) else: fillResult(p.config, resNode, prc.typ) assignParam(p, res, prc.typ.returnType) # We simplify 'unsureAsgn(result, nil); unsureAsgn(result, x)' # to 'unsureAsgn(result, x)' # Sketch why this is correct: If 'result' points to a stack location # the 'unsureAsgn' is a nop. If it points to a global variable the # global is either 'nil' or points to valid memory and so the RC operation # succeeds without touching not-initialized memory. if sfNoInit in prc.flags: discard elif allPathsAsgnResult(p, procBody) == InitSkippable: discard else: backendEnsureMutable res resetLoc(p, res.locImpl) if skipTypes(res.typ, abstractInst).kind == tyArray: #incl(res.loc.flags, lfIndirect) backendEnsureMutable res res.locImpl.storage = OnUnknown for i in 1.. 0: result.add("#define USE_NIM_NAMESPACE ") result.add(conf.cppCustomNamespace) result.add("\L") if conf.isDefined("nimEmulateOverflowChecks"): result.add("#define NIM_EmulateOverflowChecks\L") proc headerTop(): Rope = result = "/* Generated by Nim Compiler v$1 */$N" % [rope(VersionAsString)] proc getCopyright(conf: ConfigRef; cfile: Cfile): Rope = result = headerTop() if optCompileOnly notin conf.globalOptions: result.add ("/* Compiled for: $1, $2, $3 */$N" & "/* Command for C compiler:$n $4 */$N") % [rope(platform.OS[conf.target.targetOS].name), rope(platform.CPU[conf.target.targetCPU].name), rope(extccomp.CC[conf.cCompiler].name), rope(getCompileCFileCmd(conf, cfile))] proc getFileHeader(conf: ConfigRef; cfile: Cfile): Rope = var res = newBuilder(getCopyright(conf, cfile)) if conf.hcrOn: res.add("#define NIM_HOT_CODE_RELOADING\L") addNimDefines(res, conf) result = extract(res) proc getSomeNameForModule(conf: ConfigRef, filename: AbsoluteFile): Rope = ## Returns a mangled module name. result = mangleModuleName(conf, filename).mangle proc getSomeNameForModule(m: BModule): Rope = ## Returns a mangled module name. assert m.module.kind == skModule assert m.module.owner.kind == skPackage result = mangleModuleName(m.g.config, m.filename).mangle proc getSomeInitName(m: BModule, suffix: string): Rope = if not m.hcrOn: result = getSomeNameForModule(m) else: result = "" result.add suffix proc getInitName(m: BModule): Rope = if sfMainModule in m.module.flags: # generate constant name for main module, for "easy" debugging. result = rope(m.config.nimMainPrefix) & rope"NimMainModule" else: result = getSomeInitName(m, "Init000") proc getDatInitName(m: BModule): Rope = getSomeInitName(m, "DatInit000") proc getHcrInitName(m: BModule): Rope = getSomeInitName(m, "HcrInit000") proc hcrGetProcLoadCode(builder: var Builder, m: BModule, sym, prefix, handle, getProcFunc: string) # The use of a volatile function pointer to call Pre/NimMainInner # prevents inlining of the NimMainInner function and dependent # functions, which might otherwise merge their stack frames. proc isInnerMainVolatile(m: BModule): bool = m.config.selectedGC notin {gcNone, gcArc, gcAtomicArc, gcOrc} proc genPreMain(m: BModule) = m.s[cfsProcs].addDeclWithVisibility(Private): m.s[cfsProcs].addProcHeader(m.config.nimMainPrefix & "PreMainInner", CVoid, cProcParams()) m.s[cfsProcs].finishProcHeaderWithBody(): m.s[cfsProcs].add(extract(m.g.otherModsInit)) if optNoMain notin m.config.globalOptions: m.s[cfsProcs].addDeclWithVisibility(Private): m.s[cfsProcs].addVar(name = "cmdCount", typ = CInt) m.s[cfsProcs].addDeclWithVisibility(Private): m.s[cfsProcs].addVar(name = "cmdLine", typ = ptrType(ptrType(CChar))) m.s[cfsProcs].addDeclWithVisibility(Private): m.s[cfsProcs].addVar(name = "gEnv", typ = ptrType(ptrType(CChar))) m.s[cfsProcs].addDeclWithVisibility(Private): m.s[cfsProcs].addProcHeader(m.config.nimMainPrefix & "PreMain", CVoid, cProcParams()) m.s[cfsProcs].finishProcHeaderWithBody(): if isInnerMainVolatile(m): m.s[cfsProcs].addProcVar(name = "inner", rettype = CVoid, params = cProcParams(), isVolatile = true) m.s[cfsProcs].addAssignment("inner", m.config.nimMainPrefix & "PreMainInner") m.s[cfsProcs].add(extract(m.g.mainDatInit)) m.s[cfsProcs].addCallStmt(cDeref("inner")) else: # not volatile m.s[cfsProcs].add(extract(m.g.mainDatInit)) m.s[cfsProcs].addCallStmt(m.config.nimMainPrefix & "PreMainInner") proc genMainProcs(m: BModule) = m.s[cfsProcs].addCallStmt(m.config.nimMainPrefix & "NimMain") proc genMainProcsWithResult(m: BModule) = genMainProcs(m) if m.config.cmd != cmdNifC: var res = "nim_program_result" if m.hcrOn: res = cDeref(res) m.s[cfsProcs].addReturn(res) else: m.s[cfsProcs].addReturn(cIntValue(0)) proc genNimMainInner(m: BModule) = m.s[cfsProcs].addDeclWithVisibility(Private): m.s[cfsProcs].addProcHeader(ccCDecl, m.config.nimMainPrefix & "NimMainInner", CVoid, cProcParams()) m.s[cfsProcs].finishProcHeaderWithBody(): m.s[cfsProcs].add(extract(m.g.mainModInit)) m.s[cfsProcs].addNewline() proc initStackBottom(m: BModule): bool = not (m.config.target.targetOS == osStandalone or m.config.selectedGC in {gcNone, gcArc, gcAtomicArc, gcOrc}) proc genNimMainProc(m: BModule, preMainCode: Snippet) = m.s[cfsProcs].addProcHeader(ccCDecl, m.config.nimMainPrefix & "NimMain", CVoid, cProcParams()) m.s[cfsProcs].finishProcHeaderWithBody(): if isInnerMainVolatile(m): m.s[cfsProcs].addProcVar(name = "inner", rettype = CVoid, params = cProcParams(), isVolatile = true) m.s[cfsProcs].add(preMainCode) m.s[cfsProcs].addAssignment("inner", m.config.nimMainPrefix & "NimMainInner") if initStackBottom(m): m.s[cfsProcs].addCallStmt("initStackBottomWith", cCast(CPointer, cAddr("inner"))) m.s[cfsProcs].addCallStmt(cDeref("inner")) else: # not volatile m.s[cfsProcs].add(preMainCode) if initStackBottom(m): m.s[cfsProcs].addCallStmt("initStackBottomWith", cCast(CPointer, cAddr("inner"))) m.s[cfsProcs].addCallStmt(m.config.nimMainPrefix & "NimMainInner") m.s[cfsProcs].addNewline() proc genNimMainBody(m: BModule, preMainCode: Snippet) = genNimMainInner(m) genNimMainProc(m, preMainCode) proc genPosixCMain(m: BModule) = m.s[cfsProcs].addProcHeader("main", CInt, cProcParams( (name: "argc", typ: CInt), (name: "args", typ: ptrType(ptrType(CChar))), (name: "env", typ: ptrType(ptrType(CChar))))) m.s[cfsProcs].finishProcHeaderWithBody(): m.s[cfsProcs].addAssignment("cmdLine", "args") m.s[cfsProcs].addAssignment("cmdCount", "argc") m.s[cfsProcs].addAssignment("gEnv", "env") genMainProcsWithResult(m) m.s[cfsProcs].addNewline() proc genStandaloneCMain(m: BModule) = m.s[cfsProcs].addProcHeader("main", CInt, cProcParams()) m.s[cfsProcs].finishProcHeaderWithBody(): genMainProcs(m) m.s[cfsProcs].addReturn(cIntValue(0)) m.s[cfsProcs].addNewline() proc genWinNimMain(m: BModule, preMainCode: Snippet) = genNimMainBody(m, preMainCode) proc genWinCMain(m: BModule) = m.s[cfsProcs].addProcHeader(ccStdCall, "WinMain", CInt, cProcParams( (name: "hCurInstance", typ: "HINSTANCE"), (name: "hPrevInstance", typ: "HINSTANCE"), (name: "lpCmdLine", typ: "LPSTR"), (name: "nCmdShow", typ: CInt))) m.s[cfsProcs].finishProcHeaderWithBody(): genMainProcsWithResult(m) m.s[cfsProcs].addNewline() proc genWinNimDllMain(m: BModule, preMainCode: Snippet) = genNimMainInner(m) m.s[cfsProcs].addDeclWithVisibility(ExportLib): genNimMainProc(m, preMainCode) proc genWinCDllMain(m: BModule) = # used to use WINAPI macro, now ccStdCall: m.s[cfsProcs].addProcHeader(ccStdCall, "DllMain", "BOOL", cProcParams( (name: "hinstDLL", typ: "HINSTANCE"), (name: "fwdreason", typ: "DWORD"), (name: "lpvReserved", typ: "LPVOID"))) m.s[cfsProcs].finishProcHeaderWithBody(): m.s[cfsProcs].addSingleIfStmt(removeSinglePar(cOp(Equal, "fwdreason", "DLL_PROCESS_ATTACH"))): genMainProcs(m) m.s[cfsProcs].addReturn(cIntValue(1)) m.s[cfsProcs].addNewline() proc genPosixNimDllMain(m: BModule, preMainCode: Snippet) = genWinNimDllMain(m, preMainCode) proc genPosixCDllMain(m: BModule) = # used to use NIM_POSIX_INIT, now uses direct constructor attribute m.s[cfsProcs].addProcHeader("NimMainInit", CVoid, cProcParams(), isConstructor = true) m.s[cfsProcs].finishProcHeaderWithBody(): genMainProcs(m) m.s[cfsProcs].addNewline() proc genGenodeNimMain(m: BModule, preMainCode: Snippet) = let typName = "Genode::Env" m.s[cfsProcs].addDeclWithVisibility(Extern): m.s[cfsProcs].addVar(name = "nim_runtime_env", typ = ptrType(typName)) m.s[cfsProcs].addDeclWithVisibility(ExternC): m.s[cfsProcs].addProcHeader("nim_component_construct", CVoid, cProcParams((name: "", typ: ptrType(typName)))) m.s[cfsProcs].finishProcHeaderAsProto() genNimMainBody(m, preMainCode) proc genComponentConstruct(m: BModule) = let fnName = "Libc::Component::construct" let typName = "Libc::Env" m.s[cfsProcs].addProcHeader(fnName, CVoid, cProcParams((name: "env", typ: cppRefType(typName)))) m.s[cfsProcs].finishProcHeaderWithBody(): m.s[cfsProcs].addLineComment("Set Env used during runtime initialization") m.s[cfsProcs].addAssignment("nim_runtime_env", cAddr("env")) let callFn = "Libc::with_libc" var call: CallBuilder m.s[cfsProcs].addStmt(): m.s[cfsProcs].addCall(call, callFn): m.s[cfsProcs].addArgument(call): m.s[cfsProcs].addCppLambda(ByReference, cProcParams()): m.s[cfsProcs].addLineComment("Initialize runtime and globals") genMainProcs(m) m.s[cfsProcs].addLineComment("Call application construct") m.s[cfsProcs].addCallStmt("nim_component_construct", cAddr("env")) m.s[cfsProcs].addNewline() proc genMainProc(m: BModule) = ## this function is called in cgenWriteModules after all modules are closed, ## it means raising dependency on the symbols is too late as it will not propagate ## into other modules, only simple rope manipulations are allowed var preMainBuilder = newBuilder("") if m.hcrOn: proc loadLib(builder: var Builder, handle: string, name: string) = let prc = magicsys.getCompilerProc(m.g.graph, name) assert prc != nil let n = newStrNode(nkStrLit, prc.annex.path.strVal) n.info = prc.annex.path.info var strLitBuilder = newBuilder("") genStringLiteral(m, n, strLitBuilder) let strLit = extract(strLitBuilder) builder.addAssignment(handle, cCall(cgsymValue(m, "nimLoadLibrary"), strLit)) builder.addSingleIfStmt(cOp(Not, handle)): builder.addCallStmt(cgsymValue(m, "nimLoadLibraryError"), strLit) loadLib(preMainBuilder, "hcr_handle", "hcrGetProc") if m.config.selectedGC in {gcArc, gcAtomicArc, gcOrc}: preMainBuilder.addCallStmt(m.config.nimMainPrefix & "PreMain") else: preMainBuilder.addVar(name = "rtl_handle", typ = CPointer) loadLib(preMainBuilder, "rtl_handle", "nimGC_setStackBottom") hcrGetProcLoadCode(preMainBuilder, m, "nimGC_setStackBottom", "nimrtl_", "rtl_handle", "nimGetProcAddr") preMainBuilder.addAssignment("inner", m.config.nimMainPrefix & "PreMain") preMainBuilder.addCallStmt("initStackBottomWith_actual", cCast(CPointer, cAddr("inner"))) preMainBuilder.addCallStmt(cDeref("inner")) else: preMainBuilder.addCallStmt(m.config.nimMainPrefix & "PreMain") let preMainCode = extract(preMainBuilder) if m.config.target.targetOS == osWindows and m.config.globalOptions * {optGenGuiApp, optGenDynLib} != {}: m.includeHeader("") elif m.config.target.targetOS == osGenode: m.includeHeader("") if initStackBottom(m): cgsym(m, "initStackBottomWith") inc(m.labels) genPreMain(m) if m.config.target.targetOS == osWindows and m.config.globalOptions * {optGenGuiApp, optGenDynLib} != {}: if optGenGuiApp in m.config.globalOptions: genWinNimMain(m, preMainCode) else: genWinNimDllMain(m, preMainCode) elif m.config.target.targetOS == osGenode: genGenodeNimMain(m, preMainCode) elif optGenDynLib in m.config.globalOptions: genPosixNimDllMain(m, preMainCode) else: genNimMainBody(m, preMainCode) if optNoMain notin m.config.globalOptions: if m.config.cppCustomNamespace.len > 0: closeNamespaceNim(m.s[cfsProcs]) m.s[cfsProcs].add "using namespace " & m.config.cppCustomNamespace & ";\L" if m.config.target.targetOS == osWindows and m.config.globalOptions * {optGenGuiApp, optGenDynLib} != {}: if optGenGuiApp in m.config.globalOptions: genWinCMain(m) else: genWinCDllMain(m) elif m.config.target.targetOS == osGenode: genComponentConstruct(m) elif optGenDynLib in m.config.globalOptions: genPosixCDllMain(m) elif m.config.target.targetOS == osStandalone: genStandaloneCMain(m) else: genPosixCMain(m) if m.config.cppCustomNamespace.len > 0: openNamespaceNim(m.config.cppCustomNamespace, m.s[cfsProcs]) proc registerModuleToMain(g: BModuleList; m: BModule) = let init = m.getInitName datInit = m.getDatInitName if m.hcrOn: var hcrModuleMeta = newBuilder("") let systemModulePath = getModuleDllPath(m, g.mods[g.graph.config.m.systemFileIdx.int].module) let mainModulePath = getModuleDllPath(m, m.module) hcrModuleMeta.addDeclWithVisibility(Private): hcrModuleMeta.addArrayVarWithInitializer(kind = Local, name = "hcr_module_list", elementType = ptrConstType(CChar), len = g.graph.importDeps.getOrDefault(FileIndex(m.module.position)).len + ord(sfMainModule in m.module.flags) + 1): var modules: StructInitializer hcrModuleMeta.addStructInitializer(modules, siArray): if sfMainModule in m.module.flags: hcrModuleMeta.addField(modules, ""): hcrModuleMeta.add(systemModulePath) g.graph.importDeps.withValue(FileIndex(m.module.position), deps): for curr in deps[]: hcrModuleMeta.addField(modules, ""): hcrModuleMeta.add(getModuleDllPath(m, g.mods[curr.int].module)) hcrModuleMeta.addField(modules, ""): hcrModuleMeta.add("\"\"") hcrModuleMeta.addDeclWithVisibility(ExportLib): hcrModuleMeta.addProcHeader(ccNimCall, "HcrGetImportedModules", ptrType(CPointer), cProcParams()) hcrModuleMeta.finishProcHeaderWithBody(): hcrModuleMeta.addReturn(cCast(ptrType(CPointer), "hcr_module_list")) hcrModuleMeta.addDeclWithVisibility(ExportLib): hcrModuleMeta.addProcHeader(ccNimCall, "HcrGetSigHash", ptrType(CChar), cProcParams()) hcrModuleMeta.finishProcHeaderWithBody(): hcrModuleMeta.addReturn('"' & $sigHash(m.module, m.config) & '"') if sfMainModule in m.module.flags: g.mainModProcs.add(extract(hcrModuleMeta)) g.mainModProcs.addDeclWithVisibility(StaticProc): g.mainModProcs.addVar(name = "hcr_handle", typ = CPointer) g.mainModProcs.addDeclWithVisibility(ExportLib): g.mainModProcs.addProcHeader(ccNimCall, init, CVoid, cProcParams()) g.mainModProcs.finishProcHeaderAsProto() g.mainModProcs.addDeclWithVisibility(ExportLib): g.mainModProcs.addProcHeader(ccNimCall, datInit, CVoid, cProcParams()) g.mainModProcs.finishProcHeaderAsProto() g.mainModProcs.addDeclWithVisibility(ExportLib): g.mainModProcs.addProcHeaderWithParams(ccNimCall, m.getHcrInitName, CVoid): var hcrInitParams: ProcParamBuilder g.mainModProcs.addProcParams(hcrInitParams): g.mainModProcs.addUnnamedParam(hcrInitParams, CPointer) g.mainModProcs.addProcTypedParam(hcrInitParams, ccNimCall, "getProcAddr", CPointer, cProcParams( (name: "", typ: CPointer), (name: "", typ: ptrType(CChar)))) g.mainModProcs.finishProcHeaderAsProto() g.mainModProcs.addDeclWithVisibility(ExportLib): g.mainModProcs.addProcHeader(ccNimCall, "HcrCreateTypeInfos", CVoid, cProcParams()) g.mainModProcs.finishProcHeaderAsProto() g.mainModInit.addCallStmt(init) g.otherModsInit.addCallStmt("hcrInit", cCast(ptrType(CPointer), "hcr_module_list"), mainModulePath, systemModulePath, datInit, "hcr_handle", "nimGetProcAddr") g.mainDatInit.addCallStmt(m.getHcrInitName, "hcr_handle", "nimGetProcAddr") g.mainDatInit.addCallStmt("hcrAddModule", mainModulePath) g.mainDatInit.addCallStmt("HcrCreateTypeInfos") # nasty nasty hack to get the command line functionality working with HCR # register the 2 variables on behalf of the os module which might not even # be loaded (in which case it will get collected but that is not a problem) # EDIT: indeed, this hack, in combination with another un-necessary one # (`makeCString` was doing line wrap of string litterals) was root cause for # bug #16265. let osModulePath = ($systemModulePath).replace("stdlib_system", "stdlib_os").rope g.mainDatInit.addCallStmt("hcrAddModule", osModulePath) let cmdCountTyp = ptrType(CInt) let cmdLineTyp = ptrType(ptrType(ptrType(CChar))) g.mainDatInit.addVar(name = "cmd_count", typ = cmdCountTyp) g.mainDatInit.addVar(name = "cmd_line", typ = cmdLineTyp) g.mainDatInit.addCallStmt("hcrRegisterGlobal", osModulePath, "\"cmdCount\"", cSizeof(cmdCountTyp), CNil, cCast(ptrType(CPointer), cAddr("cmd_count"))) g.mainDatInit.addCallStmt("hcrRegisterGlobal", osModulePath, "\"cmdLine\"", cSizeof(cmdLineTyp), CNil, cCast(ptrType(CPointer), cAddr("cmd_line"))) g.mainDatInit.addAssignment(cDeref("cmd_count"), "cmdCount") g.mainDatInit.addAssignment(cDeref("cmd_line"), "cmdLine") else: m.s[cfsInitProc].add(extract(hcrModuleMeta)) return if m.s[cfsDatInitProc].buf.len > 0: g.mainModProcs.addDeclWithVisibility(Private): g.mainModProcs.addProcHeader(ccNimCall, datInit, CVoid, cProcParams()) g.mainModProcs.finishProcHeaderAsProto() g.mainDatInit.addCallStmt(datInit) # Initialization of TLS and GC should be done in between # systemDatInit and systemInit calls if any if sfSystemModule in m.module.flags: if emulatedThreadVars(m.config) and m.config.target.targetOS != osStandalone: g.mainDatInit.addCallStmt(cgsymValue(m, "initThreadVarsEmulation")) if m.config.target.targetOS != osStandalone and m.config.selectedGC notin {gcNone, gcArc, gcAtomicArc, gcOrc}: g.mainDatInit.addCallStmt(cgsymValue(m, "initStackBottomWith"), cCast(CPointer, cAddr("inner"))) if m.s[cfsInitProc].buf.len > 0: g.mainModProcs.addDeclWithVisibility(Private): g.mainModProcs.addProcHeader(ccNimCall, init, CVoid, cProcParams()) g.mainModProcs.finishProcHeaderAsProto() if sfMainModule in m.module.flags: g.mainModInit.addCallStmt(init) elif sfSystemModule in m.module.flags: g.mainDatInit.addCallStmt(init) # systemInit must called right after systemDatInit if any else: g.otherModsInit.addCallStmt(init) proc genDatInitCode(m: BModule) = ## this function is called in cgenWriteModules after all modules are closed, ## it means raising dependency on the symbols is too late as it will not propagate ## into other modules, only simple rope manipulations are allowed var moduleDatInitRequired = m.hcrOn var prc = newBuilder("") let vis = if m.hcrOn: ExportLib else: Private prc.addDeclWithVisibility(vis): prc.addProcHeader(ccNimCall, getDatInitName(m), CVoid, cProcParams()) prc.finishProcHeaderWithBody(): # we don't want to break into such init code - could happen if a line # directive from a function written by the user spills after itself genCLineDir(prc, InvalidFileIdx, 999999, m.config) for i in cfsTypeInit1..cfsDynLibInit: if m.s[i].buf.len != 0: moduleDatInitRequired = true prc.add(extract(m.s[i])) prc.addNewline() if moduleDatInitRequired: m.s[cfsDatInitProc].add(extract(prc)) #rememberFlag(m.g.graph, m.module, HasDatInitProc) # Very similar to the contents of symInDynamicLib - basically only the # things needed for the hot code reloading runtime procs to be loaded proc hcrGetProcLoadCode(builder: var Builder, m: BModule, sym, prefix, handle, getProcFunc: string) = let prc = magicsys.getCompilerProc(m.g.graph, sym) assert prc != nil fillProcLoc(m, prc.ast[namePos]) var extname = prefix & sym var tmp = mangleDynLibProc(prc) backendEnsureMutable prc prc.locImpl.snippet = tmp prc.typ.sym = nil if not containsOrIncl(m.declaredThings, prc.id): m.s[cfsVars].addVar(Global, name = prc.loc.snippet, typ = getTypeDesc(m, prc.loc.t, dkVar)) builder.addAssignment(tmp, cCast(getTypeDesc(m, prc.typ, dkVar), cCall(getProcFunc, handle, makeCString(prefix & sym)))) proc genInitCode(m: BModule) = ## this function is called in cgenWriteModules after all modules are closed, ## it means raising dependency on the symbols is too late as it will not propagate ## into other modules, only simple rope manipulations are allowed var moduleInitRequired = m.hcrOn let initname = getInitName(m) var prcBody = newBuilder("") # we don't want to break into such init code - could happen if a line # directive from a function written by the user spills after itself genCLineDir(prcBody, InvalidFileIdx, 999999, m.config) if m.typeNodes > 0: if m.hcrOn: m.s[cfsTypeInit1].addVar(name = m.typeNodesName, typ = ptrType(cgsymValue(m, "TNimNode"))) m.s[cfsTypeInit1].addCallStmt("hcrRegisterGlobal", getModuleDllPath(m, m.module), '"' & m.typeNodesName & '_' & $m.typeNodes & '"', cOp(Mul, NimInt, cSizeof("TNimNode"), cIntValue(m.typeNodes)), CNil, cCast(ptrType(CPointer), cAddr(m.typeNodesName))) else: m.s[cfsTypeInit1].addArrayVar(Global, name = m.typeNodesName, elementType = cgsymValue(m, "TNimNode"), len = m.typeNodes) if m.nimTypes > 0: m.s[cfsTypeInit1].addArrayVar(Global, name = m.nimTypesName, elementType = cgsymValue(m, "TNimType"), len = m.nimTypes) if m.hcrOn: prcBody.addVar(name = "nim_hcr_dummy_", typ = ptrType(CInt), initializer = cIntValue(0)) prcBody.addVar(name = "nim_hcr_do_init_", typ = NimBool, initializer = cCall("hcrRegisterGlobal", getModuleDllPath(m, m.module), "\"module_initialized_\"", cIntValue(1), CNil, cCast(ptrType(CPointer), cAddr("nim_hcr_dummy_")))) template writeSection(thing: untyped, section: TCProcSection, addHcrGuards = false) = if m.thing.s(section).buf.len > 0: moduleInitRequired = true if addHcrGuards: prcBody.addSingleIfStmt("nim_hcr_do_init_"): prcBody.addNewline() prcBody.add(extract(m.thing.s(section))) prcBody.addNewline() else: prcBody.add(extract(m.thing.s(section))) #echo "PRE INIT PROC ", m.module.name.s, " ", m.s[cfsVars].buf.len if m.preInitProc.s(cpsInit).buf.len > 0 or m.preInitProc.s(cpsStmts).buf.len > 0: # Give this small function its own scope prcBody.addScope(): # Keep a bogus frame in case the code needs one prcBody.addVar(name = "FR_", typ = "TFrame") prcBody.addFieldAssignment("FR_", "len", cIntValue(0)) writeSection(preInitProc, cpsLocals) writeSection(preInitProc, cpsInit, m.hcrOn) writeSection(preInitProc, cpsStmts) when false: m.initProc.blocks[0].sections[cpsLocals].add m.preInitProc.s(cpsLocals) m.initProc.blocks[0].sections[cpsInit].prepend m.preInitProc.s(cpsInit) m.initProc.blocks[0].sections[cpsStmts].prepend m.preInitProc.s(cpsStmts) # add new scope for following code, because old vcc compiler need variable # be defined at the top of the block prcBody.addScope(): writeSection(initProc, cpsLocals) if m.initProc.s(cpsInit).buf.len > 0 or m.initProc.s(cpsStmts).buf.len > 0: moduleInitRequired = true if optStackTrace in m.initProc.options and frameDeclared notin m.flags: # BUT: the generated init code might depend on a current frame, so # declare it nevertheless: incl m.flags, frameDeclared if preventStackTrace notin m.flags: var procname = makeCString(m.module.name.s) prcBody.add(initFrame(m.initProc, procname, quotedFilename(m.config, m.module.info))) else: prcBody.addVar(name = "FR_", typ = "TFrame") prcBody.addFieldAssignment("FR_", "len", cIntValue(0)) writeSection(initProc, cpsInit, m.hcrOn) writeSection(initProc, cpsStmts) if beforeRetNeeded in m.initProc.flags: prcBody.addLabel("BeforeRet_") if m.config.exc == excGoto: if getCompilerProc(m.g.graph, "nimTestErrorFlag") != nil: prcBody.addCallStmt(cgsymValue(m, "nimTestErrorFlag")) if optStackTrace in m.initProc.options and preventStackTrace notin m.flags: prcBody.add(deinitFrame(m.initProc)) var procs = newBuilder("") let vis = if m.hcrOn: ExportLib else: Private procs.addDeclWithVisibility(vis): procs.addProcHeader(ccNimCall, initname, CVoid, cProcParams()) procs.finishProcHeaderWithBody(): procs.add(extract(prcBody)) # we cannot simply add the init proc to ``m.s[cfsProcs]`` anymore because # that would lead to a *nesting* of merge sections which the merger does # not support. So we add it to another special section: ``cfsInitProc`` if m.hcrOn: var procsToLoad = @["hcrRegisterProc", "hcrGetProc", "hcrRegisterGlobal", "hcrGetGlobal"] m.s[cfsInitProc].addDeclWithVisibility(ExportLib): m.s[cfsInitProc].addProcHeaderWithParams(ccNimCall, getHcrInitName(m), CVoid): var hcrInitParams: ProcParamBuilder m.s[cfsInitProc].addProcParams(hcrInitParams): m.s[cfsInitProc].addParam(hcrInitParams, "handle", CPointer) m.s[cfsInitProc].addProcTypedParam(hcrInitParams, ccNimCall, "getProcAddr", CPointer, cProcParams( (name: "", typ: CPointer), (name: "", typ: ptrType(CChar)))) m.s[cfsInitProc].finishProcHeaderWithBody(): if sfMainModule in m.module.flags: # additional procs to load procsToLoad.add("hcrInit") procsToLoad.add("hcrAddModule") # load procs for curr in procsToLoad: hcrGetProcLoadCode(m.s[cfsInitProc], m, curr, "", "handle", "getProcAddr") for i, el in pairs(m.extensionLoaders): if el.buf.len != 0: moduleInitRequired = true procs.addDeclWithVisibility(ExternC): procs.addProcHeader(ccNimCall, "nimLoadProcs" & $(i.ord - '0'.ord), CVoid, cProcParams()) procs.finishProcHeaderWithBody(): procs.add(extract(el)) if moduleInitRequired or sfMainModule in m.module.flags: m.s[cfsInitProc].add(extract(procs)) #rememberFlag(m.g.graph, m.module, HasModuleInitProc) genDatInitCode(m) if m.hcrOn: m.s[cfsInitProc].addDeclWithVisibility(ExportLib): m.s[cfsInitProc].addProcHeader(ccNimCall, "HcrCreateTypeInfos", CVoid, cProcParams()) m.s[cfsInitProc].finishProcHeaderWithBody(): m.s[cfsInitProc].add(extract(m.hcrCreateTypeInfosProc)) m.s[cfsInitProc].addNewline() registerModuleToMain(m.g, m) proc postprocessCode(conf: ConfigRef, r: var Rope) = # find the first directive var f = r.find(postprocessDirStart) if f == -1: return var nimlnDirLastF = "" var res: Rope = r.substr(0, f - 1) while f != -1: var e = r.find(postprocessDirEnd, f + 1) dir = r.substr(f + 1, e - 1).split(postprocessDirSep) case dir[0] of "nimln": if dir[2] == nimlnDirLastF: res.add("nimln_(" & dir[1] & ");") else: res.add("nimlf_(" & dir[1] & ", " & quotedFilename(conf, dir[2].parseInt.FileIndex) & ");") nimlnDirLastF = dir[2] else: raiseAssert "unexpected postprocess directive" # find the next directive f = r.find(postprocessDirStart, e + 1) # copy the code until the next directive if f != -1: res.add(r.substr(e + 1, f - 1)) else: res.add(r.substr(e + 1)) r = res proc genModule(m: BModule, cfile: Cfile): Rope = var moduleIsEmpty = true var res = newBuilder(getFileHeader(m.config, cfile)) generateThreadLocalStorage(m) generateHeaders(m) res.add(extract(m.s[cfsHeaders])) if m.config.cppCustomNamespace.len > 0: openNamespaceNim(m.config.cppCustomNamespace, res) if m.s[cfsFrameDefines].buf.len > 0: res.add(extract(m.s[cfsFrameDefines])) for i in cfsForwardTypes..cfsProcs: if m.s[i].buf.len > 0: moduleIsEmpty = false res.add(extract(m.s[i])) if m.s[cfsInitProc].buf.len > 0: moduleIsEmpty = false res.add(extract(m.s[cfsInitProc])) if m.s[cfsDatInitProc].buf.len > 0 or m.hcrOn: moduleIsEmpty = false res.add(extract(m.s[cfsDatInitProc])) if m.config.cppCustomNamespace.len > 0: closeNamespaceNim(res) result = extract(res) if optLineDir in m.config.options: var srcFileDefs = "" for fi in 0..m.config.m.fileInfos.high: srcFileDefs.add("#define FX_" & $fi & " " & makeSingleLineCString(toFullPath(m.config, fi.FileIndex)) & "\n") result = srcFileDefs & result if moduleIsEmpty: result = "" postprocessCode(m.config, result) proc initProcOptions(m: BModule): TOptions = let opts = m.config.options if sfSystemModule in m.module.flags: opts-{optStackTrace} else: opts proc rawNewModule(g: BModuleList; module: PSym, filename: AbsoluteFile): BModule = new(result) result.g = g result.tmpBase = rope("TM" & $hashOwner(module) & "_") result.headerFiles = @[] result.declaredThings = initIntSet() result.declaredProtos = initIntSet() result.cfilename = filename result.filename = filename result.typeCache = initTable[SigHash, Rope]() result.forwTypeCache = initTable[SigHash, Rope]() result.module = module result.typeInfoMarker = initTable[SigHash, Rope]() result.sigConflicts = initCountTable[SigHash]() result.initProc = newProc(nil, result) for i in low(result.s)..high(result.s): result.s[i] = newBuilder("") result.initProc.options = initProcOptions(result) result.preInitProc = newProc(nil, result) result.preInitProc.flags.incl nimErrorFlagDisabled result.preInitProc.labels = 100_000 # little hack so that unique temporaries are generated result.hcrCreateTypeInfosProc = newBuilder("") result.dataCache = initNodeTable() result.typeStack = @[] result.typeNodesName = getTempName(result) result.nimTypesName = getTempName(result) # no line tracing for the init sections of the system module so that we # don't generate a TFrame which can confuse the stack bottom initialization: if sfSystemModule in module.flags: incl result.flags, preventStackTrace excl(result.preInitProc.options, optStackTrace) proc rawNewModule(g: BModuleList; module: PSym; conf: ConfigRef): BModule = result = rawNewModule(g, module, AbsoluteFile toFullPath(conf, module.position.FileIndex)) proc newModule(g: BModuleList; module: PSym; conf: ConfigRef; idgen: IdGenerator): BModule = # we should create only one cgen module for each module sym result = rawNewModule(g, module, conf) result.idgen = idgen if module.position >= g.mods.len: setLen(g.mods, module.position + 1) #growCache g.modules, module.position g.mods[module.position] = result template injectG() {.dirty.} = if graph.backend == nil: graph.backend = newModuleList(graph) let g = BModuleList(graph.backend) proc setupCgen*(graph: ModuleGraph; module: PSym; idgen: IdGenerator): PPassContext = injectG() result = newModule(g, module, graph.config, idgen) if optGenIndex in graph.config.globalOptions and g.generatedHeader == nil: let f = if graph.config.headerFile.len > 0: AbsoluteFile graph.config.headerFile else: graph.config.projectFull g.generatedHeader = rawNewModule(g, module, changeFileExt(completeCfilePath(graph.config, f), hExt)) incl g.generatedHeader.flags, isHeaderFile proc writeHeader(m: BModule) = var result = newBuilder(headerTop()) var guard = "__$1__" % [m.filename.splitFile.name.rope] result.addf("#ifndef $1$n#define $1$n", [guard]) addNimDefines(result, m.config) generateHeaders(m) generateThreadLocalStorage(m) for i in cfsHeaders..cfsProcs: result.add(extract(m.s[i])) if m.config.cppCustomNamespace.len > 0 and i == cfsHeaders: openNamespaceNim(m.config.cppCustomNamespace, result) result.add(extract(m.s[cfsInitProc])) let vis = if optGenDynLib in m.config.globalOptions: ImportLib else: None result.addDeclWithVisibility(vis): result.addProcHeader(ccCDecl, m.config.nimMainPrefix & "NimMain", CVoid, cProcParams()) result.finishProcHeaderAsProto() if m.config.cppCustomNamespace.len > 0: closeNamespaceNim(result) result.addf("#endif /* $1 */$n", [guard]) if not writeRope(extract(result), m.filename): rawMessage(m.config, errCannotOpenFile, m.filename.string) proc getCFile(m: BModule): AbsoluteFile = let ext = if m.compileToCpp: ".nim.cpp" elif m.config.backend == backendObjc or sfCompileToObjc in m.module.flags: ".nim.m" else: ".nim.c" result = changeFileExt(completeCfilePath(m.config, mangleModuleName(m.config, m.cfilename).AbsoluteFile), ext) when false: proc myOpenCached(graph: ModuleGraph; module: PSym, rd: PRodReader): PPassContext = injectG() var m = newModule(g, module, graph.config) readMergeInfo(getCFile(m), m) result = m proc addHcrInitGuards(p: BProc, n: PNode, inInitGuard: var bool, init: var IfBuilder) = if n.kind == nkStmtList: for child in n: addHcrInitGuards(p, child, inInitGuard, init) else: let stmtShouldExecute = n.kind in {nkVarSection, nkLetSection} or nfExecuteOnReload in n.flags if inInitGuard: if stmtShouldExecute: endBlockWith(p): finishBranch(p.s(cpsStmts), init) finishIfStmt(p.s(cpsStmts), init) inInitGuard = false else: if not stmtShouldExecute: startBlockWith(p): init = initIfStmt(p.s(cpsStmts)) initElifBranch(p.s(cpsStmts), init, "nim_hcr_do_init_") inInitGuard = true genStmts(p, n) proc handleProcGlobals(m: BModule) = var procGlobals: seq[PNode] = move m.g.graph.procGlobals for i in 0.. 0: let sym = m.queue.pop() genProcLvl2(m, sym) finishTypeDescriptions(m) if sfMainModule in m.module.flags: # generate main file: genMainProc(m) m.s[cfsProcHeaders].add(extract(m.g.mainModProcs)) generateThreadVarsSize(m) var cf = Cfile(nimname: m.module.name.s, cname: cfile, obj: completeCfilePath(m.config, toObjFile(m.config, cfile)), flags: {}) var code = genModule(m, cf) if code != "" or m.config.symbolFiles != disabledSf: when hasTinyCBackend: if m.config.cmd == cmdTcc: tccgen.compileCCode($code, m.config) return if not shouldRecompile(m, code, cf): cf.flags = {CfileFlag.Cached} addFileToCompile(m.config, cf) proc updateCachedModule(m: BModule) = let cfile = getCFile(m) var cf = Cfile(nimname: m.module.name.s, cname: cfile, obj: completeCfilePath(m.config, toObjFile(m.config, cfile)), flags: {}) if sfMainModule notin m.module.flags: genMainProc(m) cf.flags = {CfileFlag.Cached} addFileToCompile(m.config, cf) proc generateLibraryDestroyGlobals(graph: ModuleGraph; m: BModule; body: PNode; isDynlib: bool): PSym = let prefixedName = m.config.nimMainPrefix & "NimDestroyGlobals" let procname = getIdent(graph.cache, prefixedName) result = newSym(skProc, procname, m.idgen, m.module.owner, m.module.info) result.typ = newProcType(m.module.info, m.idgen, m.module.owner) result.typ.callConv = ccCDecl backendEnsureMutable result incl result.flagsImpl, sfExportc result.locImpl.snippet = prefixedName if isDynlib: incl(result.locImpl.flags, lfExportLib) let theProc = newNodeI(nkProcDef, m.module.info, bodyPos+1) for i in 0.. 0: let prc = g.forwardedProcs.pop() m = g.mods[prc.itemId.module] if sfForward in prc.flags: internalError(m.config, prc.info, "still forwarded: " & prc.name.s) genProcLvl2(m, prc) proc cgenWriteModules*(backend: RootRef, config: ConfigRef) = let g = BModuleList(backend) g.config = config # we need to process the transitive closure because recursive module # deps are allowed (and the system module is processed in the wrong # order anyway) genForwardedProcs(g) for m in cgenModules(g): m.writeModule() writeMapping(config, g.mapping) if g.generatedHeader != nil: writeHeader(g.generatedHeader)