BeIRCodeGen.cpp 94 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793
  1. #include "BeIRCodeGen.h"
  2. #include "../Compiler/BfIRCodeGen.h"
  3. #include "BeDbgModule.h"
  4. #include "BeefySysLib/util/BeefPerf.h"
  5. #include "llvm/IR/DIBuilder.h"
  6. #include "BeefySysLib/util/AllocDebug.h"
  7. #include "BeefySysLib/util/Hash.h"
  8. #include "BeModule.h"
  9. #include "BeContext.h"
  10. #include "../Compiler/CeMachine.h"
  11. #ifdef _DEBUG
  12. #define BE_EXTRA_CHECKS
  13. #endif
  14. USING_NS_BF;
  15. //#define CODEGEN_TRACK
  16. #ifdef CODEGEN_TRACK
  17. #include "../Compiler/MemReporter.h"
  18. MemReporter gBEMemReporter;
  19. CritSect gBEMemReporterCritSect;
  20. int gBEMemReporterSize = 0;
  21. #define BE_MEM_START \
  22. int startPos = mStream->GetReadPos();
  23. #define BE_MEM_END(name) \
  24. gBEMemReporter.Add(name, mStream->GetReadPos() - startPos);
  25. static const char* gIRCmdNames[] =
  26. {
  27. "Module_Start",
  28. "Module_SetTargetTriple",
  29. "Module_AddModuleFlag",
  30. "WriteIR",
  31. "SetType",
  32. "SetInstType",
  33. "PrimitiveType",
  34. "CreateStruct",
  35. "StructSetBody",
  36. "Type",
  37. "TypeInst",
  38. "TypeInstPtr",
  39. "GetType",
  40. "GetPointerToFuncType",
  41. "GetPointerToType",
  42. "GetSizedArrayType",
  43. "CreateConstStruct",
  44. "CreateConstStructZero",
  45. "CreateConstArray",
  46. "CreateConstString",
  47. "SetName",
  48. "CreateUndefValue",
  49. "NumericCast",
  50. "CmpEQ",
  51. "CmpNE",
  52. "CmpSLT",
  53. "CmpULT",
  54. "CmpSLE",
  55. "CmpULE",
  56. "CmpSGT",
  57. "CmpUGT",
  58. "CmpSGE",
  59. "CmpUGE",
  60. "Add",
  61. "Sub",
  62. "Mul",
  63. "SDiv",
  64. "UDiv",
  65. "SRem",
  66. "URem",
  67. "And",
  68. "Or",
  69. "Xor",
  70. "Shl",
  71. "AShr",
  72. "LShr",
  73. "Neg",
  74. "Not",
  75. "BitCast",
  76. "PtrToInt",
  77. "IntToPtr",
  78. "InboundsGEP1_32",
  79. "InboundsGEP2_32",
  80. "InBoundsGEP1",
  81. "InBoundsGEP2",
  82. "IsNull",
  83. "IsNotNull",
  84. "ExtractValue",
  85. "InsertValue",
  86. "Alloca",
  87. "AllocaArray",
  88. "SetAllocaAlignment",
  89. "SetAllocaNoChkStkHint",
  90. "SetAllocaForceMem",
  91. "LifetimeStart",
  92. "LifetimeEnd",
  93. "LifetimeExtend",
  94. "ValueScopeStart",
  95. "ValueScopeRetain",
  96. "ValueScopeSoftEnd",
  97. "ValueScopeHardEnd",
  98. "Load",
  99. "AlignedLoad",
  100. "Store",
  101. "AlignedStore",
  102. "MemSet",
  103. "Fence",
  104. "StackSave",
  105. "StackRestore",
  106. "GlobalVariable",
  107. "GlobalVar_SetUnnamedAddr",
  108. "GlobalVar_SetInitializer",
  109. "GlobalVar_SetAlignment",
  110. "GlobalStringPtr",
  111. "CreateBlock",
  112. "MaybeChainNewBlock",
  113. "AddBlock",
  114. "DropBlocks",
  115. "MergeBlockDown",
  116. "SetInsertPoint",
  117. "SetInsertPointAtStart",
  118. "EraseFromParent",
  119. "DeleteBlock",
  120. "EraseInstFromParent",
  121. "CreateBr",
  122. "CreateBr_NoCollapse",
  123. "CreateCondBr",
  124. "MoveBlockToEnd",
  125. "CreateSwitch",
  126. "AddSwitchCase",
  127. "SetSwitchDefaultDest",
  128. "CreatePhi",
  129. "AddPhiIncoming",
  130. "GetIntrinsic",
  131. "CreateFunctionType",
  132. "CreateFunction",
  133. "EnsureFunctionPatchable",
  134. "RemapBindFunction",
  135. "SetActiveFunction",
  136. "CreateCall",
  137. "SetCallCallingConv",
  138. "SetFuncCallingConv",
  139. "SetTailCall",
  140. "SetCallAttribute",
  141. "CreateRet",
  142. "CreateRetVoid",
  143. "CreateUnreachable",
  144. "Call_AddAttribute",
  145. "Call_AddAttribute1",
  146. "Func_AddAttribute",
  147. "Func_AddAttribute1",
  148. "Func_SetParamName",
  149. "Func_DeleteBody",
  150. "Func_EraseFromParent",
  151. "Func_SetLinkage",
  152. "SaveDebugLocation",
  153. "RestoreDebugLocation",
  154. "ClearDebugLocation",
  155. "ClearDebugLocationInst",
  156. "UpdateDebugLocation",
  157. "SetCurrentDebugLocation",
  158. "Nop",
  159. "EnsureInstructionAt",
  160. "StatementStart",
  161. "ObjectAccessCheck",
  162. "DbgInit",
  163. "DbgFinalize",
  164. "DbgCreateCompileUnit",
  165. "DbgCreateFile",
  166. "ConstValueI64",
  167. "DbgGetCurrentLocation",
  168. "DbgSetType",
  169. "DbgSetInstType",
  170. "DbgGetType",
  171. "DbgGetTypeInst",
  172. "DbgTrackDITypes",
  173. "DbgCreateNamespace",
  174. "DbgCreateImportedModule",
  175. "DbgCreateBasicType",
  176. "DbgCreateStructType",
  177. "DbgCreateEnumerationType",
  178. "DbgCreatePointerType",
  179. "DbgCreateReferenceType",
  180. "DbgCreateConstType",
  181. "DbgCreateArtificialType",
  182. "DbgCreateArrayType",
  183. "DbgCreateReplaceableCompositeType",
  184. "DbgCreateForwardDecl",
  185. "DbgCreateSizedForwardDecl",
  186. "BeIRCmd_DbgSetTypeSize",
  187. "DbgReplaceAllUses",
  188. "DbgDeleteTemporary",
  189. "DbgMakePermanent",
  190. "CreateEnumerator",
  191. "DbgCreateMemberType",
  192. "DbgStaticCreateMemberType",
  193. "DbgCreateInheritance",
  194. "DbgCreateMethod",
  195. "DbgCreateFunction",
  196. "DbgCreateParameterVariable",
  197. "DbgCreateSubroutineType",
  198. "DbgCreateAutoVariable",
  199. "DbgInsertValueIntrinsic",
  200. "DbgInsertDeclare",
  201. "DbgLifetimeEnd",
  202. "DbgCreateGlobalVariable",
  203. "DbgCreateLexicalBlock",
  204. "DbgCreateLexicalBlockFile",
  205. "DbgCreateAnnotation"
  206. };
  207. BF_STATIC_ASSERT(BF_ARRAY_COUNT(gIRCmdNames) == BfIRCmd_COUNT);
  208. #else
  209. #define BE_MEM_START
  210. #define BE_MEM_END(name)
  211. #endif
  212. #pragma warning(disable:4146)
  213. #define CMD_PARAM(ty, name) ty name; Read(name);
  214. BeIRCodeGen::BeIRCodeGen()
  215. {
  216. mBfIRBuilder = NULL;
  217. mStream = NULL;
  218. mActiveFunction = NULL;
  219. mBeContext = NULL;
  220. mBeModule = NULL;
  221. mHasDebugLoc = false;
  222. mDebugging = false;
  223. mCmdCount = 0;
  224. }
  225. BeIRCodeGen::~BeIRCodeGen()
  226. {
  227. BF_ASSERT(mSavedDebugLocs.size() == 0);
  228. delete mBeModule;
  229. delete mBeContext;
  230. delete mStream;
  231. }
  232. void BeIRCodeGen::Hash(BeHashContext& hashCtx)
  233. {
  234. // if (mBeModule->mModuleName == "IDE_IDEApp")
  235. // {
  236. // hashCtx.mDbgViz = true;
  237. // NOP;
  238. // }
  239. hashCtx.Mixin(mPtrSize);
  240. hashCtx.Mixin(mIsOptimized);
  241. if (mBeModule != NULL)
  242. mBeModule->Hash(hashCtx);
  243. Array<BeStructType*> structHashList;
  244. for (auto beType : mBeContext->mTypes)
  245. {
  246. if (!beType->IsStruct())
  247. continue;
  248. auto beStructType = (BeStructType*)beType;
  249. if (beStructType->mHashId != -1)
  250. continue;
  251. structHashList.Add(beStructType);
  252. }
  253. structHashList.Sort([](BeStructType* lhs, BeStructType* rhs)
  254. {
  255. return lhs->mName < rhs->mName;
  256. });
  257. for (auto beStructType : structHashList)
  258. {
  259. beStructType->HashReference(hashCtx);
  260. }
  261. }
  262. bool BeIRCodeGen::IsModuleEmpty()
  263. {
  264. if (!mBeModule->mFunctions.IsEmpty())
  265. return false;
  266. if (!mBeModule->mGlobalVariables.IsEmpty())
  267. return false;
  268. return true;
  269. }
  270. void BeIRCodeGen::FatalError(const StringImpl& err)
  271. {
  272. String failStr = "Fatal Error in Module: ";
  273. if (mBeModule != NULL)
  274. failStr += mBeModule->mModuleName;
  275. failStr += "\n";
  276. if (mBeModule != NULL)
  277. {
  278. BeDumpContext dumpCtx;
  279. if (mActiveFunction != NULL)
  280. {
  281. failStr += "Function: ";
  282. failStr += mActiveFunction->mName;
  283. failStr += "\n";
  284. }
  285. if (mBeModule->mCurDbgLoc != NULL)
  286. {
  287. failStr += "DbgLoc: ";
  288. dumpCtx.ToString(failStr, mBeModule->mCurDbgLoc);
  289. failStr += "\n";
  290. }
  291. }
  292. failStr += err;
  293. BF_FATAL(failStr);
  294. }
  295. void BeIRCodeGen::NotImpl()
  296. {
  297. BF_FATAL("Not implemented");
  298. }
  299. BeType* BeIRCodeGen::GetBeType(BfTypeCode typeCode, bool& isSigned)
  300. {
  301. isSigned = false;
  302. BeTypeCode beTypeCode = BeTypeCode_None;
  303. switch (typeCode)
  304. {
  305. case BfTypeCode_None:
  306. beTypeCode = BeTypeCode_None;
  307. break;
  308. case BfTypeCode_NullPtr:
  309. beTypeCode = BeTypeCode_NullPtr;
  310. break;
  311. case BfTypeCode_Boolean:
  312. beTypeCode = BeTypeCode_Boolean;
  313. break;
  314. case BfTypeCode_Int8:
  315. isSigned = true;
  316. beTypeCode = BeTypeCode_Int8;
  317. break;
  318. case BfTypeCode_UInt8:
  319. case BfTypeCode_Char8:
  320. beTypeCode = BeTypeCode_Int8;
  321. break;
  322. case BfTypeCode_Int16:
  323. isSigned = true;
  324. beTypeCode = BeTypeCode_Int16;
  325. break;
  326. case BfTypeCode_Char16:
  327. case BfTypeCode_UInt16:
  328. beTypeCode = BeTypeCode_Int16;
  329. break;
  330. case BfTypeCode_Int32:
  331. isSigned = true;
  332. beTypeCode = BeTypeCode_Int32;
  333. break;
  334. case BfTypeCode_UInt32:
  335. case BfTypeCode_Char32:
  336. beTypeCode = BeTypeCode_Int32;
  337. break;
  338. case BfTypeCode_Int64:
  339. isSigned = true;
  340. beTypeCode = BeTypeCode_Int64;
  341. break;
  342. case BfTypeCode_UInt64:
  343. beTypeCode = BeTypeCode_Int64;
  344. break;
  345. case BfTypeCode_IntPtr:
  346. BF_FATAL("Illegal");
  347. /*isSigned = true;
  348. if (mModule->mSystem->mPtrSize == 4)
  349. return llvm::Type::getInt32Ty(*mLLVMContext);
  350. else
  351. return llvm::Type::getInt64Ty(*mLLVMContext);*/
  352. case BfTypeCode_UIntPtr:
  353. BF_FATAL("Illegal");
  354. /*if (mModule->mSystem->mPtrSize == 4)
  355. return llvm::Type::getInt32Ty(*mLLVMContext);
  356. else
  357. return llvm::Type::getInt64Ty(*mLLVMContext);*/
  358. case BfTypeCode_Float:
  359. isSigned = true;
  360. beTypeCode = BeTypeCode_Float;
  361. break;
  362. case BfTypeCode_Double:
  363. isSigned = true;
  364. beTypeCode = BeTypeCode_Double;
  365. break;
  366. }
  367. return mBeContext->GetPrimitiveType(beTypeCode);
  368. }
  369. BeIRTypeEntry& BeIRCodeGen::GetTypeEntry(int typeId)
  370. {
  371. BeIRTypeEntry& typeEntry = mTypes[typeId];
  372. if (typeEntry.mTypeId == -1)
  373. typeEntry.mTypeId = typeId;
  374. return typeEntry;
  375. }
  376. void BeIRCodeGen::FixValues(BeStructType* structType, CmdParamVec<BeValue*>& values)
  377. {
  378. if (values.size() >= structType->mMembers.size())
  379. return;
  380. int readIdx = values.size() - 1;
  381. values.resize(structType->mMembers.size());
  382. for (int i = (int)values.size() - 1; i >= 0; i--)
  383. {
  384. if (mBeContext->AreTypesEqual(values[readIdx]->GetType(), structType->mMembers[i].mType))
  385. {
  386. values[i] = values[readIdx];
  387. readIdx--;
  388. }
  389. else if (structType->mMembers[i].mType->IsSizedArray())
  390. {
  391. auto beConst = mBeModule->mAlloc.Alloc<BeConstant>();
  392. beConst->mType = structType->mMembers[i].mType;
  393. values[i] = beConst;
  394. }
  395. else
  396. {
  397. FatalError("Malformed structure values");
  398. }
  399. }
  400. }
  401. void BeIRCodeGen::FixValues(BeStructType* structType, SizedArrayImpl<BeConstant*>& values)
  402. {
  403. if (values.size() >= structType->mMembers.size())
  404. return;
  405. int readIdx = values.size() - 1;
  406. values.resize(structType->mMembers.size());
  407. for (int i = (int)values.size() - 1; i >= 0; i--)
  408. {
  409. if (mBeContext->AreTypesEqual(values[readIdx]->GetType(), structType->mMembers[i].mType))
  410. {
  411. values[i] = values[readIdx];
  412. readIdx--;
  413. }
  414. else if (structType->mMembers[i].mType->IsSizedArray())
  415. {
  416. auto beConst = mBeModule->mAlloc.Alloc<BeConstant>();
  417. beConst->mType = structType->mMembers[i].mType;
  418. values[i] = beConst;
  419. }
  420. else
  421. {
  422. FatalError("Malformed structure values");
  423. }
  424. }
  425. }
  426. void BeIRCodeGen::Init(const BfSizedArray<uint8>& buffer)
  427. {
  428. BP_ZONE("BeIRCodeGen::Init");
  429. BF_ASSERT(mStream == NULL);
  430. mStream = new ChunkedDataBuffer();
  431. mStream->InitFlatRef(buffer.mVals, buffer.mSize);
  432. #ifdef CODEGEN_TRACK
  433. AutoCrit autoCrit(gBEMemReporterCritSect);
  434. AutoMemReporter autoMemReporter(&gBEMemReporter, "BeIRCodeGen");
  435. #endif
  436. //
  437. {
  438. BP_ZONE("BeIRCodeGen::ProcessBfIRData.HandleNextCmds");
  439. while (mStream->GetReadPos() < buffer.mSize)
  440. {
  441. if (mFailed)
  442. break;
  443. HandleNextCmd();
  444. }
  445. }
  446. BF_ASSERT((mFailed) || (mStream->GetReadPos() == buffer.mSize));
  447. }
  448. void BeIRCodeGen::Process()
  449. {
  450. BP_ZONE("BeIRCodeGen::process");
  451. //mDebugging |= ((mBeModule->mDbgModule != NULL) && (mBeModule->mDbgModule->mFileName == "ClassQ"));
  452. if (mDebugging)
  453. {
  454. String dbgStr;
  455. dbgStr = mBeModule->ToString();
  456. OutputDebugStr(dbgStr);
  457. }
  458. mBeModule->DoInlining();
  459. if (mDebugging)
  460. {
  461. String dbgStr = "-------------- AFTER INLINING --------------\n";
  462. dbgStr += mBeModule->ToString();
  463. OutputDebugStr(dbgStr);
  464. }
  465. }
  466. void BeIRCodeGen::ProcessBfIRData(const BfSizedArray<uint8>& buffer)
  467. {
  468. BP_ZONE("BeIRCodeGen::ProcessBfIRData");
  469. Init(buffer);
  470. Process();
  471. }
  472. BfTypeCode BeIRCodeGen::GetTypeCode(BeType * type, bool isSigned)
  473. {
  474. switch (type->mTypeCode)
  475. {
  476. case BeTypeCode_Int8:
  477. return (isSigned) ? BfTypeCode_Int8 : BfTypeCode_UInt8;
  478. case BeTypeCode_Int16:
  479. return (isSigned) ? BfTypeCode_Int16 : BfTypeCode_UInt16;
  480. case BeTypeCode_Int32:
  481. return (isSigned) ? BfTypeCode_Int32 : BfTypeCode_UInt32;
  482. case BeTypeCode_Int64:
  483. return (isSigned) ? BfTypeCode_Int64 : BfTypeCode_UInt64;
  484. case BeTypeCode_Float:
  485. return BfTypeCode_Float;
  486. case BeTypeCode_Double:
  487. return BfTypeCode_Double;
  488. default:
  489. return BfTypeCode_None;
  490. }
  491. }
  492. void BeIRCodeGen::SetResult(int id, BeValue* value)
  493. {
  494. BeIRCodeGenEntry entry;
  495. entry.mKind = BeIRCodeGenEntryKind_Value;
  496. entry.mBeValue = value;
  497. mResults.TryAdd(id, entry);
  498. }
  499. void BeIRCodeGen::SetResult(int id, BeType* type)
  500. {
  501. BeIRCodeGenEntry entry;
  502. entry.mKind = BeIRCodeGenEntryKind_Type;
  503. entry.mBeType = type;
  504. mResults.TryAdd(id, entry);
  505. }
  506. void BeIRCodeGen::SetResult(int id, BeBlock* value)
  507. {
  508. BeIRCodeGenEntry entry;
  509. entry.mKind = BeIRCodeGenEntryKind_Block;
  510. entry.mBeBlock = value;
  511. mResults.TryAdd(id, entry);
  512. }
  513. void BeIRCodeGen::SetResult(int id, BeMDNode* md)
  514. {
  515. BeIRCodeGenEntry entry;
  516. entry.mKind = BeIRCodeGenEntryKind_Metadata;
  517. entry.mBeMetadata = md;
  518. mResults.TryAdd(id, entry);
  519. }
  520. int64 BeIRCodeGen::ReadSLEB128()
  521. {
  522. int64 val = 0;
  523. int64 shift = 0;
  524. uint8 byteVal;
  525. do
  526. {
  527. byteVal = mStream->Read();
  528. val |= ((int64)(byteVal & 0x7f)) << shift;
  529. shift += 7;
  530. } while (byteVal >= 128);
  531. // Sign extend negative numbers.
  532. if ((byteVal & 0x40) && (shift < 64))
  533. val |= (-1ULL) << shift;
  534. return val;
  535. }
  536. void BeIRCodeGen::Read(StringImpl& str)
  537. {
  538. BE_MEM_START;
  539. int len = (int)ReadSLEB128();
  540. str.Append('?', len);
  541. mStream->Read((void*)str.c_str(), len);
  542. BE_MEM_END("String");
  543. }
  544. void BeIRCodeGen::Read(int& i)
  545. {
  546. BE_MEM_START;
  547. i = (int)ReadSLEB128();
  548. BE_MEM_END("int");
  549. }
  550. void BeIRCodeGen::Read(int64& i)
  551. {
  552. BE_MEM_START;
  553. i = ReadSLEB128();
  554. BE_MEM_END("int64");
  555. }
  556. void BeIRCodeGen::Read(Val128& i)
  557. {
  558. i.mLow = (uint64)ReadSLEB128();
  559. i.mHigh = (uint64)ReadSLEB128();
  560. }
  561. void BeIRCodeGen::Read(bool& val)
  562. {
  563. BE_MEM_START;
  564. val = mStream->Read() != 0;
  565. BE_MEM_END("bool");
  566. }
  567. void BeIRCodeGen::Read(int8& val)
  568. {
  569. val = mStream->Read();
  570. }
  571. void BeIRCodeGen::Read(BeIRTypeEntry*& type)
  572. {
  573. BE_MEM_START;
  574. int typeId = (int)ReadSLEB128();
  575. type = &GetTypeEntry(typeId);
  576. BE_MEM_END("BeIRTypeEntry");
  577. }
  578. void BeIRCodeGen::Read(BeType*& beType)
  579. {
  580. BE_MEM_START;
  581. BfIRType::TypeKind typeKind = (BfIRType::TypeKind)mStream->Read();
  582. if (typeKind == BfIRType::TypeKind::TypeKind_None)
  583. {
  584. beType = NULL;
  585. BE_MEM_END("BeType");
  586. return;
  587. }
  588. if (typeKind == BfIRType::TypeKind::TypeKind_Stream)
  589. {
  590. int streamId = (int)ReadSLEB128();
  591. if (streamId == -1)
  592. {
  593. beType = NULL;
  594. BE_MEM_END("BeType");
  595. return;
  596. }
  597. auto& result = mResults[streamId];
  598. BF_ASSERT(result.mKind == BeIRCodeGenEntryKind_Type);
  599. beType = result.mBeType;
  600. BE_MEM_END("BeType");
  601. return;
  602. }
  603. if (typeKind == BfIRType::TypeKind::TypeKind_SizedArray)
  604. {
  605. CMD_PARAM(BeType*, elementType);
  606. CMD_PARAM(int, length);
  607. beType = mBeContext->CreateSizedArrayType(elementType, length);
  608. return;
  609. }
  610. int typeId = (int)ReadSLEB128();
  611. if (typeKind == BfIRType::TypeKind::TypeKind_TypeCode)
  612. {
  613. bool isSigned = false;
  614. beType = GetBeType((BfTypeCode)typeId, isSigned);
  615. return;
  616. }
  617. auto& typeEntry = GetTypeEntry(typeId);
  618. if (typeKind == BfIRType::TypeKind::TypeKind_TypeId)
  619. beType = typeEntry.mBeType;
  620. else if (typeKind == BfIRType::TypeKind::TypeKind_TypeInstId)
  621. beType = typeEntry.mInstBeType;
  622. else if (typeKind == BfIRType::TypeKind::TypeKind_TypeInstPtrId)
  623. beType = mBeContext->GetPointerTo(typeEntry.mInstBeType);
  624. BE_MEM_END("BeType");
  625. }
  626. void BeIRCodeGen::Read(BeFunctionType*& beType)
  627. {
  628. BE_MEM_START;
  629. int streamId = (int)ReadSLEB128();
  630. auto& result = mResults[streamId];
  631. BF_ASSERT(result.mKind == BeIRCodeGenEntryKind_Type);
  632. beType = (BeFunctionType*)result.mBeType;
  633. BE_MEM_END("BeFunctionType");
  634. }
  635. void BeIRCodeGen::Read(BeValue*& beValue)
  636. {
  637. BE_MEM_START;
  638. BfIRParamType paramType = (BfIRParamType)mStream->Read();
  639. if (paramType == BfIRParamType_None)
  640. {
  641. beValue = NULL;
  642. BE_MEM_END("ParamType_None");
  643. }
  644. else if (paramType == BfIRParamType_Const)
  645. {
  646. BfTypeCode typeCode = (BfTypeCode)mStream->Read();
  647. BfConstType constType = (BfConstType)typeCode;
  648. if (constType == BfConstType_GlobalVar)
  649. {
  650. CMD_PARAM(int, streamId);
  651. if (streamId == -1)
  652. {
  653. int streamId = mCmdCount++;
  654. CMD_PARAM(BeType*, varType);
  655. CMD_PARAM(bool, isConstant);
  656. BfIRLinkageType linkageType = (BfIRLinkageType)mStream->Read();
  657. CMD_PARAM(BeConstant*, initializer);
  658. CMD_PARAM(String, name);
  659. CMD_PARAM(bool, isTLS);
  660. BF_ASSERT(varType != NULL);
  661. auto globalVariable = mBeModule->mGlobalVariables.Alloc();
  662. globalVariable->mModule = mBeModule;
  663. globalVariable->mType = varType;
  664. globalVariable->mIsConstant = isConstant;
  665. globalVariable->mLinkageType = linkageType;
  666. globalVariable->mInitializer = initializer;
  667. globalVariable->mName = name;
  668. globalVariable->mIsTLS = isTLS;
  669. globalVariable->mAlign = varType->mAlign;
  670. globalVariable->mUnnamedAddr = false;
  671. globalVariable->mStorageKind = BfIRStorageKind_Normal;
  672. if (initializer != NULL)
  673. BF_ASSERT(varType->mAlign > 0);
  674. SetResult(streamId, globalVariable);
  675. beValue = globalVariable;
  676. }
  677. else
  678. beValue = GetBeValue(streamId);
  679. beValue->mRefCount++;
  680. BE_MEM_END("ParamType_Const_GlobalVar");
  681. return;
  682. }
  683. else if ((constType == BfConstType_BitCast) || (constType == BfConstType_BitCastNull))
  684. {
  685. CMD_PARAM(BeConstant*, target);
  686. CMD_PARAM(BeType*, toType);
  687. auto castedVal = mBeModule->mAlloc.Alloc<BeCastConstant>();
  688. castedVal->mInt64 = target->mInt64;
  689. castedVal->mType = toType;
  690. castedVal->mTarget = target;
  691. BF_ASSERT(target->GetType() != NULL);
  692. BF_ASSERT(!target->GetType()->IsComposite());
  693. beValue = castedVal;
  694. BE_MEM_END("ParamType_Const_BitCast");
  695. return;
  696. }
  697. else if (constType == BfConstType_GEP32_1)
  698. {
  699. CMD_PARAM(BeConstant*, target);
  700. CMD_PARAM(int, idx0);
  701. BF_ASSERT(target->GetType()->IsPointer());
  702. auto gepConstant = mBeModule->mAlloc.Alloc<BeGEP1Constant>();
  703. gepConstant->mTarget = target;
  704. gepConstant->mIdx0 = idx0;
  705. beValue = gepConstant;
  706. BE_MEM_END("ParamType_Const_GEP32_1");
  707. return;
  708. }
  709. else if (constType == BfConstType_GEP32_2)
  710. {
  711. CMD_PARAM(BeConstant*, target);
  712. CMD_PARAM(int, idx0);
  713. CMD_PARAM(int, idx1);
  714. BF_ASSERT(target->GetType()->IsPointer());
  715. auto gepConstant = mBeModule->mAlloc.Alloc<BeGEP2Constant>();
  716. gepConstant->mTarget = target;
  717. gepConstant->mIdx0 = idx0;
  718. gepConstant->mIdx1 = idx1;
  719. beValue = gepConstant;
  720. BE_MEM_END("ParamType_Const_GEP32_2");
  721. return;
  722. }
  723. else if (constType == BfConstType_ExtractValue)
  724. {
  725. CMD_PARAM(BeConstant*, target);
  726. CMD_PARAM(int, idx0);
  727. auto gepConstant = mBeModule->mAlloc.Alloc<BeExtractValueConstant>();
  728. gepConstant->mTarget = target;
  729. gepConstant->mIdx0 = idx0;
  730. beValue = gepConstant;
  731. BE_MEM_END("ParamType_Const_ExtractValue");
  732. return;
  733. }
  734. else if (constType == BfConstType_PtrToInt)
  735. {
  736. CMD_PARAM(BeConstant*, target);
  737. BfTypeCode toTypeCode = (BfTypeCode)mStream->Read();
  738. BF_ASSERT(target->GetType()->IsPointer());
  739. bool isSigned = false;
  740. BeType* toType = GetBeType(toTypeCode, isSigned);
  741. auto castedVal = mBeModule->mAlloc.Alloc<BeCastConstant>();
  742. castedVal->mInt64 = target->mInt64;
  743. castedVal->mType = toType;
  744. castedVal->mTarget = target;
  745. BF_ASSERT(target->GetType() != NULL);
  746. beValue = castedVal;
  747. BE_MEM_END("ParamType_Const_PtrToInt");
  748. return;
  749. }
  750. else if (constType == BfConstType_IntToPtr)
  751. {
  752. CMD_PARAM(BeConstant*, target);
  753. CMD_PARAM(BeType*, toType);
  754. auto castedVal = mBeModule->mAlloc.Alloc<BeCastConstant>();
  755. castedVal->mInt64 = target->mInt64;
  756. castedVal->mType = toType;
  757. castedVal->mTarget = target;
  758. BF_ASSERT(target->GetType() != NULL);
  759. BF_ASSERT(!target->GetType()->IsComposite());
  760. BF_ASSERT(toType->IsPointer());
  761. beValue = castedVal;
  762. BE_MEM_END("ParamType_Const_IntToPtr");
  763. return;
  764. }
  765. else if (constType == BfConstType_AggZero)
  766. {
  767. CMD_PARAM(BeType*, type);
  768. auto beConst = mBeModule->mAlloc.Alloc<BeConstant>();
  769. beConst->mType = type;
  770. beValue = beConst;
  771. BE_MEM_END("ParamType_Const_AggZero");
  772. return;
  773. }
  774. else if (constType == BfConstType_Agg)
  775. {
  776. CMD_PARAM(BeType*, type);
  777. CMD_PARAM(CmdParamVec<BeConstant*>, values);
  778. if (type->IsSizedArray())
  779. {
  780. auto arrayType = (BeSizedArrayType*)type;
  781. int fillCount = (int)(arrayType->mLength - values.size());
  782. if (fillCount > 0)
  783. {
  784. auto lastValue = values.back();
  785. for (int i = 0; i < fillCount; i++)
  786. values.push_back(lastValue);
  787. }
  788. }
  789. else if (type->IsVector())
  790. {
  791. auto vecType = (BeVectorType*)type;
  792. int fillCount = (int)(vecType->mLength - values.size());
  793. if (fillCount > 0)
  794. {
  795. auto lastValue = values.back();
  796. for (int i = 0; i < fillCount; i++)
  797. values.push_back(lastValue);
  798. }
  799. }
  800. else
  801. {
  802. BF_ASSERT(type->IsStruct());
  803. }
  804. auto constStruct = mBeModule->mOwnedValues.Alloc<BeStructConstant>();
  805. constStruct->mType = type;
  806. if (type->IsStruct())
  807. FixValues((BeStructType*)type, values);
  808. for (int i = 0; i < (int)values.size(); i++)
  809. {
  810. auto val = values[i];
  811. BeConstant* constant = BeValueDynCast<BeConstant>(val);
  812. if (type->IsSizedArray())
  813. {
  814. auto arrayType = (BeSizedArrayType*)type;
  815. auto memberType = constant->GetType();
  816. if (memberType != arrayType->mElementType)
  817. Fail("ConstAgg array member type mismatch");
  818. }
  819. else if (type->IsVector())
  820. {
  821. auto vecType = (BeVectorType*)type;
  822. auto memberType = constant->GetType();
  823. if (memberType != vecType->mElementType)
  824. Fail("ConstAgg vector member type mismatch");
  825. }
  826. else
  827. {
  828. BF_ASSERT(type->IsStruct());
  829. auto structType = (BeStructType*)type;
  830. auto valType = constant->GetType();
  831. if (structType->mIsOpaque)
  832. {
  833. Fail("ConstAgg with opaque struct");
  834. }
  835. else if (valType != structType->mMembers[i].mType)
  836. {
  837. if (valType->IsSizedArray())
  838. {
  839. auto valSizedType = (BeSizedArrayType*)valType;
  840. if (valSizedType->mSize == 0)
  841. {
  842. constant->mType = structType->mMembers[i].mType;
  843. constStruct->mMemberValues.Add(constant);
  844. continue;
  845. }
  846. }
  847. Fail("ConstAgg struct member type mismatch");
  848. }
  849. }
  850. constStruct->mMemberValues.Add(constant);
  851. }
  852. beValue = constStruct;
  853. BE_MEM_END("ParamType_Const_Array");
  854. return;
  855. }
  856. else if (constType == BfConstType_ArrayZero8)
  857. {
  858. CMD_PARAM(int, count);
  859. auto beType = mBeContext->CreateSizedArrayType(mBeContext->GetPrimitiveType(BeTypeCode_Int8), count);
  860. auto beConst = mBeModule->mAlloc.Alloc<BeConstant>();
  861. beConst->mType = beType;
  862. beValue = beConst;
  863. return;
  864. }
  865. else if (constType == BfConstType_Undef)
  866. {
  867. CMD_PARAM(BeType*, type);
  868. auto constUndef = mBeModule->mOwnedValues.Alloc<BeUndefConstant>();
  869. constUndef->mType = type;
  870. beValue = constUndef;
  871. return;
  872. }
  873. else if (constType == BfConstType_TypeOf)
  874. {
  875. CMD_PARAM(BeType*, type);
  876. beValue = mReflectDataMap[type];
  877. BF_ASSERT(beValue != NULL);
  878. return;
  879. }
  880. else if (constType == BfConstType_TypeOf_Comptime)
  881. {
  882. CMD_PARAM(BeType*, typeType);
  883. CMD_PARAM(int, bfTypeId);
  884. auto beConst = mBeModule->mAlloc.Alloc<BeTypeOfConstant>();
  885. beConst->mType = typeType;
  886. beConst->mBfTypeId = bfTypeId;
  887. beValue = beConst;
  888. return;
  889. }
  890. else if (constType == BfConstType_TypeOf_WithData)
  891. {
  892. CMD_PARAM(BeType*, type);
  893. CMD_PARAM(BeValue*, value);
  894. mReflectDataMap[type] = value;
  895. beValue = value;
  896. return;
  897. }
  898. bool isSigned = false;
  899. BeType* llvmConstType = GetBeType(typeCode, isSigned);
  900. if (typeCode == BfTypeCode_Float)
  901. {
  902. float f;
  903. mStream->Read(&f, sizeof(float));
  904. beValue = mBeModule->GetConstant(llvmConstType, f);
  905. BE_MEM_END("ParamType_Single");
  906. }
  907. else if (typeCode == BfTypeCode_Double)
  908. {
  909. double d;
  910. mStream->Read(&d, sizeof(double));
  911. beValue = mBeModule->GetConstant(llvmConstType, d);
  912. BE_MEM_END("ParamType_Const_Double");
  913. }
  914. else if (typeCode == BfTypeCode_Boolean)
  915. {
  916. CMD_PARAM(bool, boolVal);
  917. beValue = mBeModule->GetConstant(llvmConstType, boolVal);
  918. BE_MEM_END("ParamType_Const_Boolean");
  919. }
  920. else if (typeCode == BfTypeCode_None)
  921. {
  922. beValue = NULL;
  923. BE_MEM_END("ParamType_Const_None");
  924. }
  925. else if (typeCode == BfTypeCode_NullPtr)
  926. {
  927. CMD_PARAM(BeType*, nullType);
  928. beValue = mBeModule->GetConstantNull((BePointerType*)nullType);
  929. BE_MEM_END("ParamType_Const_NullPtr");
  930. }
  931. else if (BfIRBuilder::IsInt(typeCode))
  932. {
  933. int64 intVal = ReadSLEB128();
  934. auto constVal = mBeModule->GetConstant(llvmConstType, intVal);
  935. beValue = constVal;
  936. BE_MEM_END("ParamType_Const_Int");
  937. }
  938. else
  939. {
  940. BF_FATAL("Unhandled");
  941. }
  942. }
  943. else if (paramType == BfIRParamType_Arg)
  944. {
  945. int argIdx = mStream->Read();
  946. beValue = mBeModule->GetArgument(argIdx);
  947. BE_MEM_END("ParamType_Arg");
  948. }
  949. else if (paramType == BfIRParamType_StreamId_Abs8)
  950. {
  951. int cmdId = mStream->Read();
  952. auto& result = mResults[cmdId];
  953. BF_ASSERT(result.mKind == BeIRCodeGenEntryKind_Value);
  954. beValue = result.mBeValue;
  955. BE_MEM_END("ParamType_StreamId");
  956. }
  957. else if (paramType == BfIRParamType_StreamId_Rel)
  958. {
  959. int cmdId = mCmdCount - (int)ReadSLEB128();
  960. auto& result = mResults[cmdId];
  961. BF_ASSERT(result.mKind == BeIRCodeGenEntryKind_Value);
  962. beValue = result.mBeValue;
  963. BE_MEM_END("ParamType_StreamId");
  964. }
  965. else
  966. {
  967. int cmdId = mCmdCount - (paramType - BfIRParamType_StreamId_Back1) - 1;
  968. auto& result = mResults[cmdId];
  969. BF_ASSERT(result.mKind == BeIRCodeGenEntryKind_Value);
  970. beValue = result.mBeValue;
  971. BE_MEM_END("ParamType_StreamId");
  972. }
  973. if (beValue != NULL)
  974. beValue->mRefCount++;
  975. }
  976. void BeIRCodeGen::Read(BeConstant*& llvmConstant)
  977. {
  978. BE_MEM_START;
  979. BeValue* value;
  980. Read(value);
  981. if (value == NULL)
  982. {
  983. llvmConstant = NULL;
  984. }
  985. else
  986. {
  987. BF_ASSERT(BeValueDynCast<BeConstant>(value));
  988. llvmConstant = (BeConstant*)value;
  989. }
  990. BE_MEM_END("BeConstant");
  991. }
  992. void BeIRCodeGen::Read(BeFunction*& beFunc)
  993. {
  994. BE_MEM_START;
  995. int streamId = (int)ReadSLEB128();
  996. if (streamId == -1)
  997. {
  998. beFunc = NULL;
  999. return;
  1000. }
  1001. auto& result = mResults[streamId];
  1002. BF_ASSERT(result.mKind == BeIRCodeGenEntryKind_Value);
  1003. BF_ASSERT(BeValueDynCast<BeFunction>(result.mBeValue));
  1004. beFunc = (BeFunction*)result.mBeValue;
  1005. BE_MEM_END("BeFunction");
  1006. beFunc->mRefCount++;
  1007. }
  1008. void BeIRCodeGen::Read(BeBlock*& beBlock)
  1009. {
  1010. BE_MEM_START;
  1011. int streamId = (int)ReadSLEB128();
  1012. auto& result = mResults[streamId];
  1013. BF_ASSERT(result.mKind == BeIRCodeGenEntryKind_Block);
  1014. beBlock = (BeBlock*)result.mBeType;
  1015. BE_MEM_END("BeBlock");
  1016. beBlock->mRefCount++;
  1017. }
  1018. void BeIRCodeGen::Read(BeMDNode*& llvmMD)
  1019. {
  1020. BE_MEM_START;
  1021. int streamId = (int)ReadSLEB128();
  1022. if (streamId == -1)
  1023. {
  1024. llvmMD = NULL;
  1025. return;
  1026. }
  1027. auto& result = mResults[streamId];
  1028. BF_ASSERT(result.mKind == BeIRCodeGenEntryKind_Metadata);
  1029. llvmMD = result.mBeMetadata;
  1030. BE_MEM_END("BeMDNode");
  1031. if (llvmMD != NULL)
  1032. llvmMD->mRefCount++;
  1033. }
  1034. void BeIRCodeGen::HandleNextCmd()
  1035. {
  1036. if (mFailed)
  1037. {
  1038. mStream->SetReadPos(mStream->GetSize());
  1039. return;
  1040. }
  1041. int curId = mCmdCount;
  1042. BfIRCmd cmd = (BfIRCmd)mStream->Read();
  1043. mCmdCount++;
  1044. #ifdef CODEGEN_TRACK
  1045. gBEMemReporter.BeginSection(gIRCmdNames[cmd]);
  1046. gBEMemReporter.Add(1);
  1047. #endif
  1048. switch (cmd)
  1049. {
  1050. case BfIRCmd_Module_Start:
  1051. {
  1052. CMD_PARAM(String, moduleName);
  1053. CMD_PARAM(int, ptrSize);
  1054. CMD_PARAM(bool, isOptimized);
  1055. BF_ASSERT(mBeModule == NULL);
  1056. mPtrSize = ptrSize;
  1057. mIsOptimized = isOptimized;
  1058. mBeContext = new BeContext();
  1059. mBeModule = new BeModule(moduleName, mBeContext);
  1060. mBeModule->mBeIRCodeGen = this;
  1061. mBeContext->mPointerSize = ptrSize;
  1062. for (auto constInt : mConfigConsts)
  1063. {
  1064. auto constVal = mBeModule->mAlloc.Alloc<BeConstant>();
  1065. constVal->mType = mBeContext->GetPrimitiveType(BeTypeCode_Int32);
  1066. constVal->mInt64 = constInt;
  1067. mBeModule->mConfigConsts32.Add(constVal);
  1068. constVal = mBeModule->mAlloc.Alloc<BeConstant>();
  1069. constVal->mType = mBeContext->GetPrimitiveType(BeTypeCode_Int64);
  1070. constVal->mInt64 = constInt;
  1071. mBeModule->mConfigConsts64.Add(constVal);
  1072. }
  1073. }
  1074. break;
  1075. case BfIRCmd_Module_SetTargetTriple:
  1076. {
  1077. CMD_PARAM(String, targetTriple);
  1078. CMD_PARAM(String, targetCPU);
  1079. mBeModule->mTargetTriple = targetTriple;
  1080. mBeModule->mTargetCPU = targetCPU;
  1081. }
  1082. break;
  1083. case BfIRCmd_Module_AddModuleFlag:
  1084. {
  1085. CMD_PARAM(String, flag);
  1086. CMD_PARAM(int, val);
  1087. //mBeModule->addModuleFlag(BeModule::Warning, flag, val);
  1088. }
  1089. break;
  1090. case BfIRCmd_WriteIR:
  1091. {
  1092. /*CMD_PARAM(String, fileName);
  1093. std::error_code ec;
  1094. Beraw_fd_ostream outStream(fileName.c_str(), ec, Besys::fs::OpenFlags::F_Text);
  1095. if (ec)
  1096. {
  1097. Fail("Failed writing IR '" + fileName + "': " + ec.message());
  1098. }
  1099. else
  1100. mBeModule->print(outStream, NULL);*/
  1101. }
  1102. break;
  1103. case BfIRCmd_Abort:
  1104. {
  1105. Fail("Stream aborted");
  1106. }
  1107. break;
  1108. case BfIRCmd_SetType:
  1109. {
  1110. CMD_PARAM(int, typeId);
  1111. CMD_PARAM(BeType*, type);
  1112. auto& typeEntry = GetTypeEntry(typeId);
  1113. typeEntry.mBeType = type;
  1114. if (typeEntry.mInstBeType == NULL)
  1115. typeEntry.mInstBeType = type;
  1116. }
  1117. break;
  1118. case BfIRCmd_SetInstType:
  1119. {
  1120. CMD_PARAM(int, typeId);
  1121. CMD_PARAM(BeType*, type);
  1122. GetTypeEntry(typeId).mInstBeType = type;
  1123. }
  1124. break;
  1125. case BfIRCmd_PrimitiveType:
  1126. {
  1127. BfTypeCode typeCode = (BfTypeCode)mStream->Read();
  1128. bool isSigned = false;
  1129. SetResult(curId, GetBeType(typeCode, isSigned));
  1130. }
  1131. break;
  1132. case BfIRCmd_CreateStruct:
  1133. {
  1134. CMD_PARAM(String, typeName);
  1135. SetResult(curId, mBeContext->CreateStruct(typeName));
  1136. }
  1137. break;
  1138. case BfIRCmd_CreateAnonymousStruct:
  1139. {
  1140. CMD_PARAM(CmdParamVec<BeType*>, members);
  1141. BeStructType* structType = mBeContext->CreateStruct(members);
  1142. SetResult(curId, structType);
  1143. }
  1144. break;
  1145. case BfIRCmd_StructSetBody:
  1146. {
  1147. CMD_PARAM(BeType*, type);
  1148. CMD_PARAM(CmdParamVec<BeType*>, members);
  1149. CMD_PARAM(int, instSize);
  1150. CMD_PARAM(int, instAlign);
  1151. CMD_PARAM(bool, isPacked);
  1152. if ((type == NULL) || (type->mTypeCode != BeTypeCode_Struct))
  1153. {
  1154. Fail("StructSetBody invalid type");
  1155. break;
  1156. }
  1157. BF_ASSERT(type->mTypeCode == BeTypeCode_Struct);
  1158. auto structType = (BeStructType*)type;
  1159. mBeContext->SetStructBody(structType, members, isPacked);
  1160. structType->mSize = instSize;
  1161. structType->mAlign = instAlign;
  1162. }
  1163. break;
  1164. case BfIRCmd_Type:
  1165. {
  1166. CMD_PARAM(BeIRTypeEntry*, typeEntry);
  1167. auto type = typeEntry->mBeType;
  1168. SetResult(curId, type);
  1169. }
  1170. break;
  1171. case BfIRCmd_TypeInst:
  1172. {
  1173. CMD_PARAM(BeIRTypeEntry*, typeEntry);
  1174. SetResult(curId, typeEntry->mInstBeType);
  1175. }
  1176. break;
  1177. case BfIRCmd_TypeInstPtr:
  1178. {
  1179. CMD_PARAM(BeIRTypeEntry*, typeEntry);
  1180. SetResult(curId, mBeContext->GetPointerTo(typeEntry->mInstBeType));
  1181. }
  1182. break;
  1183. case BfIRCmd_GetType:
  1184. {
  1185. CMD_PARAM(BeValue*, value);
  1186. auto type = value->GetType();
  1187. SetResult(curId, type);
  1188. }
  1189. break;
  1190. case BfIRCmd_GetPointerToFuncType:
  1191. {
  1192. CMD_PARAM(BeFunctionType*, funcType);
  1193. SetResult(curId, mBeContext->GetPointerTo(funcType));
  1194. }
  1195. break;
  1196. case BfIRCmd_GetPointerToType:
  1197. {
  1198. CMD_PARAM(BeType*, type);
  1199. SetResult(curId, mBeContext->GetPointerTo(type));
  1200. }
  1201. break;
  1202. case BfIRCmd_GetSizedArrayType:
  1203. {
  1204. CMD_PARAM(BeType*, elementType);
  1205. CMD_PARAM(int, length);
  1206. SetResult(curId, mBeContext->CreateSizedArrayType(elementType, length));
  1207. }
  1208. break;
  1209. case BfIRCmd_GetVectorType:
  1210. {
  1211. CMD_PARAM(BeType*, elementType);
  1212. CMD_PARAM(int, length);
  1213. SetResult(curId, mBeContext->CreateVectorType(elementType, length));
  1214. }
  1215. break;
  1216. case BfIRCmd_CreateConstAgg:
  1217. {
  1218. CMD_PARAM(BeType*, type);
  1219. CMD_PARAM(CmdParamVec<BeValue*>, values);
  1220. auto constStruct = mBeModule->mOwnedValues.Alloc<BeStructConstant>();
  1221. constStruct->mType = type;
  1222. if (type->IsStruct())
  1223. {
  1224. FixValues((BeStructType*)type, values);
  1225. BF_ASSERT(((BeStructType*)type)->mMembers.size() == values.size());
  1226. for (int i = 0; i < (int)values.size(); i++)
  1227. {
  1228. auto val = values[i];
  1229. BF_ASSERT(mBeContext->AreTypesEqual(((BeStructType*)type)->mMembers[i].mType, val->GetType()));
  1230. constStruct->mMemberValues.push_back(BeValueDynCast<BeConstant>(val));
  1231. }
  1232. }
  1233. else
  1234. {
  1235. BF_ASSERT(type->IsSizedArray());
  1236. auto arrayType = (BeSizedArrayType*)type;
  1237. int fillCount = (int)(arrayType->mLength - values.size());
  1238. if (fillCount > 0)
  1239. {
  1240. auto lastValue = values.back();
  1241. for (int i = 0; i < fillCount; i++)
  1242. values.push_back(lastValue);
  1243. }
  1244. BF_ASSERT(arrayType->mLength == values.size());
  1245. for (int i = 0; i < (int)values.size(); i++)
  1246. {
  1247. auto val = values[i];
  1248. BF_ASSERT(mBeContext->AreTypesEqual(((BeSizedArrayType*)type)->mElementType, val->GetType()));
  1249. constStruct->mMemberValues.push_back(BeValueDynCast<BeConstant>(val));
  1250. }
  1251. }
  1252. SetResult(curId, constStruct);
  1253. }
  1254. break;
  1255. case BfIRCmd_CreateConstStructZero:
  1256. {
  1257. CMD_PARAM(BeType*, type);
  1258. auto beConst = mBeModule->mAlloc.Alloc<BeConstant>();
  1259. beConst->mType = type;
  1260. SetResult(curId, beConst);
  1261. }
  1262. break;
  1263. case BfIRCmd_CreateConstString:
  1264. {
  1265. CMD_PARAM(String, str);
  1266. auto constStruct = mBeModule->mOwnedValues.Alloc<BeStringConstant>();
  1267. constStruct->mString = str;
  1268. auto charType = mBeContext->GetPrimitiveType(BeTypeCode_Int8);
  1269. constStruct->mType = mBeContext->CreateSizedArrayType(charType, str.length() + 1);
  1270. SetResult(curId, constStruct);
  1271. }
  1272. break;
  1273. case BfIRCmd_ConfigConst:
  1274. {
  1275. CMD_PARAM(int, constIdx);
  1276. BfTypeCode typeCode = (BfTypeCode)mStream->Read();
  1277. if (typeCode == BfTypeCode_IntPtr)
  1278. typeCode = (mPtrSize == 4) ? BfTypeCode_Int32 : BfTypeCode_Int64;
  1279. BeConstant* constVal = (typeCode == BfTypeCode_Int32) ?
  1280. mBeModule->mConfigConsts32[constIdx] :
  1281. mBeModule->mConfigConsts64[constIdx];
  1282. SetResult(curId, constVal);
  1283. }
  1284. break;
  1285. case BfIRCmd_SetName:
  1286. {
  1287. CMD_PARAM(BeValue*, val);
  1288. CMD_PARAM(String, name);
  1289. val->SetName(name);
  1290. }
  1291. break;
  1292. case BfIRCmd_CreateUndefValue:
  1293. {
  1294. CMD_PARAM(BeType*, type);
  1295. SetResult(curId, mBeModule->CreateUndefValue(type));
  1296. }
  1297. break;
  1298. case BfIRCmd_NumericCast:
  1299. {
  1300. CMD_PARAM(BeValue*, val);
  1301. CMD_PARAM(bool, valIsSigned);
  1302. BfTypeCode typeCode = (BfTypeCode)mStream->Read();
  1303. auto valType = val->GetType();
  1304. if ((!valType->IsIntable()) && (!valType->IsFloat()))
  1305. {
  1306. Fail("Invalid NumericCast target");
  1307. }
  1308. BfTypeCode valTypeCode = GetTypeCode(valType, valIsSigned);
  1309. if (auto srcCastConstant = BeValueDynCast<BeCastConstant>(val))
  1310. {
  1311. BeType* toType = GetBeType(typeCode, valIsSigned);
  1312. auto castedVal = mBeModule->mAlloc.Alloc<BeCastConstant>();
  1313. castedVal->mInt64 = srcCastConstant->mInt64;
  1314. castedVal->mType = toType;
  1315. castedVal->mTarget = srcCastConstant->mTarget;
  1316. SetResult(curId, castedVal);
  1317. break;
  1318. }
  1319. bool toSigned = false;
  1320. auto toBeType = GetBeType(typeCode, toSigned);
  1321. BeValue* retVal = mBeModule->CreateNumericCast(val, toBeType, valIsSigned, toSigned);
  1322. SetResult(curId, retVal);
  1323. }
  1324. break;
  1325. case BfIRCmd_CmpEQ:
  1326. {
  1327. CMD_PARAM(BeValue*, lhs);
  1328. CMD_PARAM(BeValue*, rhs);
  1329. if (lhs->GetType() != rhs->GetType())
  1330. Fail("Type mismatch for CmpEQ");
  1331. SetResult(curId, mBeModule->CreateCmp(BeCmpKind_EQ, lhs, rhs));
  1332. }
  1333. break;
  1334. case BfIRCmd_CmpNE:
  1335. {
  1336. CMD_PARAM(BeValue*, lhs);
  1337. CMD_PARAM(BeValue*, rhs);
  1338. SetResult(curId, mBeModule->CreateCmp(BeCmpKind_NE, lhs, rhs));
  1339. }
  1340. break;
  1341. case BfIRCmd_CmpSLT:
  1342. {
  1343. CMD_PARAM(BeValue*, lhs);
  1344. CMD_PARAM(BeValue*, rhs);
  1345. SetResult(curId, mBeModule->CreateCmp(BeCmpKind_SLT, lhs, rhs));
  1346. }
  1347. break;
  1348. case BfIRCmd_CmpULT:
  1349. {
  1350. CMD_PARAM(BeValue*, lhs);
  1351. CMD_PARAM(BeValue*, rhs);
  1352. SetResult(curId, mBeModule->CreateCmp(BeCmpKind_ULT, lhs, rhs));
  1353. }
  1354. break;
  1355. case BfIRCmd_CmpSLE:
  1356. {
  1357. CMD_PARAM(BeValue*, lhs);
  1358. CMD_PARAM(BeValue*, rhs);
  1359. SetResult(curId, mBeModule->CreateCmp(BeCmpKind_SLE, lhs, rhs));
  1360. }
  1361. break;
  1362. case BfIRCmd_CmpULE:
  1363. {
  1364. CMD_PARAM(BeValue*, lhs);
  1365. CMD_PARAM(BeValue*, rhs);
  1366. SetResult(curId, mBeModule->CreateCmp(BeCmpKind_ULE, lhs, rhs));
  1367. }
  1368. break;
  1369. case BfIRCmd_CmpSGT:
  1370. {
  1371. CMD_PARAM(BeValue*, lhs);
  1372. CMD_PARAM(BeValue*, rhs);
  1373. SetResult(curId, mBeModule->CreateCmp(BeCmpKind_SGT, lhs, rhs));
  1374. }
  1375. break;
  1376. case BfIRCmd_CmpUGT:
  1377. {
  1378. CMD_PARAM(BeValue*, lhs);
  1379. CMD_PARAM(BeValue*, rhs);
  1380. SetResult(curId, mBeModule->CreateCmp(BeCmpKind_UGT, lhs, rhs));
  1381. }
  1382. break;
  1383. case BfIRCmd_CmpSGE:
  1384. {
  1385. CMD_PARAM(BeValue*, lhs);
  1386. CMD_PARAM(BeValue*, rhs);
  1387. SetResult(curId, mBeModule->CreateCmp(BeCmpKind_SGE, lhs, rhs));
  1388. }
  1389. break;
  1390. case BfIRCmd_CmpUGE:
  1391. {
  1392. CMD_PARAM(BeValue*, lhs);
  1393. CMD_PARAM(BeValue*, rhs);
  1394. SetResult(curId, mBeModule->CreateCmp(BeCmpKind_UGE, lhs, rhs));
  1395. }
  1396. break;
  1397. case BfIRCmd_Add:
  1398. {
  1399. CMD_PARAM(BeValue*, lhs);
  1400. CMD_PARAM(BeValue*, rhs);
  1401. CMD_PARAM(int8, overflowCheckKind);
  1402. SetResult(curId, mBeModule->CreateBinaryOp(BeBinaryOpKind_Add, lhs, rhs, (BfOverflowCheckKind)overflowCheckKind));
  1403. }
  1404. break;
  1405. case BfIRCmd_Sub:
  1406. {
  1407. CMD_PARAM(BeValue*, lhs);
  1408. CMD_PARAM(BeValue*, rhs);
  1409. CMD_PARAM(int8, overflowCheckKind);
  1410. SetResult(curId, mBeModule->CreateBinaryOp(BeBinaryOpKind_Subtract, lhs, rhs, (BfOverflowCheckKind)overflowCheckKind));
  1411. }
  1412. break;
  1413. case BfIRCmd_Mul:
  1414. {
  1415. CMD_PARAM(BeValue*, lhs);
  1416. CMD_PARAM(BeValue*, rhs);
  1417. CMD_PARAM(int8, overflowCheckKind);
  1418. SetResult(curId, mBeModule->CreateBinaryOp(BeBinaryOpKind_Multiply, lhs, rhs, (BfOverflowCheckKind)overflowCheckKind));
  1419. }
  1420. break;
  1421. case BfIRCmd_SDiv:
  1422. {
  1423. CMD_PARAM(BeValue*, lhs);
  1424. CMD_PARAM(BeValue*, rhs);
  1425. SetResult(curId, mBeModule->CreateBinaryOp(BeBinaryOpKind_SDivide, lhs, rhs));
  1426. }
  1427. break;
  1428. case BfIRCmd_UDiv:
  1429. {
  1430. CMD_PARAM(BeValue*, lhs);
  1431. CMD_PARAM(BeValue*, rhs);
  1432. SetResult(curId, mBeModule->CreateBinaryOp(BeBinaryOpKind_UDivide, lhs, rhs));
  1433. }
  1434. break;
  1435. case BfIRCmd_SRem:
  1436. {
  1437. CMD_PARAM(BeValue*, lhs);
  1438. CMD_PARAM(BeValue*, rhs);
  1439. SetResult(curId, mBeModule->CreateBinaryOp(BeBinaryOpKind_SModulus, lhs, rhs));
  1440. }
  1441. break;
  1442. case BfIRCmd_URem:
  1443. {
  1444. CMD_PARAM(BeValue*, lhs);
  1445. CMD_PARAM(BeValue*, rhs);
  1446. SetResult(curId, mBeModule->CreateBinaryOp(BeBinaryOpKind_UModulus, lhs, rhs));
  1447. }
  1448. break;
  1449. case BfIRCmd_And:
  1450. {
  1451. CMD_PARAM(BeValue*, lhs);
  1452. CMD_PARAM(BeValue*, rhs);
  1453. SetResult(curId, mBeModule->CreateBinaryOp(BeBinaryOpKind_BitwiseAnd, lhs, rhs));
  1454. }
  1455. break;
  1456. case BfIRCmd_Or:
  1457. {
  1458. CMD_PARAM(BeValue*, lhs);
  1459. CMD_PARAM(BeValue*, rhs);
  1460. SetResult(curId, mBeModule->CreateBinaryOp(BeBinaryOpKind_BitwiseOr, lhs, rhs));
  1461. }
  1462. break;
  1463. case BfIRCmd_Xor:
  1464. {
  1465. CMD_PARAM(BeValue*, lhs);
  1466. CMD_PARAM(BeValue*, rhs);
  1467. SetResult(curId, mBeModule->CreateBinaryOp(BeBinaryOpKind_ExclusiveOr, lhs, rhs));
  1468. }
  1469. break;
  1470. case BfIRCmd_Shl:
  1471. {
  1472. CMD_PARAM(BeValue*, lhs);
  1473. CMD_PARAM(BeValue*, rhs);
  1474. SetResult(curId, mBeModule->CreateBinaryOp(BeBinaryOpKind_LeftShift, lhs, rhs));
  1475. }
  1476. break;
  1477. case BfIRCmd_AShr:
  1478. {
  1479. CMD_PARAM(BeValue*, lhs);
  1480. CMD_PARAM(BeValue*, rhs);
  1481. SetResult(curId, mBeModule->CreateBinaryOp(BeBinaryOpKind_ARightShift, lhs, rhs));
  1482. }
  1483. break;
  1484. case BfIRCmd_LShr:
  1485. {
  1486. CMD_PARAM(BeValue*, lhs);
  1487. CMD_PARAM(BeValue*, rhs);
  1488. SetResult(curId, mBeModule->CreateBinaryOp(BeBinaryOpKind_RightShift, lhs, rhs));
  1489. }
  1490. break;
  1491. case BfIRCmd_Neg:
  1492. {
  1493. CMD_PARAM(BeValue*, val);
  1494. auto negInst = mBeModule->AllocInst<BeNegInst>();
  1495. negInst->mValue = val;
  1496. SetResult(curId, negInst);
  1497. }
  1498. break;
  1499. case BfIRCmd_Not:
  1500. {
  1501. CMD_PARAM(BeValue*, val);
  1502. auto negInst = mBeModule->AllocInst<BeNotInst>();
  1503. negInst->mValue = val;
  1504. SetResult(curId, negInst);
  1505. }
  1506. break;
  1507. case BfIRCmd_BitCast:
  1508. {
  1509. CMD_PARAM(BeValue*, val);
  1510. CMD_PARAM(BeType*, toType);
  1511. if (auto funcVal = BeValueDynCast<BeFunction>(val))
  1512. {
  1513. auto beConst = mBeModule->mAlloc.Alloc<BeConstant>();
  1514. beConst->mTarget = funcVal;
  1515. BF_ASSERT(funcVal->mType != NULL);
  1516. beConst->mType = toType;
  1517. SetResult(curId, beConst);
  1518. break;
  1519. }
  1520. SetResult(curId, mBeModule->CreateBitCast(val, toType));
  1521. }
  1522. break;
  1523. case BfIRCmd_PtrToInt:
  1524. {
  1525. CMD_PARAM(BeValue*, val);
  1526. auto typeCode = (BfTypeCode)mStream->Read();
  1527. bool isSigned = false;
  1528. auto beType = GetBeType(typeCode, isSigned);
  1529. BF_ASSERT(beType != NULL);
  1530. auto numericCastInst = mBeModule->AllocInst<BeNumericCastInst>();
  1531. numericCastInst->mValue = val;
  1532. numericCastInst->mValSigned = false;
  1533. numericCastInst->mToType = beType;
  1534. numericCastInst->mToSigned = isSigned;
  1535. SetResult(curId, numericCastInst);
  1536. }
  1537. break;
  1538. case BfIRCmd_IntToPtr:
  1539. {
  1540. CMD_PARAM(BeValue*, val);
  1541. CMD_PARAM(BeType*, toType);
  1542. auto bitcastInst = mBeModule->AllocInst<BeBitCastInst>();
  1543. bitcastInst->mValue = val;
  1544. bitcastInst->mToType = toType;
  1545. SetResult(curId, bitcastInst);
  1546. }
  1547. break;
  1548. case BfIRCmd_InboundsGEP1_32:
  1549. {
  1550. CMD_PARAM(BeValue*, val);
  1551. CMD_PARAM(int, idx0);
  1552. BF_ASSERT(val->GetType()->IsPointer());
  1553. BeType* int32Type = mBeContext->GetPrimitiveType(BeTypeCode_Int32);
  1554. SetResult(curId, mBeModule->CreateGEP(val, mBeModule->GetConstant(int32Type, (int64)idx0), NULL));
  1555. }
  1556. break;
  1557. case BfIRCmd_InboundsGEP2_32:
  1558. {
  1559. CMD_PARAM(BeValue*, val);
  1560. CMD_PARAM(int, idx0);
  1561. CMD_PARAM(int, idx1);
  1562. BF_ASSERT(val->GetType()->IsPointer());
  1563. BeType* int32Type = mBeContext->GetPrimitiveType(BeTypeCode_Int32);
  1564. SetResult(curId, mBeModule->CreateGEP(val, mBeModule->GetConstant(int32Type, (int64)idx0), mBeModule->GetConstant(int32Type, (int64)idx1)));
  1565. }
  1566. break;
  1567. case BfIRCmd_InBoundsGEP1:
  1568. {
  1569. CMD_PARAM(BeValue*, val);
  1570. CMD_PARAM(BeValue*, idx0);
  1571. BF_ASSERT(val->GetType()->IsPointer());
  1572. SetResult(curId, mBeModule->CreateGEP(val, idx0, NULL));
  1573. }
  1574. break;
  1575. case BfIRCmd_InBoundsGEP2:
  1576. {
  1577. CMD_PARAM(BeValue*, val);
  1578. CMD_PARAM(BeValue*, idx0);
  1579. CMD_PARAM(BeValue*, idx1);
  1580. BF_ASSERT(val->GetType()->IsPointer());
  1581. SetResult(curId, mBeModule->CreateGEP(val, idx0, idx1));
  1582. }
  1583. break;
  1584. case BfIRCmd_IsNull:
  1585. {
  1586. CMD_PARAM(BeValue*, val);
  1587. BF_ASSERT(val->GetType()->IsPointer());
  1588. SetResult(curId, mBeModule->CreateCmp(BeCmpKind_EQ, val, mBeModule->GetConstantNull((BePointerType*)val->GetType())));
  1589. }
  1590. break;
  1591. case BfIRCmd_IsNotNull:
  1592. {
  1593. CMD_PARAM(BeValue*, val);
  1594. BF_ASSERT(val->GetType()->IsPointer());
  1595. SetResult(curId, mBeModule->CreateCmp(BeCmpKind_NE, val, mBeModule->GetConstantNull((BePointerType*)val->GetType())));
  1596. }
  1597. break;
  1598. case BfIRCmd_ExtractValue:
  1599. {
  1600. CMD_PARAM(BeValue*, val);
  1601. CMD_PARAM(int, idx);
  1602. BF_ASSERT(val->GetType()->IsComposite());
  1603. auto extractValueInst = mBeModule->AllocInst<BeExtractValueInst>();
  1604. extractValueInst->mAggVal = val;
  1605. extractValueInst->mIdx = idx;
  1606. SetResult(curId, extractValueInst);
  1607. }
  1608. break;
  1609. case BfIRCmd_InsertValue:
  1610. {
  1611. CMD_PARAM(BeValue*, agg);
  1612. CMD_PARAM(BeValue*, val);
  1613. CMD_PARAM(int, idx);
  1614. auto insertValueInst = mBeModule->AllocInst<BeInsertValueInst>();
  1615. insertValueInst->mAggVal = agg;
  1616. insertValueInst->mMemberVal = val;
  1617. insertValueInst->mIdx = idx;
  1618. SetResult(curId, insertValueInst);
  1619. }
  1620. break;
  1621. case BfIRCmd_Alloca:
  1622. {
  1623. CMD_PARAM(BeType*, type);
  1624. if (type->IsStruct())
  1625. {
  1626. BF_ASSERT(!((BeStructType*)type)->mIsOpaque);
  1627. }
  1628. auto allocaInst = mBeModule->CreateAlloca(type);
  1629. allocaInst->mAlign = type->mAlign;
  1630. SetResult(curId, allocaInst);
  1631. }
  1632. break;
  1633. case BfIRCmd_AllocaArray:
  1634. {
  1635. CMD_PARAM(BeType*, type);
  1636. CMD_PARAM(BeValue*, arraySize);
  1637. if (auto constant = BeValueDynCast<BeConstant>(arraySize))
  1638. {
  1639. //BF_ASSERT(constant->mInt64 >= 0);
  1640. }
  1641. auto allocaInst = mBeModule->AllocInst<BeAllocaInst>();
  1642. allocaInst->mType = type;
  1643. allocaInst->mAlign = type->mAlign;
  1644. allocaInst->mArraySize = arraySize;
  1645. SetResult(curId, allocaInst);
  1646. }
  1647. break;
  1648. case BfIRCmd_SetAllocaAlignment:
  1649. {
  1650. CMD_PARAM(BeValue*, val);
  1651. CMD_PARAM(int, alignment);
  1652. auto inst = BeValueDynCast<BeAllocaInst>(val);
  1653. inst->mAlign = alignment;
  1654. //TODO: Implement
  1655. /*inst->setAlignment(alignment);*/
  1656. }
  1657. break;
  1658. case BfIRCmd_AliasValue:
  1659. {
  1660. CMD_PARAM(BeValue*, val);
  1661. auto inst = mBeModule->AllocInst<BeAliasValueInst>();
  1662. inst->mPtr = val;
  1663. SetResult(curId, inst);
  1664. }
  1665. break;
  1666. case BfIRCmd_LifetimeStart:
  1667. {
  1668. CMD_PARAM(BeValue*, val);
  1669. auto inst = mBeModule->AllocInst<BeLifetimeStartInst>();
  1670. inst->mPtr = val;
  1671. SetResult(curId, inst);
  1672. }
  1673. break;
  1674. case BfIRCmd_LifetimeEnd:
  1675. {
  1676. CMD_PARAM(BeValue*, val);
  1677. #ifdef _DEBUG
  1678. val->mLifetimeEnded = true;
  1679. #endif
  1680. auto inst = mBeModule->AllocInst<BeLifetimeEndInst>();
  1681. inst->mPtr = val;
  1682. SetResult(curId, inst);
  1683. }
  1684. break;
  1685. case BfIRCmd_LifetimeSoftEnd:
  1686. {
  1687. CMD_PARAM(BeValue*, val);
  1688. auto inst = mBeModule->AllocInst<BeLifetimeSoftEndInst>();
  1689. inst->mPtr = val;
  1690. SetResult(curId, inst);
  1691. }
  1692. break;
  1693. case BfIRCmd_LifetimeExtend:
  1694. {
  1695. CMD_PARAM(BeValue*, val);
  1696. auto inst = mBeModule->AllocInst<BeLifetimeExtendInst>();
  1697. inst->mPtr = val;
  1698. SetResult(curId, inst);
  1699. }
  1700. break;
  1701. case BfIRCmd_ValueScopeStart:
  1702. {
  1703. auto inst = mBeModule->AllocInst<BeValueScopeStartInst>();
  1704. SetResult(curId, inst);
  1705. }
  1706. break;
  1707. case BfIRCmd_ValueScopeRetain:
  1708. {
  1709. CMD_PARAM(BeValue*, val);
  1710. auto inst = mBeModule->AllocInst<BeValueScopeRetainInst>();
  1711. inst->mValue = val;
  1712. }
  1713. break;
  1714. case BfIRCmd_ValueScopeSoftEnd:
  1715. {
  1716. CMD_PARAM(BeValue*, val);
  1717. auto inst = mBeModule->AllocInst<BeValueScopeEndInst>();
  1718. inst->mScopeStart = (BeValueScopeStartInst*)val;
  1719. inst->mIsSoft = true;
  1720. //TODO: Is this always correct? This keeps there from being nops inserted on block opens ( { )
  1721. inst->mDbgLoc = NULL;
  1722. }
  1723. break;
  1724. case BfIRCmd_ValueScopeHardEnd:
  1725. {
  1726. CMD_PARAM(BeValue*, val);
  1727. auto inst = mBeModule->AllocInst<BeValueScopeEndInst>();
  1728. inst->mScopeStart = (BeValueScopeStartInst*)val;
  1729. inst->mIsSoft = false;
  1730. //TODO: Is this always correct? This keeps there from being nops inserted on block opens ( { )
  1731. inst->mDbgLoc = NULL;
  1732. }
  1733. break;
  1734. case BfIRCmd_SetAllocaNoChkStkHint:
  1735. {
  1736. CMD_PARAM(BeValue*, val);
  1737. auto inst = BeValueDynCast<BeAllocaInst>(val);
  1738. inst->mNoChkStk = true;
  1739. }
  1740. break;
  1741. case BfIRCmd_SetAllocaForceMem:
  1742. {
  1743. CMD_PARAM(BeValue*, val);
  1744. auto inst = BeValueDynCast<BeAllocaInst>(val);
  1745. inst->mForceMem = true;
  1746. }
  1747. break;
  1748. case BfIRCmd_Load:
  1749. {
  1750. CMD_PARAM(BeValue*, val);
  1751. #ifdef _DEBUG
  1752. auto ptrType = val->GetType();
  1753. BF_ASSERT(ptrType->IsPointer());
  1754. // We call via a function pointer so there's never a reason to allow loading of a funcPtr
  1755. BF_ASSERT(((BePointerType*)ptrType)->mElementType->mTypeCode != BeTypeCode_Function);
  1756. // Disallow loading from a NULL constant
  1757. if (val->GetTypeId() == BeConstant::TypeId)
  1758. {
  1759. if (auto constant = BeValueDynCast<BeConstant>(val))
  1760. {
  1761. BF_ASSERT(constant->mTarget != NULL);
  1762. }
  1763. }
  1764. #endif
  1765. CMD_PARAM(bool, isVolatile);
  1766. SetResult(curId, mBeModule->CreateLoad(val, isVolatile));
  1767. }
  1768. break;
  1769. case BfIRCmd_AlignedLoad:
  1770. {
  1771. CMD_PARAM(BeValue*, val);
  1772. CMD_PARAM(int, alignment);
  1773. CMD_PARAM(bool, isVolatile);
  1774. #ifdef _DEBUG
  1775. // Disallow loading from a NULL constant
  1776. if (val->GetTypeId() == BeConstant::TypeId)
  1777. {
  1778. if (auto constant = BeValueDynCast<BeConstant>(val))
  1779. {
  1780. BF_ASSERT(constant->mTarget != NULL);
  1781. }
  1782. }
  1783. auto ptrType = val->GetType();
  1784. BF_ASSERT(ptrType->IsPointer());
  1785. #endif
  1786. SetResult(curId, mBeModule->CreateAlignedLoad(val, alignment, isVolatile));
  1787. }
  1788. break;
  1789. case BfIRCmd_Store:
  1790. {
  1791. CMD_PARAM(BeValue*, val);
  1792. CMD_PARAM(BeValue*, ptr);
  1793. #ifdef _DEBUG
  1794. auto ptrType = ptr->GetType();
  1795. auto valType = val->GetType();
  1796. if ((!ptrType->IsPointer()) || (!mBeContext->AreTypesEqual(((BePointerType*)ptrType)->mElementType, valType)))
  1797. {
  1798. String errStr;
  1799. errStr += "BfIRCmd_Store Match Failure:\n";
  1800. BeDumpContext dumpCtx;
  1801. errStr += "Val: ";
  1802. dumpCtx.ToString(errStr, val);
  1803. errStr += "\nPtr: ";
  1804. dumpCtx.ToString(errStr, ptr);
  1805. FatalError(errStr);
  1806. }
  1807. #endif
  1808. CMD_PARAM(bool, isVolatile);
  1809. SetResult(curId, mBeModule->CreateStore(val, ptr, isVolatile));
  1810. }
  1811. break;
  1812. case BfIRCmd_AlignedStore:
  1813. {
  1814. CMD_PARAM(BeValue*, val);
  1815. CMD_PARAM(BeValue*, ptr);
  1816. CMD_PARAM(int, alignment);
  1817. CMD_PARAM(bool, isVolatile);
  1818. #ifdef _DEBUG
  1819. auto ptrType = ptr->GetType();
  1820. auto valType = val->GetType();
  1821. if ((!ptrType->IsPointer()) || (!mBeContext->AreTypesEqual(((BePointerType*)ptrType)->mElementType, valType)))
  1822. {
  1823. String errStr;
  1824. errStr += "BfIRCmd_Store Match Failure:\n";
  1825. BeDumpContext dumpCtx;
  1826. errStr += "Val: ";
  1827. dumpCtx.ToString(errStr, val);
  1828. errStr += "\nPtr: ";
  1829. dumpCtx.ToString(errStr, ptr);
  1830. FatalError(errStr);
  1831. }
  1832. #endif
  1833. SetResult(curId, mBeModule->CreateAlignedStore(val, ptr, alignment, isVolatile));
  1834. }
  1835. break;
  1836. case BfIRCmd_MemSet:
  1837. {
  1838. auto inst = mBeModule->AllocInst<BeMemSetInst>();
  1839. Read(inst->mAddr);
  1840. Read(inst->mVal);
  1841. Read(inst->mSize);
  1842. Read(inst->mAlignment);
  1843. SetResult(curId, inst);
  1844. }
  1845. break;
  1846. case BfIRCmd_Fence:
  1847. {
  1848. BfIRFenceType fenceType = (BfIRFenceType)mStream->Read();
  1849. mBeModule->AllocInst<BeFenceInst>();
  1850. }
  1851. break;
  1852. case BfIRCmd_StackSave:
  1853. {
  1854. SetResult(curId, mBeModule->AllocInst<BeStackSaveInst>());
  1855. }
  1856. break;
  1857. case BfIRCmd_StackRestore:
  1858. {
  1859. CMD_PARAM(BeValue*, stackVal);
  1860. auto stackRestoreInst = mBeModule->AllocInst<BeStackRestoreInst>();
  1861. stackRestoreInst->mStackVal = stackVal;
  1862. SetResult(curId, stackRestoreInst);
  1863. }
  1864. break;
  1865. case BfIRCmd_GlobalVariable:
  1866. {
  1867. CMD_PARAM(BeType*, varType);
  1868. CMD_PARAM(bool, isConstant);
  1869. BfIRLinkageType linkageType = (BfIRLinkageType)mStream->Read();
  1870. CMD_PARAM(StringT<256>, name);
  1871. CMD_PARAM(bool, isTLS);
  1872. CMD_PARAM(BeConstant*, initializer);
  1873. BF_ASSERT(varType != NULL);
  1874. auto globalVariable = mBeModule->mGlobalVariables.Alloc();
  1875. globalVariable->mModule = mBeModule;
  1876. globalVariable->mType = varType;
  1877. globalVariable->mIsConstant = isConstant;
  1878. globalVariable->mLinkageType = linkageType;
  1879. globalVariable->mInitializer = initializer;
  1880. globalVariable->mName = name;
  1881. globalVariable->mIsTLS = isTLS;
  1882. globalVariable->mUnnamedAddr = false;
  1883. globalVariable->mStorageKind = BfIRStorageKind_Normal;
  1884. if (initializer != NULL)
  1885. {
  1886. globalVariable->mAlign = varType->mAlign;
  1887. BF_ASSERT(varType->mAlign > 0);
  1888. BF_ASSERT(mBeContext->AreTypesEqual(varType, initializer->GetType()));
  1889. }
  1890. else
  1891. globalVariable->mAlign = -1;
  1892. SetResult(curId, globalVariable);
  1893. }
  1894. break;
  1895. case BfIRCmd_GlobalVar_SetUnnamedAddr:
  1896. {
  1897. CMD_PARAM(BeValue*, val);
  1898. CMD_PARAM(bool, unnamedAddr);
  1899. BF_ASSERT(BeValueDynCast<BeGlobalVariable>(val) != NULL);
  1900. ((BeGlobalVariable*)val)->mUnnamedAddr = true;
  1901. }
  1902. break;
  1903. case BfIRCmd_GlobalVar_SetInitializer:
  1904. {
  1905. CMD_PARAM(BeValue*, val);
  1906. CMD_PARAM(BeConstant*, initializer);
  1907. BF_ASSERT(BeValueDynCast<BeGlobalVariable>(val) != NULL);
  1908. auto globalVariable = (BeGlobalVariable*)val;
  1909. globalVariable->mInitializer = initializer;
  1910. if (globalVariable->mInitializer != NULL)
  1911. {
  1912. globalVariable->mAlign = globalVariable->mType->mAlign;
  1913. BF_ASSERT(globalVariable->mAlign != -1);
  1914. }
  1915. }
  1916. break;
  1917. case BfIRCmd_GlobalVar_SetAlignment:
  1918. {
  1919. CMD_PARAM(BeValue*, val);
  1920. CMD_PARAM(int, alignment);
  1921. BF_ASSERT(BeValueDynCast<BeGlobalVariable>(val) != NULL);
  1922. auto globalVariable = (BeGlobalVariable*)val;
  1923. globalVariable->mAlign = alignment;
  1924. BF_ASSERT(alignment > 0);
  1925. if (globalVariable->mInitializer != NULL)
  1926. {
  1927. BF_ASSERT(globalVariable->mAlign != -1);
  1928. }
  1929. }
  1930. break;
  1931. case BfIRCmd_GlobalVar_SetStorageKind:
  1932. {
  1933. CMD_PARAM(BeValue*, val);
  1934. CMD_PARAM(int, storageKind);
  1935. BF_ASSERT(BeValueDynCast<BeGlobalVariable>(val) != NULL);
  1936. auto globalVariable = (BeGlobalVariable*)val;
  1937. globalVariable->mStorageKind = (BfIRStorageKind)storageKind;
  1938. }
  1939. break;
  1940. case BfIRCmd_GlobalStringPtr:
  1941. {
  1942. CMD_PARAM(String, str);
  1943. auto constStruct = mBeModule->mOwnedValues.Alloc<BeStringConstant>();
  1944. constStruct->mString = str;
  1945. auto charType = mBeContext->GetPrimitiveType(BeTypeCode_Int8);
  1946. constStruct->mType = mBeContext->CreateSizedArrayType(charType, str.length() + 1);
  1947. auto globalVariable = mBeModule->mGlobalVariables.Alloc();
  1948. globalVariable->mModule = mBeModule;
  1949. globalVariable->mType = constStruct->mType;
  1950. globalVariable->mIsConstant = true;
  1951. globalVariable->mLinkageType = BfIRLinkageType_Internal;
  1952. globalVariable->mInitializer = constStruct;
  1953. globalVariable->mName = StrFormat("__str%d", (int)mBeModule->mGlobalVariables.size() - 1);
  1954. globalVariable->mIsTLS = false;
  1955. globalVariable->mAlign = 1;
  1956. globalVariable->mUnnamedAddr = false;
  1957. auto castedVal = mBeModule->mAlloc.Alloc<BeCastConstant>();
  1958. castedVal->mType = mBeContext->GetPointerTo(charType);
  1959. castedVal->mTarget = globalVariable;
  1960. SetResult(curId, castedVal);
  1961. //SetResult(curId, globalVariable);
  1962. }
  1963. break;
  1964. case BfIRCmd_SetReflectTypeData:
  1965. {
  1966. CMD_PARAM(BeType*, type);
  1967. CMD_PARAM(BeValue*, value);
  1968. mReflectDataMap[type] = value;
  1969. }
  1970. break;
  1971. case BfIRCmd_CreateBlock:
  1972. {
  1973. CMD_PARAM(String, name);
  1974. CMD_PARAM(bool, addNow);
  1975. auto block = mBeModule->CreateBlock(name);
  1976. if (addNow)
  1977. mBeModule->AddBlock(mActiveFunction, block);
  1978. SetResult(curId, block);
  1979. }
  1980. break;
  1981. case BfIRCmd_MaybeChainNewBlock:
  1982. {
  1983. CMD_PARAM(String, name);
  1984. auto newBlock = mBeModule->GetInsertBlock();
  1985. if (!newBlock->IsEmpty())
  1986. {
  1987. auto bb = mBeModule->CreateBlock(name);
  1988. mBeModule->CreateBr(bb);
  1989. mBeModule->AddBlock(mActiveFunction, bb);
  1990. mBeModule->SetInsertPoint(bb);
  1991. newBlock = bb;
  1992. }
  1993. SetResult(curId, newBlock);
  1994. }
  1995. break;
  1996. case BfIRCmd_AddBlock:
  1997. {
  1998. CMD_PARAM(BeBlock*, block);
  1999. mBeModule->AddBlock(mActiveFunction, block);
  2000. }
  2001. break;
  2002. case BfIRCmd_DropBlocks:
  2003. {
  2004. CMD_PARAM(BeBlock*, startingBlock);
  2005. auto& basicBlockList = mActiveFunction->mBlocks;
  2006. int postExitBlockIdx = -1;
  2007. /*auto itr = basicBlockList.begin();
  2008. while (itr != basicBlockList.end())
  2009. {
  2010. auto block = *itr;
  2011. if (block == startingBlock)
  2012. {
  2013. basicBlockList.erase(itr, basicBlockList.end());
  2014. break;
  2015. }
  2016. ++itr;
  2017. }*/
  2018. for (int i = 0; i < (int)basicBlockList.size(); i++)
  2019. {
  2020. if (basicBlockList[i] == startingBlock)
  2021. {
  2022. basicBlockList.RemoveRange(i, basicBlockList.size() - i);
  2023. break;
  2024. }
  2025. }
  2026. }
  2027. break;
  2028. case BfIRCmd_MergeBlockDown:
  2029. {
  2030. CMD_PARAM(BeBlock*, fromBlock);
  2031. CMD_PARAM(BeBlock*, intoBlock);
  2032. for (auto inst : fromBlock->mInstructions)
  2033. inst->mParentBlock = intoBlock;
  2034. if (!fromBlock->mInstructions.IsEmpty())
  2035. intoBlock->mInstructions.Insert(0, &fromBlock->mInstructions[0], fromBlock->mInstructions.size());
  2036. mBeModule->RemoveBlock(mActiveFunction, fromBlock);
  2037. }
  2038. break;
  2039. case BfIRCmd_GetInsertBlock:
  2040. {
  2041. SetResult(curId, mBeModule->mActiveBlock);
  2042. }
  2043. break;
  2044. case BfIRCmd_SetInsertPoint:
  2045. {
  2046. CMD_PARAM(BeBlock*, block);
  2047. mBeModule->SetInsertPoint(block);
  2048. }
  2049. break;
  2050. case BfIRCmd_SetInsertPointAtStart:
  2051. {
  2052. CMD_PARAM(BeBlock*, block);
  2053. mBeModule->SetInsertPointAtStart(block);
  2054. }
  2055. break;
  2056. case BfIRCmd_EraseFromParent:
  2057. {
  2058. CMD_PARAM(BeBlock*, block);
  2059. mBeModule->RemoveBlock(mActiveFunction, block);
  2060. }
  2061. break;
  2062. case BfIRCmd_DeleteBlock:
  2063. {
  2064. CMD_PARAM(BeBlock*, block);
  2065. }
  2066. break;
  2067. case BfIRCmd_EraseInstFromParent:
  2068. {
  2069. CMD_PARAM(BeValue*, instVal);
  2070. BeInst* inst = (BeInst*)instVal;
  2071. bool wasRemoved = inst->mParentBlock->mInstructions.Remove(inst);
  2072. BF_ASSERT(wasRemoved);
  2073. #ifdef _DEBUG
  2074. inst->mWasRemoved = true;
  2075. #endif
  2076. }
  2077. break;
  2078. case BfIRCmd_CreateBr:
  2079. {
  2080. CMD_PARAM(BeBlock*, block);
  2081. mBeModule->CreateBr(block);
  2082. }
  2083. break;
  2084. case BfIRCmd_CreateBr_Fake:
  2085. {
  2086. CMD_PARAM(BeBlock*, block);
  2087. auto inst = mBeModule->CreateBr(block);
  2088. inst->mIsFake = true;
  2089. }
  2090. break;
  2091. case BfIRCmd_CreateBr_NoCollapse:
  2092. {
  2093. CMD_PARAM(BeBlock*, block);
  2094. auto inst = mBeModule->CreateBr(block);
  2095. inst->mNoCollapse = true;
  2096. }
  2097. break;
  2098. case BfIRCmd_CreateCondBr:
  2099. {
  2100. CMD_PARAM(BeValue*, condVal);
  2101. CMD_PARAM(BeBlock*, trueBlock);
  2102. CMD_PARAM(BeBlock*, falseBlock);
  2103. mBeModule->CreateCondBr(condVal, trueBlock, falseBlock);
  2104. }
  2105. break;
  2106. case BfIRCmd_MoveBlockToEnd:
  2107. {
  2108. CMD_PARAM(BeBlock*, block);
  2109. mBeModule->RemoveBlock(mActiveFunction, block);
  2110. mBeModule->AddBlock(mActiveFunction, block);
  2111. }
  2112. break;
  2113. case BfIRCmd_CreateSwitch:
  2114. {
  2115. CMD_PARAM(BeValue*, val);
  2116. CMD_PARAM(BeBlock*, dest);
  2117. CMD_PARAM(int, numCases);
  2118. auto switchInst = mBeModule->AllocInstOwned<BeSwitchInst>();
  2119. switchInst->mValue = val;
  2120. switchInst->mDefaultBlock = dest;
  2121. switchInst->mCases.Reserve(numCases);
  2122. SetResult(curId, switchInst);
  2123. }
  2124. break;
  2125. case BfIRCmd_AddSwitchCase:
  2126. {
  2127. CMD_PARAM(BeValue*, switchVal);
  2128. CMD_PARAM(BeValue*, caseVal);
  2129. CMD_PARAM(BeBlock*, caseBlock);
  2130. BeSwitchCase switchCase;
  2131. switchCase.mValue = (BeConstant*)caseVal;
  2132. switchCase.mBlock = caseBlock;
  2133. ((BeSwitchInst*)switchVal)->mCases.push_back(switchCase);
  2134. }
  2135. break;
  2136. case BfIRCmd_SetSwitchDefaultDest:
  2137. {
  2138. CMD_PARAM(BeValue*, switchVal);
  2139. CMD_PARAM(BeBlock*, caseBlock);
  2140. ((BeSwitchInst*)switchVal)->mDefaultBlock = caseBlock;
  2141. }
  2142. break;
  2143. case BfIRCmd_CreatePhi:
  2144. {
  2145. CMD_PARAM(BeType*, type);
  2146. CMD_PARAM(int, incomingCount);
  2147. auto phiInst = mBeModule->AllocInstOwned<BePhiInst>();
  2148. phiInst->mType = type;
  2149. SetResult(curId, phiInst);
  2150. }
  2151. break;
  2152. case BfIRCmd_AddPhiIncoming:
  2153. {
  2154. CMD_PARAM(BeValue*, phiValue);
  2155. CMD_PARAM(BeValue*, value);
  2156. CMD_PARAM(BeBlock*, comingFrom);
  2157. if (phiValue->GetType() != value->GetType())
  2158. {
  2159. Fail("AddPhiIncoming type mismatch");
  2160. break;
  2161. }
  2162. auto phiIncoming = mBeModule->mAlloc.Alloc<BePhiIncoming>();
  2163. phiIncoming->mBlock = comingFrom;
  2164. phiIncoming->mValue = value;
  2165. ((BePhiInst*)phiValue)->mIncoming.push_back(phiIncoming);
  2166. }
  2167. break;
  2168. case BfIRCmd_GetIntrinsic:
  2169. {
  2170. CMD_PARAM(String, intrinName);
  2171. CMD_PARAM(int, intrinId);
  2172. CMD_PARAM(BeType*, returnType);
  2173. CMD_PARAM(CmdParamVec<BeType*>, paramTypes);
  2174. auto intrin = mBeModule->mAlloc.Alloc<BeIntrinsic>();
  2175. intrin->mName = intrinName;
  2176. intrin->mKind = (BfIRIntrinsic)intrinId;
  2177. intrin->mReturnType = returnType;
  2178. SetResult(curId, intrin);
  2179. }
  2180. break;
  2181. case BfIRCmd_CreateFunctionType:
  2182. {
  2183. CMD_PARAM(BeType*, resultType);
  2184. CMD_PARAM(CmdParamVec<BeType*>, paramTypes);
  2185. CMD_PARAM(bool, isVarArg);
  2186. auto functionType = mBeContext->CreateFunctionType(resultType, paramTypes, isVarArg);
  2187. SetResult(curId, functionType);
  2188. }
  2189. break;
  2190. case BfIRCmd_CreateFunction:
  2191. {
  2192. CMD_PARAM(BeFunctionType*, type);
  2193. BfIRLinkageType linkageType = (BfIRLinkageType)mStream->Read();
  2194. CMD_PARAM(String, name);
  2195. SetResult(curId, mBeModule->CreateFunction(type, linkageType, name));
  2196. }
  2197. break;
  2198. case BfIRCmd_SetFunctionName:
  2199. {
  2200. CMD_PARAM(BeValue*, func);
  2201. CMD_PARAM(String, name);
  2202. BeFunction* beFunc = BeValueDynCast<BeFunction>(func);
  2203. beFunc->mName = name;
  2204. }
  2205. break;
  2206. case BfIRCmd_EnsureFunctionPatchable:
  2207. {
  2208. }
  2209. break;
  2210. case BfIRCmd_RemapBindFunction:
  2211. {
  2212. CMD_PARAM(BeValue*, func);
  2213. // We need to store this value to a data segment so we get a symbol we can remap during hot swap
  2214. // We actually do this to ensure that we don't bind to the NEW method but rather the old one- so
  2215. // delegate equality checks still work
  2216. BeFunction* beFunc = BeValueDynCast<BeFunction>(func);
  2217. if (beFunc != NULL)
  2218. {
  2219. if (beFunc->mRemapBindVar == NULL)
  2220. {
  2221. auto globalVariable = mBeModule->mGlobalVariables.Alloc();
  2222. globalVariable->mModule = mBeModule;
  2223. globalVariable->mType = beFunc->mType;
  2224. globalVariable->mIsConstant = true;
  2225. globalVariable->mLinkageType = BfIRLinkageType_External;
  2226. globalVariable->mInitializer = beFunc;
  2227. globalVariable->mName = StrFormat("bf_hs_preserve@%s_%s", beFunc->mName.c_str(), mBeModule->mModuleName.c_str());
  2228. globalVariable->mIsTLS = false;
  2229. globalVariable->mAlign = 8;
  2230. globalVariable->mUnnamedAddr = false;
  2231. beFunc->mRemapBindVar = globalVariable;
  2232. /*if (mBeModule->mDbgModule != NULL)
  2233. {
  2234. auto dbgGlobalVariable = mBeModule->mDbgModule->mGlobalVariables.Alloc();
  2235. dbgGlobalVariable->mContext = mBeContext;
  2236. dbgGlobalVariable->mName = name;
  2237. dbgGlobalVariable->mLinkageName = globalVariable->mName;
  2238. dbgGlobalVariable->mFile = (BeDbgFile*)file;
  2239. dbgGlobalVariable->mLineNum = lineNum;
  2240. dbgGlobalVariable->mType = (BeDbgType*)type;
  2241. dbgGlobalVariable->mIsLocalToUnit = isLocalToUnit;
  2242. dbgGlobalVariable->mValue = val;
  2243. dbgGlobalVariable->mDecl = decl;
  2244. }*/
  2245. }
  2246. SetResult(curId, mBeModule->CreateLoad(beFunc->mRemapBindVar, false));
  2247. }
  2248. else
  2249. SetResult(curId, func);
  2250. }
  2251. break;
  2252. case BfIRCmd_SetActiveFunction:
  2253. {
  2254. CMD_PARAM(BeFunction*, func);
  2255. mActiveFunction = func;
  2256. mBeModule->mActiveFunction = func;
  2257. }
  2258. break;
  2259. case BfIRCmd_CreateCall:
  2260. {
  2261. CMD_PARAM(BeValue*, func);
  2262. CMD_PARAM(CmdParamVec<BeValue*>, args);
  2263. #ifdef BE_EXTRA_CHECKS
  2264. auto funcPtrType = func->GetType();
  2265. if (funcPtrType != NULL)
  2266. {
  2267. BF_ASSERT(funcPtrType->IsPointer());
  2268. auto funcType = (BeFunctionType*)((BePointerType*)funcPtrType)->mElementType;
  2269. BF_ASSERT(funcType->mTypeCode == BeTypeCode_Function);
  2270. bool argsMatched = true;
  2271. if (!funcType->mIsVarArg)
  2272. {
  2273. if (funcType->mParams.size() != args.size())
  2274. {
  2275. argsMatched = false;
  2276. }
  2277. else
  2278. {
  2279. int argIdx = 0;
  2280. for (int argIdx = 0; argIdx < (int)args.size(); argIdx++)
  2281. {
  2282. if (args[argIdx] == NULL)
  2283. argsMatched = false;
  2284. else if (funcType->mParams[argIdx].mType != args[argIdx]->GetType())
  2285. argsMatched = false;
  2286. }
  2287. }
  2288. }
  2289. if (!argsMatched)
  2290. {
  2291. String errStr;
  2292. errStr += "BfIRCmd_CreateCall Match Failure:\n";
  2293. BeDumpContext dumpCtx;
  2294. dumpCtx.ToString(errStr, func);
  2295. errStr += "\n";
  2296. dumpCtx.ToString(errStr, funcType);
  2297. errStr += "\n";
  2298. for (int argIdx = 0; argIdx < (int)args.size(); argIdx++)
  2299. {
  2300. errStr += StrFormat("ARG #%d: ", argIdx);
  2301. dumpCtx.ToString(errStr, args[argIdx]);
  2302. errStr += "\n";
  2303. }
  2304. FatalError(errStr);
  2305. }
  2306. }
  2307. else
  2308. {
  2309. BF_ASSERT(func->GetTypeId() == BeIntrinsic::TypeId);
  2310. }
  2311. #endif
  2312. SetResult(curId, mBeModule->CreateCall(func, args));
  2313. }
  2314. break;
  2315. case BfIRCmd_SetCallCallingConv:
  2316. {
  2317. CMD_PARAM(BeValue*, callInst);
  2318. BfIRCallingConv callingConv = (BfIRCallingConv)mStream->Read();
  2319. ((BeCallInst*)callInst)->mCallingConv = callingConv;
  2320. }
  2321. break;
  2322. case BfIRCmd_SetFuncCallingConv:
  2323. {
  2324. CMD_PARAM(BeFunction*, func);
  2325. BfIRCallingConv callingConv = (BfIRCallingConv)mStream->Read();
  2326. func->mCallingConv = callingConv;
  2327. }
  2328. break;
  2329. case BfIRCmd_SetTailCall:
  2330. {
  2331. CMD_PARAM(BeValue*, callInstVal);
  2332. BeCallInst* callInst = (BeCallInst*)callInstVal;
  2333. callInst->mTailCall = true;
  2334. }
  2335. break;
  2336. case BfIRCmd_SetCallAttribute:
  2337. {
  2338. CMD_PARAM(BeValue*, callInstVal);
  2339. CMD_PARAM(int, paramIdx);
  2340. BfIRAttribute attribute = (BfIRAttribute)mStream->Read();
  2341. BeCallInst* callInst = (BeCallInst*)callInstVal;
  2342. if (attribute == BfIRAttribute_NoReturn)
  2343. callInst->mNoReturn = true;
  2344. }
  2345. break;
  2346. case BfIRCmd_CreateRet:
  2347. {
  2348. CMD_PARAM(BeValue*, val);
  2349. #ifdef BE_EXTRA_CHECKS
  2350. auto retType = val->GetType();
  2351. auto funcType = mActiveFunction->GetFuncType();
  2352. BF_ASSERT(retType == funcType->mReturnType);
  2353. #endif
  2354. SetResult(curId, mBeModule->CreateRet(val));
  2355. }
  2356. break;
  2357. case BfIRCmd_CreateSetRet:
  2358. {
  2359. CMD_PARAM(BeValue*, val);
  2360. CMD_PARAM(int, returnTypeId);
  2361. SetResult(curId, mBeModule->CreateSetRet(val, returnTypeId));
  2362. }
  2363. break;
  2364. case BfIRCmd_CreateRetVoid:
  2365. {
  2366. mBeModule->CreateRetVoid();
  2367. }
  2368. break;
  2369. case BfIRCmd_CreateUnreachable:
  2370. {
  2371. mBeModule->AllocInst<BeUnreachableInst>();
  2372. }
  2373. break;
  2374. case BfIRCmd_Call_AddAttribute:
  2375. {
  2376. CMD_PARAM(BeValue*, callInstVal);
  2377. CMD_PARAM(int, argIdx);
  2378. BfIRAttribute attribute = (BfIRAttribute)mStream->Read();
  2379. BeCallInst* callInst = (BeCallInst*)callInstVal;
  2380. if (argIdx > 0)
  2381. {
  2382. if (attribute == BfIRAttribute_StructRet)
  2383. {
  2384. auto valType = callInst->mArgs[argIdx - 1].mValue->GetType();
  2385. BF_ASSERT(valType->IsPointer());
  2386. callInst->mArgs[argIdx - 1].mStructRet = true;
  2387. }
  2388. else if (attribute == BfIRAttribute_ZExt)
  2389. callInst->mArgs[argIdx - 1].mZExt = true;
  2390. else if (attribute == BfIRAttribute_NoAlias)
  2391. callInst->mArgs[argIdx - 1].mNoAlias = true;
  2392. else if (attribute == BfIRAttribute_NoCapture)
  2393. callInst->mArgs[argIdx - 1].mNoCapture = true;
  2394. else if (attribute == BfIRAttribute_ByVal)
  2395. {
  2396. }
  2397. else
  2398. BF_FATAL("Unhandled");
  2399. }
  2400. else
  2401. {
  2402. if (attribute == BfIRAttribute_NoReturn)
  2403. callInst->mNoReturn = true;
  2404. else
  2405. BF_FATAL("Unhandled");
  2406. }
  2407. }
  2408. break;
  2409. case BfIRCmd_Call_AddAttribute1:
  2410. {
  2411. CMD_PARAM(BeValue*, inst);
  2412. CMD_PARAM(int, argIdx);
  2413. BfIRAttribute attribute = (BfIRAttribute)mStream->Read();
  2414. CMD_PARAM(int, arg);
  2415. BeCallInst* callInst = BeValueDynCast<BeCallInst>(inst);
  2416. if (callInst != NULL)
  2417. {
  2418. if (argIdx > 0)
  2419. {
  2420. if (attribute == BfIRAttribute_Dereferencable)
  2421. {
  2422. callInst->mArgs[argIdx - 1].mDereferenceableSize = arg;
  2423. if (auto func = BeValueDynCast<BeFunction>(callInst->mFunc))
  2424. {
  2425. BF_ASSERT(func->mParams[argIdx - 1].mDereferenceableSize == arg);
  2426. }
  2427. }
  2428. else if (attribute == BfIRAttribute_ByVal)
  2429. {
  2430. callInst->mArgs[argIdx - 1].mByRefSize = arg;
  2431. if (auto func = BeValueDynCast<BeFunction>(callInst->mFunc))
  2432. {
  2433. BF_ASSERT((func->mParams[argIdx - 1].mByValSize == arg) || (func->mParams[argIdx - 1].mByValSize == -1));
  2434. }
  2435. }
  2436. else
  2437. BF_FATAL("Unhandled");
  2438. }
  2439. else
  2440. {
  2441. BF_FATAL("Unhandled");
  2442. }
  2443. }
  2444. }
  2445. break;
  2446. case BfIRCmd_Func_AddAttribute:
  2447. {
  2448. CMD_PARAM(BeFunction*, func);
  2449. CMD_PARAM(int, argIdx);
  2450. BfIRAttribute attribute = (BfIRAttribute)mStream->Read();
  2451. if (argIdx > 0)
  2452. {
  2453. if (attribute == BfIRAttribute_StructRet)
  2454. func->mParams[argIdx - 1].mStructRet = true;
  2455. else if (attribute == BfIRAttribute_NoAlias)
  2456. func->mParams[argIdx - 1].mNoAlias = true;
  2457. else if (attribute == BfIRAttribute_NoCapture)
  2458. func->mParams[argIdx - 1].mNoCapture = true;
  2459. else if (attribute == BfIRAttribute_ZExt)
  2460. func->mParams[argIdx - 1].mZExt = true;
  2461. else
  2462. BF_FATAL("Unhandled");
  2463. }
  2464. else
  2465. {
  2466. if (attribute == BfIRAttribute_VarRet)
  2467. func->mIsVarReturn = true;
  2468. else if (attribute == BFIRAttribute_AlwaysInline)
  2469. func->mAlwaysInline = true;
  2470. else if (attribute == BFIRAttribute_NoUnwind)
  2471. func->mNoUnwind = true;
  2472. else if (attribute == BFIRAttribute_UWTable)
  2473. func->mUWTable = true;
  2474. else if (attribute == BfIRAttribute_NoReturn)
  2475. func->mNoReturn = true;
  2476. else if (attribute == BFIRAttribute_NoFramePointerElim)
  2477. func->mNoFramePointerElim = true;
  2478. else if (attribute == BFIRAttribute_DllExport)
  2479. func->mIsDLLExport = true;
  2480. else if (attribute == BFIRAttribute_DllImport)
  2481. func->mIsDLLImport = true;
  2482. else if (attribute == BFIRAttribute_NoRecurse)
  2483. {
  2484. }
  2485. else if (attribute == BFIRAttribute_Constructor)
  2486. {
  2487. }
  2488. else if (attribute == BFIRAttribute_Destructor)
  2489. {
  2490. }
  2491. else
  2492. BF_FATAL("Unhandled");
  2493. }
  2494. }
  2495. break;
  2496. case BfIRCmd_Func_AddAttribute1:
  2497. {
  2498. CMD_PARAM(BeFunction*, func);
  2499. CMD_PARAM(int, argIdx);
  2500. BfIRAttribute attribute = (BfIRAttribute)mStream->Read();
  2501. CMD_PARAM(int, arg);
  2502. // This is for adding things like Dereferencable, which we don't use
  2503. if (argIdx > 0)
  2504. {
  2505. if (attribute == BfIRAttribute_Dereferencable)
  2506. func->mParams[argIdx - 1].mDereferenceableSize = arg;
  2507. else if (attribute == BfIRAttribute_ByVal)
  2508. func->mParams[argIdx - 1].mByValSize = arg;
  2509. else
  2510. BF_FATAL("Unhandled");
  2511. }
  2512. else
  2513. BF_FATAL("Unhandled");
  2514. }
  2515. break;
  2516. case BfIRCmd_Func_SetParamName:
  2517. {
  2518. CMD_PARAM(BeFunction*, func);
  2519. CMD_PARAM(int, argIdx);
  2520. CMD_PARAM(String, name);
  2521. if (argIdx > 0)
  2522. func->mParams[argIdx - 1].mName = name;
  2523. }
  2524. break;
  2525. case BfIRCmd_Func_DeleteBody:
  2526. {
  2527. CMD_PARAM(BeFunction*, func);
  2528. func->mBlocks.Clear();
  2529. }
  2530. break;
  2531. case BfIRCmd_Func_SafeRename:
  2532. {
  2533. CMD_PARAM(BeFunction*, func);
  2534. func->mName += StrFormat("__RENAME%d", curId);
  2535. }
  2536. break;
  2537. case BfIRCmd_Func_SafeRenameFrom:
  2538. {
  2539. CMD_PARAM(BeFunction*, func);
  2540. CMD_PARAM(String, prevName);
  2541. if (func->mName == prevName)
  2542. func->mName += StrFormat("__RENAME%d", curId);
  2543. }
  2544. break;
  2545. case BfIRCmd_Func_SetLinkage:
  2546. {
  2547. CMD_PARAM(BeFunction*, func);
  2548. BfIRLinkageType linkageType = (BfIRLinkageType)mStream->Read();
  2549. func->mLinkageType = linkageType;
  2550. }
  2551. break;
  2552. case BfIRCmd_SaveDebugLocation:
  2553. {
  2554. mSavedDebugLocs.push_back(mBeModule->GetCurrentDebugLocation());
  2555. }
  2556. break;
  2557. case BfIRCmd_RestoreDebugLocation:
  2558. {
  2559. mBeModule->SetCurrentDebugLocation(mSavedDebugLocs[mSavedDebugLocs.size() - 1]);
  2560. mSavedDebugLocs.pop_back();
  2561. }
  2562. break;
  2563. case BfIRCmd_DupDebugLocation:
  2564. {
  2565. mBeModule->DupCurrentDebugLocation();
  2566. }
  2567. break;
  2568. case BfIRCmd_ClearDebugLocation:
  2569. {
  2570. mBeModule->SetCurrentDebugLocation(NULL);
  2571. }
  2572. break;
  2573. case BfIRCmd_ClearDebugLocationInst:
  2574. {
  2575. CMD_PARAM(BeValue*, instValue);
  2576. auto inst = (BeInst*)instValue;
  2577. inst->mDbgLoc = NULL;
  2578. }
  2579. break;
  2580. case BfIRCmd_ClearDebugLocationInstLast:
  2581. {
  2582. if ((mBeModule->mActiveBlock != NULL) && (!mBeModule->mActiveBlock->mInstructions.IsEmpty()))
  2583. {
  2584. auto inst = mBeModule->mActiveBlock->mInstructions.back();
  2585. inst->mDbgLoc = NULL;
  2586. }
  2587. }
  2588. break;
  2589. case BfIRCmd_UpdateDebugLocation:
  2590. {
  2591. CMD_PARAM(BeValue*, instValue);
  2592. auto inst = (BeInst*)instValue;
  2593. inst->mDbgLoc = mBeModule->mCurDbgLoc;
  2594. }
  2595. break;
  2596. case BfIRCmd_SetCurrentDebugLocation:
  2597. {
  2598. CMD_PARAM(int, line);
  2599. CMD_PARAM(int, column);
  2600. CMD_PARAM(BeMDNode*, diScope);
  2601. CMD_PARAM(BeMDNode*, diInlinedAt);
  2602. BF_ASSERT(diScope != NULL);
  2603. mBeModule->SetCurrentDebugLocation(line - 1, column - 1, diScope, (BeDbgLoc*)diInlinedAt);
  2604. }
  2605. break;
  2606. case BfIRCmd_Nop:
  2607. {
  2608. mBeModule->CreateNop();
  2609. }
  2610. break;
  2611. case BfIRCmd_EnsureInstructionAt:
  2612. {
  2613. mBeModule->AllocInst<BeEnsureInstructionAtInst>();
  2614. }
  2615. break;
  2616. case BfIRCmd_StatementStart:
  2617. {
  2618. }
  2619. break;
  2620. case BfIRCmd_ObjectAccessCheck:
  2621. {
  2622. CMD_PARAM(BeValue*, val);
  2623. CMD_PARAM(bool, useAsm);
  2624. auto inst = mBeModule->AllocInst<BeObjectAccessCheckInst>();
  2625. inst->mValue = val;
  2626. SetResult(curId, mBeModule->GetInsertBlock());
  2627. }
  2628. break;
  2629. case BfIRCmd_Comptime_Error:
  2630. {
  2631. CMD_PARAM(int32, error);
  2632. auto inst = mBeModule->AllocInst<BeComptimeError>();
  2633. inst->mError = error;
  2634. SetResult(curId, inst);
  2635. }
  2636. break;
  2637. case BfIRCmd_Comptime_GetBfType:
  2638. {
  2639. CMD_PARAM(int32, typeId);
  2640. CMD_PARAM(BeType*, resultType);
  2641. auto inst = mBeModule->AllocInst<BeComptimeGetType>();
  2642. inst->mTypeId = typeId;
  2643. inst->mResultType = resultType;
  2644. SetResult(curId, inst);
  2645. }
  2646. break;
  2647. case BfIRCmd_Comptime_GetReflectType:
  2648. {
  2649. CMD_PARAM(int32, typeId);
  2650. CMD_PARAM(BeType*, resultType);
  2651. auto inst = mBeModule->AllocInst<BeComptimeGetReflectType>();
  2652. inst->mTypeId = typeId;
  2653. inst->mResultType = resultType;
  2654. SetResult(curId, inst);
  2655. }
  2656. break;
  2657. case BfIRCmd_Comptime_DynamicCastCheck:
  2658. {
  2659. CMD_PARAM(BeValue*, value);
  2660. CMD_PARAM(int32, typeId);
  2661. CMD_PARAM(BeType*, resultType);
  2662. auto inst = mBeModule->AllocInst<BeComptimeDynamicCastCheck>();
  2663. inst->mValue = value;
  2664. inst->mTypeId = typeId;
  2665. inst->mResultType = resultType;
  2666. SetResult(curId, inst);
  2667. }
  2668. break;
  2669. case BfIRCmd_Comptime_GetVirtualFunc:
  2670. {
  2671. CMD_PARAM(BeValue*, value);
  2672. CMD_PARAM(int32, virtualTableIdx);
  2673. CMD_PARAM(BeType*, resultType);
  2674. auto inst = mBeModule->AllocInst<BeComptimeGetVirtualFunc>();
  2675. inst->mValue = value;
  2676. inst->mVirtualTableIdx = virtualTableIdx;
  2677. inst->mResultType = resultType;
  2678. SetResult(curId, inst);
  2679. }
  2680. break;
  2681. case BfIRCmd_Comptime_GetInterfaceFunc:
  2682. {
  2683. CMD_PARAM(BeValue*, value);
  2684. CMD_PARAM(int32, ifaceTypeId);
  2685. CMD_PARAM(int32, methodIdx);
  2686. CMD_PARAM(BeType*, resultType);
  2687. auto inst = mBeModule->AllocInst<BeComptimeGetInterfaceFunc>();
  2688. inst->mValue = value;
  2689. inst->mIFaceTypeId = ifaceTypeId;
  2690. inst->mMethodIdx = methodIdx;
  2691. inst->mResultType = resultType;
  2692. SetResult(curId, inst);
  2693. }
  2694. break;
  2695. case BfIRCmd_DbgInit:
  2696. {
  2697. /*mDIBuilder = new BeDIBuilder(*mBeModule); */
  2698. mBeModule->mDbgModule = new BeDbgModule();
  2699. mBeModule->mDbgModule->mBeModule = mBeModule;
  2700. }
  2701. break;
  2702. case BfIRCmd_DbgFinalize:
  2703. {
  2704. /*for (auto typeEntryPair : mTypes)
  2705. {
  2706. auto& typeEntry = typeEntryPair.second;
  2707. if (typeEntry.mInstDIType != NULL)
  2708. typeEntry.mInstDIType->resolveCycles();
  2709. }
  2710. mDIBuilder->finalize();*/
  2711. }
  2712. break;
  2713. case BfIRCmd_DbgCreateCompileUnit:
  2714. {
  2715. CMD_PARAM(int, lang);
  2716. CMD_PARAM(String, fileName);
  2717. CMD_PARAM(String, directory);
  2718. CMD_PARAM(String, producer);
  2719. CMD_PARAM(bool, isOptimized);
  2720. CMD_PARAM(String, flags);
  2721. CMD_PARAM(int, runtimeVer);
  2722. CMD_PARAM(bool, linesOnly);
  2723. mBeModule->mDbgModule->mFileName = fileName;
  2724. mBeModule->mDbgModule->mDirectory = directory;
  2725. mBeModule->mDbgModule->mProducer = producer;
  2726. //mDebugging = fileName == "TestToots";
  2727. SetResult(curId, mBeModule->mDbgModule);
  2728. }
  2729. break;
  2730. case BfIRCmd_DbgCreateFile:
  2731. {
  2732. CMD_PARAM(String, fileName);
  2733. CMD_PARAM(String, directory);
  2734. CMD_PARAM(Val128, md5Hash);
  2735. auto dbgFile = mBeModule->mDbgModule->mFiles.Alloc();
  2736. dbgFile->mFileName = fileName;
  2737. dbgFile->mDirectory = directory;
  2738. dbgFile->mMD5Hash = md5Hash;
  2739. dbgFile->mIdx = (int)mBeModule->mDbgModule->mFiles.size() - 1;
  2740. SetResult(curId, dbgFile);
  2741. }
  2742. break;
  2743. case BfIRCmd_DbgGetCurrentLocation:
  2744. {
  2745. SetResult(curId, mBeModule->mCurDbgLoc);
  2746. }
  2747. break;
  2748. case BfIRCmd_DbgSetType:
  2749. {
  2750. CMD_PARAM(int, typeId);
  2751. CMD_PARAM(BeMDNode*, type);
  2752. auto& typeEntry = GetTypeEntry(typeId);
  2753. typeEntry.mDIType = BeValueDynCast<BeDbgType>(type);
  2754. if (typeEntry.mInstDIType == NULL)
  2755. typeEntry.mInstDIType = BeValueDynCast<BeDbgType>(type);
  2756. }
  2757. break;
  2758. case BfIRCmd_DbgSetInstType:
  2759. {
  2760. CMD_PARAM(int, typeId);
  2761. CMD_PARAM(BeMDNode*, type);
  2762. GetTypeEntry(typeId).mInstDIType = BeValueDynCast<BeDbgType>(type);
  2763. }
  2764. break;
  2765. case BfIRCmd_DbgGetType:
  2766. {
  2767. CMD_PARAM(int, typeId);
  2768. if (mBeModule->mCeMachine != NULL)
  2769. {
  2770. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgTypeId>();
  2771. dbgType->mTypeId = typeId;
  2772. SetResult(curId, dbgType);
  2773. }
  2774. else
  2775. SetResult(curId, GetTypeEntry(typeId).mDIType);
  2776. }
  2777. break;
  2778. case BfIRCmd_DbgGetTypeInst:
  2779. {
  2780. CMD_PARAM(int, typeId);
  2781. auto result = GetTypeEntry(typeId).mInstDIType;
  2782. //BF_ASSERT(result);
  2783. SetResult(curId, result);
  2784. }
  2785. break;
  2786. case BfIRCmd_DbgTrackDITypes:
  2787. {
  2788. CMD_PARAM(int, typeId);
  2789. auto& typeEntry = GetTypeEntry(typeId);
  2790. /*if (typeEntry.mDIType != NULL)
  2791. BeMetadataTracking::track(*(BeMetadata**)&typeEntry.mDIType);
  2792. if (typeEntry.mInstDIType != NULL)
  2793. BeMetadataTracking::track(*(BeMetadata**)&typeEntry.mInstDIType);*/
  2794. //NotImpl();
  2795. }
  2796. break;
  2797. case BfIRCmd_DbgCreateNamespace:
  2798. {
  2799. CMD_PARAM(BeMDNode*, scope);
  2800. CMD_PARAM(String, name);
  2801. CMD_PARAM(BeMDNode*, file);
  2802. CMD_PARAM(int, lineNum);
  2803. auto dbgNamespace = mBeModule->mOwnedValues.Alloc<BeDbgNamespace>();
  2804. dbgNamespace->mScope = scope;
  2805. dbgNamespace->mName = name;
  2806. SetResult(curId, dbgNamespace);
  2807. }
  2808. break;
  2809. case BfIRCmd_DbgCreateImportedModule:
  2810. {
  2811. CMD_PARAM(BeMDNode*, context);
  2812. CMD_PARAM(BeMDNode*, namespaceNode);
  2813. CMD_PARAM(int, lineNum);
  2814. /*SetResult(curId, mDIBuilder->createImportedModule((BeDIScope*)context, (BeDINamespace*)namespaceNode, lineNum));*/
  2815. //NotImpl();
  2816. }
  2817. break;
  2818. case BfIRCmd_DbgCreateBasicType:
  2819. {
  2820. CMD_PARAM(String, name);
  2821. CMD_PARAM(int64, sizeInBits);
  2822. CMD_PARAM(int64, alignInBits);
  2823. CMD_PARAM(int, encoding);
  2824. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgBasicType>();
  2825. dbgType->mName = name;
  2826. dbgType->mSize = (int)(sizeInBits / 8);
  2827. dbgType->mAlign = (int)(alignInBits / 8);
  2828. dbgType->mEncoding = encoding;
  2829. SetResult(curId, dbgType);
  2830. }
  2831. break;
  2832. case BfIRCmd_DbgCreateStructType:
  2833. {
  2834. CMD_PARAM(BeMDNode*, context);
  2835. CMD_PARAM(String, name);
  2836. CMD_PARAM(BeMDNode*, file);
  2837. CMD_PARAM(int, lineNum);
  2838. CMD_PARAM(int64, sizeInBits);
  2839. CMD_PARAM(int64, alignInBits);
  2840. CMD_PARAM(int, flags);
  2841. CMD_PARAM(BeMDNode*, derivedFrom);
  2842. CMD_PARAM(CmdParamVec<BeMDNode*>, members);
  2843. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgStructType>();
  2844. dbgType->mScope = context;
  2845. dbgType->mName = name;
  2846. dbgType->mSize = (int)(sizeInBits / 8);
  2847. dbgType->mAlign = (int)(alignInBits / 8);
  2848. dbgType->mDerivedFrom = BeValueDynCast<BeDbgType>(derivedFrom);
  2849. dbgType->mDefFile = (BeDbgFile*)file;
  2850. dbgType->mDefLine = lineNum - 1;
  2851. dbgType->mIsFullyDefined = true;
  2852. dbgType->SetMembers(members);
  2853. SetResult(curId, dbgType);
  2854. }
  2855. break;
  2856. case BfIRCmd_DbgCreateEnumerationType:
  2857. {
  2858. CMD_PARAM(BeMDNode*, context);
  2859. CMD_PARAM(String, name);
  2860. CMD_PARAM(BeMDNode*, file);
  2861. CMD_PARAM(int, lineNum);
  2862. CMD_PARAM(int64, sizeInBits);
  2863. CMD_PARAM(int64, alignInBits);
  2864. CMD_PARAM(CmdParamVec<BeMDNode*>, members);
  2865. CMD_PARAM(BeMDNode*, underlyingType);
  2866. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgEnumType>();
  2867. dbgType->mScope = context;
  2868. dbgType->mName = name;
  2869. dbgType->mSize = (int)(sizeInBits / 8);
  2870. dbgType->mAlign = (int)(alignInBits / 8);
  2871. dbgType->mIsFullyDefined = true;
  2872. dbgType->mElementType = BeValueDynCast<BeDbgType>(underlyingType);
  2873. for (auto member : members)
  2874. {
  2875. if (auto enumMember = BeValueDynCast<BeDbgEnumMember>(member))
  2876. {
  2877. dbgType->mMembers.push_back(enumMember);
  2878. }
  2879. else
  2880. NotImpl();
  2881. }
  2882. //dbgType->mDefFile = (BeDbgFile*)file;
  2883. //dbgType->mDefLine = line - 1;
  2884. SetResult(curId, dbgType);
  2885. }
  2886. break;
  2887. case BfIRCmd_DbgCreatePointerType:
  2888. {
  2889. CMD_PARAM(BeMDNode*, elementTypeNode);
  2890. BeDbgType* elementType = BeValueDynCast<BeDbgType>(elementTypeNode);
  2891. if (elementType == NULL)
  2892. {
  2893. if (auto dbgTypeId = BeValueDynCast<BeDbgTypeId>(elementTypeNode))
  2894. {
  2895. auto bfElementType = mBeModule->mCeMachine->mCeModule->mContext->mTypes[dbgTypeId->mTypeId];
  2896. auto bfPtrType = mBeModule->mCeMachine->mCeModule->CreatePointerType(bfElementType);
  2897. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgTypeId>();
  2898. dbgType->mTypeId = bfPtrType->mTypeId;
  2899. SetResult(curId, dbgType);
  2900. break;
  2901. }
  2902. }
  2903. BeDbgType* useType = elementType->FindDerivedType(BeDbgPointerType::TypeId);
  2904. if (useType == NULL)
  2905. {
  2906. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgPointerType>();
  2907. dbgType->mElement = elementType;
  2908. dbgType->mSize = mPtrSize;
  2909. dbgType->mAlign = mPtrSize;
  2910. elementType->mDerivedTypes.PushFront(dbgType, &mBeModule->mAlloc);
  2911. useType = dbgType;
  2912. }
  2913. SetResult(curId, useType);
  2914. }
  2915. break;
  2916. case BfIRCmd_DbgCreateReferenceType:
  2917. {
  2918. CMD_PARAM(BeMDNode*, elementTypeNode);
  2919. BeDbgType* elementType = BeValueDynCast<BeDbgType>(elementTypeNode);
  2920. if (elementType == NULL)
  2921. {
  2922. if (auto dbgTypeId = BeValueDynCast<BeDbgTypeId>(elementTypeNode))
  2923. {
  2924. auto bfElementType = mBeModule->mCeMachine->mCeModule->mContext->mTypes[dbgTypeId->mTypeId];
  2925. auto bfPtrType = mBeModule->mCeMachine->mCeModule->CreateRefType(bfElementType);
  2926. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgTypeId>();
  2927. dbgType->mTypeId = bfPtrType->mTypeId;
  2928. SetResult(curId, dbgType);
  2929. break;
  2930. }
  2931. }
  2932. auto useType = mBeModule->mDbgModule->CreateReferenceType(elementType);
  2933. SetResult(curId, useType);
  2934. }
  2935. break;
  2936. case BfIRCmd_DbgCreateConstType:
  2937. {
  2938. CMD_PARAM(BeMDNode*, elementTypeNode);
  2939. BeDbgType* elementType = BeValueDynCast<BeDbgType>(elementTypeNode);
  2940. if (elementType == NULL)
  2941. {
  2942. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgConstType>();
  2943. dbgType->mElement = elementTypeNode;
  2944. SetResult(curId, dbgType);
  2945. break;
  2946. }
  2947. BeDbgType* useType = elementType->FindDerivedType(BeDbgConstType::TypeId);
  2948. if (useType == NULL)
  2949. {
  2950. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgConstType>();
  2951. dbgType->mElement = elementType;
  2952. elementType->mDerivedTypes.PushFront(dbgType, &mBeModule->mAlloc);
  2953. useType = dbgType;
  2954. }
  2955. SetResult(curId, useType);
  2956. }
  2957. break;
  2958. case BfIRCmd_DbgCreateArtificialType:
  2959. {
  2960. CMD_PARAM(BeMDNode*, diType);
  2961. //auto dbgType = mBeModule->mOwnedValues.Alloc<BeDbgArtificialType>();
  2962. //dbgType->mElement = (BeDbgType*)diType;
  2963. // Does the artificial thing do anything for us actually?
  2964. auto dbgType = diType;
  2965. SetResult(curId, dbgType);
  2966. }
  2967. break;
  2968. case BfIRCmd_DbgCreateArrayType:
  2969. {
  2970. CMD_PARAM(int64, sizeInBits);
  2971. CMD_PARAM(int64, alignInBits);
  2972. CMD_PARAM(BeMDNode*, elementType);
  2973. CMD_PARAM(int64, numElements);
  2974. auto dbgArray = mBeModule->mDbgModule->mTypes.Alloc<BeDbgArrayType>();
  2975. dbgArray->mSize = (int)(sizeInBits / 8);
  2976. dbgArray->mAlign = (int)(alignInBits / 8);
  2977. dbgArray->mElement = BeValueDynCast<BeDbgType>(elementType);
  2978. dbgArray->mNumElements = numElements;
  2979. SetResult(curId, dbgArray);
  2980. }
  2981. break;
  2982. case BfIRCmd_DbgCreateReplaceableCompositeType:
  2983. {
  2984. CMD_PARAM(int, tag);
  2985. CMD_PARAM(String, name);
  2986. CMD_PARAM(BeMDNode*, scope);
  2987. CMD_PARAM(BeMDNode*, file);
  2988. CMD_PARAM(int, line);
  2989. CMD_PARAM(int64, sizeInBits);
  2990. CMD_PARAM(int64, alignInBits);
  2991. CMD_PARAM(int, flags);
  2992. if (tag == llvm::dwarf::DW_TAG_structure_type)
  2993. {
  2994. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgStructType>();
  2995. dbgType->mScope = scope;
  2996. dbgType->mName = name;
  2997. dbgType->mSize = (int)(sizeInBits / 8);
  2998. dbgType->mAlign = (int)(alignInBits / 8);
  2999. dbgType->mDefFile = (BeDbgFile*)file;
  3000. dbgType->mDefLine = line - 1;
  3001. SetResult(curId, dbgType);
  3002. }
  3003. else if (tag == llvm::dwarf::DW_TAG_enumeration_type)
  3004. {
  3005. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgEnumType>();
  3006. dbgType->mScope = scope;
  3007. dbgType->mName = name;
  3008. dbgType->mSize = (int)(sizeInBits / 8);
  3009. dbgType->mAlign = (int)(alignInBits / 8);
  3010. //dbgType->mDefFile = (BeDbgFile*)file;
  3011. //dbgType->mDefLine = line - 1;
  3012. SetResult(curId, dbgType);
  3013. }
  3014. else
  3015. NotImpl();
  3016. }
  3017. break;
  3018. case BfIRCmd_DbgCreateForwardDecl:
  3019. {
  3020. CMD_PARAM(int, tag);
  3021. CMD_PARAM(String, name);
  3022. CMD_PARAM(BeMDNode*, scope);
  3023. CMD_PARAM(BeMDNode*, file);
  3024. CMD_PARAM(int, line);
  3025. if (tag == llvm::dwarf::DW_TAG_structure_type)
  3026. {
  3027. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgStructType>();
  3028. dbgType->mScope = scope;
  3029. dbgType->mName = name;
  3030. dbgType->mDefFile = (BeDbgFile*)file;
  3031. dbgType->mDefLine = line;
  3032. SetResult(curId, dbgType);
  3033. }
  3034. else if (tag == llvm::dwarf::DW_TAG_enumeration_type)
  3035. {
  3036. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgEnumType>();
  3037. dbgType->mScope = scope;
  3038. dbgType->mName = name;
  3039. SetResult(curId, dbgType);
  3040. }
  3041. else
  3042. NotImpl();
  3043. }
  3044. break;
  3045. case BfIRCmd_DbgCreateSizedForwardDecl:
  3046. {
  3047. CMD_PARAM(int, tag);
  3048. CMD_PARAM(String, name);
  3049. CMD_PARAM(BeMDNode*, scope);
  3050. CMD_PARAM(BeMDNode*, file);
  3051. CMD_PARAM(int, line);
  3052. CMD_PARAM(int64, sizeInBits);
  3053. CMD_PARAM(int64, alignInBits);
  3054. if (tag == llvm::dwarf::DW_TAG_structure_type)
  3055. {
  3056. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgStructType>();
  3057. dbgType->mDefFile = (BeDbgFile*)file;
  3058. dbgType->mDefLine = line;
  3059. dbgType->mScope = scope;
  3060. dbgType->mName = name;
  3061. dbgType->mSize = (int)(sizeInBits / 8);
  3062. dbgType->mAlign = (int)(alignInBits / 8);
  3063. SetResult(curId, dbgType);
  3064. }
  3065. else if (tag == llvm::dwarf::DW_TAG_enumeration_type)
  3066. {
  3067. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgEnumType>();
  3068. dbgType->mScope = scope;
  3069. dbgType->mName = name;
  3070. dbgType->mSize = (int)(sizeInBits / 8);
  3071. dbgType->mAlign = (int)(alignInBits / 8);
  3072. SetResult(curId, dbgType);
  3073. }
  3074. else
  3075. NotImpl();
  3076. }
  3077. break;
  3078. case BeIRCmd_DbgSetTypeSize:
  3079. {
  3080. CMD_PARAM(BeMDNode*, mdType);
  3081. CMD_PARAM(int64, sizeInBits);
  3082. CMD_PARAM(int64, alignInBits);
  3083. if (auto dbgType = BeValueDynCast<BeDbgType>(mdType))
  3084. {
  3085. dbgType->mSize = (int)(sizeInBits / 8);
  3086. dbgType->mAlign = (int)(alignInBits / 8);
  3087. }
  3088. }
  3089. break;
  3090. case BfIRCmd_DbgReplaceAllUses:
  3091. {
  3092. CMD_PARAM(BeMDNode*, diPrevNode);
  3093. CMD_PARAM(BeMDNode*, diNewNode);
  3094. /*diPrevNode->replaceAllUsesWith(diNewNode); */
  3095. NotImpl();
  3096. }
  3097. break;
  3098. case BfIRCmd_DbgDeleteTemporary:
  3099. {
  3100. CMD_PARAM(BeMDNode*, diNode);
  3101. /*BeMDNode::deleteTemporary(diNode);*/
  3102. NotImpl();
  3103. }
  3104. break;
  3105. case BfIRCmd_DbgMakePermanent:
  3106. {
  3107. CMD_PARAM(BeMDNode*, diNode);
  3108. CMD_PARAM(BeMDNode*, diBaseType);
  3109. CMD_PARAM(CmdParamVec<BeMDNode*>, members);
  3110. if (auto dbgType = BeValueDynCast<BeDbgStructType>(diNode))
  3111. {
  3112. dbgType->SetMembers(members);
  3113. }
  3114. else if (auto dbgType = BeValueDynCast<BeDbgEnumType>(diNode))
  3115. {
  3116. dbgType->mElementType = BeValueDynCast<BeDbgType>(diBaseType);
  3117. dbgType->SetMembers(members);
  3118. }
  3119. else
  3120. NotImpl();
  3121. SetResult(curId, diNode);
  3122. break;
  3123. }
  3124. case BfIRCmd_CreateEnumerator:
  3125. {
  3126. CMD_PARAM(String, name);
  3127. CMD_PARAM(int64, val);
  3128. auto enumValue = mBeModule->mOwnedValues.Alloc<BeDbgEnumMember>();
  3129. enumValue->mName = name;
  3130. enumValue->mValue = val;
  3131. SetResult(curId, enumValue);
  3132. }
  3133. break;
  3134. case BfIRCmd_DbgCreateMemberType:
  3135. {
  3136. CMD_PARAM(BeMDNode*, scope);
  3137. CMD_PARAM(String, name);
  3138. CMD_PARAM(BeMDNode*, file);
  3139. CMD_PARAM(int, lineNumber);
  3140. CMD_PARAM(int64, sizeInBits);
  3141. CMD_PARAM(int64, alignInBits);
  3142. CMD_PARAM(int64, offsetInBits);
  3143. CMD_PARAM(int, flags);
  3144. CMD_PARAM(BeMDNode*, type);
  3145. BF_ASSERT(type != NULL);
  3146. auto dbgMember = mBeModule->mOwnedValues.Alloc<BeDbgStructMember>();
  3147. dbgMember->mName = name;
  3148. dbgMember->mType = BeValueDynCast<BeDbgType>(type);
  3149. dbgMember->mOffset = (int)(offsetInBits / 8);
  3150. dbgMember->mFlags = flags;
  3151. SetResult(curId, dbgMember);
  3152. }
  3153. break;
  3154. case BfIRCmd_DbgStaticCreateMemberType:
  3155. {
  3156. CMD_PARAM(BeMDNode*, scope);
  3157. CMD_PARAM(String, name);
  3158. CMD_PARAM(BeMDNode*, file);
  3159. CMD_PARAM(int, lineNumber);
  3160. CMD_PARAM(BeMDNode*, type);
  3161. CMD_PARAM(int, flags);
  3162. CMD_PARAM(BeConstant*, val);
  3163. BF_ASSERT(type != NULL);
  3164. auto dbgMember = mBeModule->mOwnedValues.Alloc<BeDbgStructMember>();
  3165. dbgMember->mName = name;
  3166. dbgMember->mType = BeValueDynCast<BeDbgType>(type);
  3167. dbgMember->mOffset = -1;
  3168. dbgMember->mStaticValue = val;
  3169. dbgMember->mFlags = flags;
  3170. dbgMember->mIsStatic = true;
  3171. SetResult(curId, dbgMember);
  3172. }
  3173. break;
  3174. case BfIRCmd_DbgCreateInheritance:
  3175. {
  3176. CMD_PARAM(BeMDNode*, type);
  3177. CMD_PARAM(BeMDNode*, baseType);
  3178. CMD_PARAM(int64, baseOffset);
  3179. CMD_PARAM(int, flags);
  3180. auto dbgInheritance = mBeModule->mAlloc.Alloc<BeDbgInheritance>();
  3181. dbgInheritance->mBaseType = BeValueDynCast<BeDbgType>(baseType);
  3182. SetResult(curId, dbgInheritance);
  3183. }
  3184. break;
  3185. case BfIRCmd_DbgCreateMethod:
  3186. {
  3187. CMD_PARAM(BeMDNode*, context);
  3188. CMD_PARAM(String, name);
  3189. CMD_PARAM(String, linkageName);
  3190. CMD_PARAM(BeMDNode*, file);
  3191. CMD_PARAM(int, lineNum);
  3192. CMD_PARAM(BeMDNode*, type);
  3193. CMD_PARAM(bool, isLocalToUnit);
  3194. CMD_PARAM(bool, isDefinition);
  3195. CMD_PARAM(int, vk);
  3196. CMD_PARAM(int, vIndex);
  3197. CMD_PARAM(BeMDNode*, vTableHolder);
  3198. CMD_PARAM(int, flags);
  3199. CMD_PARAM(bool, isOptimized);
  3200. CMD_PARAM(BeValue*, fn);
  3201. CMD_PARAM(CmdParamVec<BeMDNode*>, genericArgs);
  3202. CMD_PARAM(CmdParamVec<BeConstant*>, genericConstValueArgs);
  3203. auto dbgFunc = mBeModule->mOwnedValues.Alloc<BeDbgFunction>();
  3204. dbgFunc->mScope = context;
  3205. dbgFunc->mFile = (BeDbgFile*)file;
  3206. dbgFunc->mLine = lineNum - 1;
  3207. dbgFunc->mType = (BeDbgFunctionType*)type;
  3208. dbgFunc->mName = name;
  3209. dbgFunc->mLinkageName = linkageName;
  3210. dbgFunc->mValue = (BeFunction*)fn;
  3211. dbgFunc->mIsLocalToUnit = isLocalToUnit;
  3212. dbgFunc->mVK = vk;
  3213. dbgFunc->mVIndex = vIndex;
  3214. dbgFunc->mIsStaticMethod = (flags & llvm::DINode::FlagStaticMember) != 0;
  3215. dbgFunc->mFlags = flags;
  3216. for (auto arg : genericArgs)
  3217. {
  3218. BF_ASSERT(arg != NULL);
  3219. dbgFunc->mGenericArgs.Add(BeValueDynCast<BeDbgType>(arg));
  3220. }
  3221. for (auto genericConstValue : genericConstValueArgs)
  3222. dbgFunc->mGenericConstValueArgs.Add(genericConstValue);
  3223. if (dbgFunc->mValue != NULL)
  3224. dbgFunc->mValue->mDbgFunction = dbgFunc;
  3225. dbgFunc->mIdx = (int)mBeModule->mDbgModule->mFuncs.size();
  3226. mBeModule->mDbgModule->mFuncs.push_back(dbgFunc);
  3227. SetResult(curId, dbgFunc);
  3228. }
  3229. break;
  3230. case BfIRCmd_DbgCreateFunction:
  3231. {
  3232. CMD_PARAM(BeMDNode*, context);
  3233. CMD_PARAM(String, name);
  3234. CMD_PARAM(String, linkageName);
  3235. CMD_PARAM(BeMDNode*, file);
  3236. CMD_PARAM(int, lineNum);
  3237. CMD_PARAM(BeMDNode*, type);
  3238. CMD_PARAM(bool, isLocalToUnit);
  3239. CMD_PARAM(bool, isDefinition);
  3240. CMD_PARAM(int, scopeLine);
  3241. CMD_PARAM(int, flags);
  3242. CMD_PARAM(bool, isOptimized);
  3243. CMD_PARAM(BeValue*, fn);
  3244. auto dbgFunc = mBeModule->mOwnedValues.Alloc<BeDbgFunction>();
  3245. dbgFunc->mScope = context;
  3246. dbgFunc->mFile = (BeDbgFile*)file;
  3247. dbgFunc->mLine = lineNum - 1;
  3248. dbgFunc->mType = (BeDbgFunctionType*)type;
  3249. dbgFunc->mName = name;
  3250. dbgFunc->mLinkageName = linkageName;
  3251. dbgFunc->mValue = (BeFunction*)fn;
  3252. dbgFunc->mIsLocalToUnit = isLocalToUnit;
  3253. dbgFunc->mFlags = flags;
  3254. /*if (auto dbgStructType = BeValueDynCast<BeDbgStructType>(context))
  3255. {
  3256. // This will get added to the struct later
  3257. }
  3258. else
  3259. {
  3260. }*/
  3261. if (dbgFunc->mValue != NULL)
  3262. dbgFunc->mValue->mDbgFunction = dbgFunc;
  3263. dbgFunc->mIdx = (int)mBeModule->mDbgModule->mFuncs.size();
  3264. mBeModule->mDbgModule->mFuncs.push_back(dbgFunc);
  3265. SetResult(curId, dbgFunc);
  3266. }
  3267. break;
  3268. case BfIRCmd_DbgCreateParameterVariable:
  3269. {
  3270. CMD_PARAM(BeMDNode*, scope);
  3271. CMD_PARAM(String, name);
  3272. CMD_PARAM(int, argNo);
  3273. CMD_PARAM(BeMDNode*, file);
  3274. CMD_PARAM(int, lineNum);
  3275. CMD_PARAM(BeMDNode*, type);
  3276. CMD_PARAM(bool, alwaysPreserve);
  3277. CMD_PARAM(int, flags);
  3278. BF_ASSERT(type != NULL);
  3279. auto dbgFunc = (BeDbgFunction*)scope;
  3280. auto dbgVar = mBeModule->mOwnedValues.Alloc<BeDbgVariable>();
  3281. dbgVar->mName = name;
  3282. dbgVar->mType = type;
  3283. dbgVar->mParamNum = argNo - 1;
  3284. int argIdx = argNo - 1;
  3285. //for (int i = ; i <= argNo - 1; i++)
  3286. while (argIdx >= (int)dbgFunc->mVariables.size())
  3287. dbgFunc->mVariables.push_back(NULL);
  3288. if (dbgFunc->mVariables[argIdx] == NULL)
  3289. dbgFunc->mVariables[argIdx] = dbgVar;
  3290. else
  3291. {
  3292. BF_ASSERT(dbgFunc->mVariables[argIdx]->mParamNum == -1);
  3293. dbgFunc->mVariables.Insert(argIdx, dbgVar);
  3294. }
  3295. //mActiveFunction->mDbgFunction->mVariables.push_back(dbgVar);
  3296. //dbgVar->mValue = mBeModule->GetArgument(argNo - 1);
  3297. SetResult(curId, dbgVar);
  3298. }
  3299. break;
  3300. case BfIRCmd_DbgCreateSubroutineType:
  3301. {
  3302. CMD_PARAM(CmdParamVec<BeMDNode*>, elements);
  3303. auto dbgFuncType = mBeModule->mOwnedValues.Alloc<BeDbgFunctionType>();
  3304. if (!elements.empty())
  3305. {
  3306. dbgFuncType->mReturnType = BeValueDynCast<BeDbgType>(elements[0]);
  3307. for (int i = 1; i < (int)elements.size(); i++)
  3308. dbgFuncType->mParams.push_back(BeValueDynCast<BeDbgType>(elements[i]));
  3309. }
  3310. SetResult(curId, dbgFuncType);
  3311. }
  3312. break;
  3313. case BfIRCmd_DbgCreateAutoVariable:
  3314. {
  3315. CMD_PARAM(BeMDNode*, scope);
  3316. CMD_PARAM(String, name);
  3317. CMD_PARAM(BeMDNode*, file);
  3318. CMD_PARAM(int, lineNo);
  3319. CMD_PARAM(BeMDNode*, type);
  3320. CMD_PARAM(int, initType);
  3321. BF_ASSERT(type != NULL);
  3322. auto dbgVar = mBeModule->mOwnedValues.Alloc<BeDbgVariable>();
  3323. dbgVar->mName = name;
  3324. dbgVar->mType = type;
  3325. dbgVar->mScope = scope;
  3326. dbgVar->mInitType = (BfIRInitType)initType;
  3327. mActiveFunction->mDbgFunction->mVariables.push_back(dbgVar);
  3328. BF_ASSERT(name != "__CRASH_AUTOVARIABLE__");
  3329. SetResult(curId, dbgVar);
  3330. }
  3331. break;
  3332. case BfIRCmd_DbgInsertValueIntrinsic:
  3333. {
  3334. CMD_PARAM(BeValue*, val);
  3335. CMD_PARAM(BeMDNode*, varInfo);
  3336. auto dbgVar = BeValueDynCast<BeDbgVariable>(varInfo);
  3337. //dbgVar->mValue = val;
  3338. //dbgVar->mDeclDbgLoc = mBeModule->mCurDbgLoc;
  3339. if (val == NULL)
  3340. {
  3341. val = mBeModule->GetConstant(mBeContext->GetPrimitiveType(BeTypeCode_Int32), (int64)0);
  3342. }
  3343. auto inst = mBeModule->AllocInst<BeDbgDeclareInst>();
  3344. inst->mValue = val;
  3345. inst->mDbgVar = dbgVar;
  3346. inst->mIsValue = true;
  3347. SetResult(curId, inst);
  3348. }
  3349. break;
  3350. case BfIRCmd_DbgInsertDeclare:
  3351. {
  3352. CMD_PARAM(BeValue*, val);
  3353. CMD_PARAM(BeMDNode*, varInfo);
  3354. CMD_PARAM(BeValue*, insertBefore);
  3355. auto dbgVar = BeValueDynCast<BeDbgVariable>(varInfo);
  3356. //dbgVar->mValue = val;
  3357. //dbgVar->mDeclDbgLoc = mBeModule->mCurDbgLoc;
  3358. auto inst = mBeModule->mAlloc.Alloc<BeDbgDeclareInst>();
  3359. inst->mValue = val;
  3360. inst->mDbgVar = dbgVar;
  3361. inst->mIsValue = false;
  3362. if (insertBefore == NULL)
  3363. mBeModule->AddInst(inst);
  3364. else
  3365. NotImpl();
  3366. SetResult(curId, inst);
  3367. }
  3368. break;
  3369. case BfIRCmd_DbgLifetimeEnd:
  3370. {
  3371. CMD_PARAM(BeMDNode*, varInfo);
  3372. auto inst = mBeModule->AllocInst<BeLifetimeEndInst>();
  3373. inst->mPtr = varInfo;
  3374. }
  3375. break;
  3376. case BfIRCmd_DbgCreateGlobalVariable:
  3377. {
  3378. CMD_PARAM(BeMDNode*, context);
  3379. CMD_PARAM(String, name);
  3380. CMD_PARAM(String, linkageName);
  3381. CMD_PARAM(BeMDNode*, file);
  3382. CMD_PARAM(int, lineNum);
  3383. CMD_PARAM(BeMDNode*, type);
  3384. CMD_PARAM(bool, isLocalToUnit);
  3385. CMD_PARAM(BeConstant*, val);
  3386. CMD_PARAM(BeMDNode*, decl);
  3387. auto dbgGlobalVariable = mBeModule->mDbgModule->mGlobalVariables.Alloc();
  3388. dbgGlobalVariable->mContext = context;
  3389. dbgGlobalVariable->mName = name;
  3390. dbgGlobalVariable->mLinkageName = linkageName;
  3391. dbgGlobalVariable->mFile = (BeDbgFile*)file;
  3392. dbgGlobalVariable->mLineNum = lineNum;
  3393. dbgGlobalVariable->mType = type;
  3394. dbgGlobalVariable->mIsLocalToUnit = isLocalToUnit;
  3395. dbgGlobalVariable->mValue = val;
  3396. dbgGlobalVariable->mDecl = decl;
  3397. SetResult(curId, dbgGlobalVariable);
  3398. }
  3399. break;
  3400. case BfIRCmd_DbgCreateLexicalBlock:
  3401. {
  3402. CMD_PARAM(BeMDNode*, scope);
  3403. CMD_PARAM(BeMDNode*, file);
  3404. CMD_PARAM(int, lineNum);
  3405. CMD_PARAM(int, col);
  3406. auto dbgLexicalBlock = mBeModule->mOwnedValues.Alloc<BeDbgLexicalBlock>();
  3407. BF_ASSERT(BeValueDynCast<BeDbgFile>(file) != NULL);
  3408. dbgLexicalBlock->mFile = (BeDbgFile*)file;
  3409. dbgLexicalBlock->mScope = scope;
  3410. dbgLexicalBlock->mId = mBeModule->mCurLexBlockId++;
  3411. SetResult(curId, dbgLexicalBlock);
  3412. }
  3413. break;
  3414. case BfIRCmd_DbgCreateAnnotation:
  3415. {
  3416. CMD_PARAM(BeMDNode*, scope);
  3417. CMD_PARAM(String, name);
  3418. CMD_PARAM(BeValue*, value);
  3419. if (auto dbgFunc = BeValueDynCast<BeDbgFunction>(scope))
  3420. {
  3421. auto beType = value->GetType();
  3422. BeDbgType** dbgTypePtr;
  3423. if (mOnDemandTypeMap.TryAdd(beType, NULL, &dbgTypePtr))
  3424. {
  3425. auto dbgType = mBeModule->mDbgModule->mTypes.Alloc<BeDbgBasicType>();
  3426. dbgType->mSize = beType->mSize;
  3427. dbgType->mAlign = beType->mAlign;
  3428. dbgType->mEncoding = llvm::dwarf::DW_ATE_signed;
  3429. *dbgTypePtr = dbgType;
  3430. }
  3431. auto dbgVar = mBeModule->mOwnedValues.Alloc<BeDbgVariable>();
  3432. dbgVar->mName = "#" + name;
  3433. dbgVar->mType = *dbgTypePtr;
  3434. dbgVar->mValue = value;
  3435. dbgVar->mScope = scope;
  3436. dbgFunc->mVariables.Add(dbgVar);
  3437. auto inst = mBeModule->AllocInst<BeDbgDeclareInst>();
  3438. inst->mValue = value;
  3439. inst->mDbgVar = dbgVar;
  3440. inst->mIsValue = true;
  3441. }
  3442. else
  3443. NotImpl();
  3444. }
  3445. break;
  3446. default:
  3447. BF_FATAL("Unhandled");
  3448. break;
  3449. }
  3450. #ifdef CODEGEN_TRACK
  3451. gBEMemReporter.EndSection();
  3452. gBEMemReporterSize += mStream->GetReadPos() - curId;
  3453. #endif
  3454. }
  3455. void BeIRCodeGen::SetConfigConst(int idx, int value)
  3456. {
  3457. BF_ASSERT(idx == (int)mConfigConsts.size());
  3458. mConfigConsts.Add(value);
  3459. }
  3460. BeValue* BeIRCodeGen::GetBeValue(int id)
  3461. {
  3462. auto& result = mResults[id];
  3463. BF_ASSERT(result.mKind == BeIRCodeGenEntryKind_Value);
  3464. #ifdef BE_EXTRA_CHECKS
  3465. BF_ASSERT(!result.mBeValue->mLifetimeEnded);
  3466. BF_ASSERT(!result.mBeValue->mWasRemoved);
  3467. #endif
  3468. return result.mBeValue;
  3469. }
  3470. BeType* BeIRCodeGen::GetBeType(int id)
  3471. {
  3472. auto& result = mResults[id];
  3473. BF_ASSERT(result.mKind == BeIRCodeGenEntryKind_Type);
  3474. return result.mBeType;
  3475. }
  3476. BeBlock* BeIRCodeGen::GetBeBlock(int id)
  3477. {
  3478. auto& result = mResults[id];
  3479. BF_ASSERT(result.mKind == BeIRCodeGenEntryKind_Block);
  3480. return result.mBeBlock;
  3481. }
  3482. BeMDNode* BeIRCodeGen::GetBeMetadata(int id)
  3483. {
  3484. auto& result = mResults[id];
  3485. BF_ASSERT(result.mKind == BeIRCodeGenEntryKind_Metadata);
  3486. return result.mBeMetadata;
  3487. }
  3488. BeType* BeIRCodeGen::GetBeTypeById(int id)
  3489. {
  3490. return GetTypeEntry(id).mBeType;
  3491. }
  3492. BeState BeIRCodeGen::GetState()
  3493. {
  3494. BeState state;
  3495. state.mActiveFunction = mActiveFunction;
  3496. state.mSavedDebugLocs = mSavedDebugLocs;
  3497. state.mHasDebugLoc = mHasDebugLoc;
  3498. state.mActiveBlock = mBeModule->mActiveBlock;
  3499. state.mInsertPos = mBeModule->mInsertPos;
  3500. state.mCurDbgLoc = mBeModule->mCurDbgLoc;
  3501. state.mPrevDbgLocInline = mBeModule->mPrevDbgLocInline;
  3502. state.mLastDbgLoc = mBeModule->mLastDbgLoc;
  3503. return state;
  3504. }
  3505. void BeIRCodeGen::SetState(const BeState& state)
  3506. {
  3507. mActiveFunction = state.mActiveFunction;
  3508. mBeModule->mActiveFunction = mActiveFunction;
  3509. mSavedDebugLocs = state.mSavedDebugLocs;
  3510. mHasDebugLoc = state.mHasDebugLoc;
  3511. mBeModule->mActiveBlock = state.mActiveBlock;
  3512. mBeModule->mInsertPos = state.mInsertPos;
  3513. mBeModule->mCurDbgLoc = state.mCurDbgLoc;
  3514. mBeModule->mPrevDbgLocInline = state.mPrevDbgLocInline;
  3515. mBeModule->mLastDbgLoc = state.mLastDbgLoc;
  3516. }