src/cmd/internal/obj/arm64/obj7.go

   1 // cmd/7l/noop.c, cmd/7l/obj.c, cmd/ld/pass.c from Vita Nuova.
   2 // https://code.google.com/p/ken-cc/source/browse/
   3 //
   4 //      Copyright © 1994-1999 Lucent Technologies Inc. All rights reserved.
   5 //      Portions Copyright © 1995-1997 C H Forsyth (forsyth@terzarima.net)
   6 //      Portions Copyright © 1997-1999 Vita Nuova Limited
   7 //      Portions Copyright © 2000-2007 Vita Nuova Holdings Limited (www.vitanuova.com)
   8 //      Portions Copyright © 2004,2006 Bruce Ellis
   9 //      Portions Copyright © 2005-2007 C H Forsyth (forsyth@terzarima.net)
  10 //      Revisions Copyright © 2000-2007 Lucent Technologies Inc. and others
  11 //      Portions Copyright © 2009 The Go Authors. All rights reserved.
  12 //
  13 // Permission is hereby granted, free of charge, to any person obtaining a copy
  14 // of this software and associated documentation files (the "Software"), to deal
  15 // in the Software without restriction, including without limitation the rights
  16 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  17 // copies of the Software, and to permit persons to whom the Software is
  18 // furnished to do so, subject to the following conditions:
  19 //
  20 // The above copyright notice and this permission notice shall be included in
  21 // all copies or substantial portions of the Software.
  22 //
  23 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  24 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  25 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
  26 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  27 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  28 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  29 // THE SOFTWARE.
  30
  31 package arm64
  32
  33 import (
  34         "cmd/internal/obj"
  35         "cmd/internal/objabi"
  36         "cmd/internal/src"
  37         "cmd/internal/sys"
  38         "internal/buildcfg"
  39         "log"
  40         "math"
  41 )
  42
  43 var complements = []obj.As{
  44         AADD:  ASUB,
  45         AADDW: ASUBW,
  46         ASUB:  AADD,
  47         ASUBW: AADDW,
  48         ACMP:  ACMN,
  49         ACMPW: ACMNW,
  50         ACMN:  ACMP,
  51         ACMNW: ACMPW,
  52 }
  53
  54 // noZRreplace is the set of instructions for which $0 in the To operand
  55 // should NOT be replaced with REGZERO.
  56 var noZRreplace = map[obj.As]bool{
  57         APRFM: true,
  58 }
  59
  60 func (c *ctxt7) stacksplit(p *obj.Prog, framesize int32) *obj.Prog {
  61         if c.ctxt.Flag_maymorestack != "" {
  62                 p = c.cursym.Func().SpillRegisterArgs(p, c.newprog)
  63
  64                 // Save LR and make room for FP, REGCTXT. Leave room
  65                 // for caller's saved FP.
  66                 const frameSize = 32
  67                 p = obj.Appendp(p, c.newprog)
  68                 p.As = AMOVD
  69                 p.From.Type = obj.TYPE_REG
  70                 p.From.Reg = REGLINK
  71                 p.To.Type = obj.TYPE_MEM
  72                 p.Scond = C_XPRE
  73                 p.To.Offset = -frameSize
  74                 p.To.Reg = REGSP
  75                 p.Spadj = frameSize
  76
  77                 // Save FP.
  78                 p = obj.Appendp(p, c.newprog)
  79                 p.As = AMOVD
  80                 p.From.Type = obj.TYPE_REG
  81                 p.From.Reg = REGFP
  82                 p.To.Type = obj.TYPE_MEM
  83                 p.To.Reg = REGSP
  84                 p.To.Offset = -8
  85
  86                 p = obj.Appendp(p, c.newprog)
  87                 p.As = ASUB
  88                 p.From.Type = obj.TYPE_CONST
  89                 p.From.Offset = 8
  90                 p.Reg = REGSP
  91                 p.To.Type = obj.TYPE_REG
  92                 p.To.Reg = REGFP
  93
  94                 // Save REGCTXT (for simplicity we do this whether or
  95                 // not we need it.)
  96                 p = obj.Appendp(p, c.newprog)
  97                 p.As = AMOVD
  98                 p.From.Type = obj.TYPE_REG
  99                 p.From.Reg = REGCTXT
 100                 p.To.Type = obj.TYPE_MEM
 101                 p.To.Reg = REGSP
 102                 p.To.Offset = 8
 103
 104                 // BL maymorestack
 105                 p = obj.Appendp(p, c.newprog)
 106                 p.As = ABL
 107                 p.To.Type = obj.TYPE_BRANCH
 108                 // See ../x86/obj6.go
 109                 p.To.Sym = c.ctxt.LookupABI(c.ctxt.Flag_maymorestack, c.cursym.ABI())
 110
 111                 // Restore REGCTXT.
 112                 p = obj.Appendp(p, c.newprog)
 113                 p.As = AMOVD
 114                 p.From.Type = obj.TYPE_MEM
 115                 p.From.Reg = REGSP
 116                 p.From.Offset = 8
 117                 p.To.Type = obj.TYPE_REG
 118                 p.To.Reg = REGCTXT
 119
 120                 // Restore FP.
 121                 p = obj.Appendp(p, c.newprog)
 122                 p.As = AMOVD
 123                 p.From.Type = obj.TYPE_MEM
 124                 p.From.Reg = REGSP
 125                 p.From.Offset = -8
 126                 p.To.Type = obj.TYPE_REG
 127                 p.To.Reg = REGFP
 128
 129                 // Restore LR and SP.
 130                 p = obj.Appendp(p, c.newprog)
 131                 p.As = AMOVD
 132                 p.From.Type = obj.TYPE_MEM
 133                 p.Scond = C_XPOST
 134                 p.From.Offset = frameSize
 135                 p.From.Reg = REGSP
 136                 p.To.Type = obj.TYPE_REG
 137                 p.To.Reg = REGLINK
 138                 p.Spadj = -frameSize
 139
 140                 p = c.cursym.Func().UnspillRegisterArgs(p, c.newprog)
 141         }
 142
 143         // Jump back to here after morestack returns.
 144         startPred := p
 145
 146         // MOV  g_stackguard(g), RT1
 147         p = obj.Appendp(p, c.newprog)
 148
 149         p.As = AMOVD
 150         p.From.Type = obj.TYPE_MEM
 151         p.From.Reg = REGG
 152         p.From.Offset = 2 * int64(c.ctxt.Arch.PtrSize) // G.stackguard0
 153         if c.cursym.CFunc() {
 154                 p.From.Offset = 3 * int64(c.ctxt.Arch.PtrSize) // G.stackguard1
 155         }
 156         p.To.Type = obj.TYPE_REG
 157         p.To.Reg = REGRT1
 158
 159         // Mark the stack bound check and morestack call async nonpreemptible.
 160         // If we get preempted here, when resumed the preemption request is
 161         // cleared, but we'll still call morestack, which will double the stack
 162         // unnecessarily. See issue #35470.
 163         p = c.ctxt.StartUnsafePoint(p, c.newprog)
 164
 165         q := (*obj.Prog)(nil)
 166         if framesize <= objabi.StackSmall {
 167                 // small stack: SP < stackguard
 168                 //      CMP     stackguard, SP
 169
 170                 p = obj.Appendp(p, c.newprog)
 171                 p.As = ACMP
 172                 p.From.Type = obj.TYPE_REG
 173                 p.From.Reg = REGRT1
 174                 p.Reg = REGSP
 175         } else if framesize <= objabi.StackBig {
 176                 // large stack: SP-framesize < stackguard-StackSmall
 177                 //      SUB     $(framesize-StackSmall), SP, RT2
 178                 //      CMP     stackguard, RT2
 179                 p = obj.Appendp(p, c.newprog)
 180
 181                 p.As = ASUB
 182                 p.From.Type = obj.TYPE_CONST
 183                 p.From.Offset = int64(framesize) - objabi.StackSmall
 184                 p.Reg = REGSP
 185                 p.To.Type = obj.TYPE_REG
 186                 p.To.Reg = REGRT2
 187
 188                 p = obj.Appendp(p, c.newprog)
 189                 p.As = ACMP
 190                 p.From.Type = obj.TYPE_REG
 191                 p.From.Reg = REGRT1
 192                 p.Reg = REGRT2
 193         } else {
 194                 // Such a large stack we need to protect against underflow.
 195                 // The runtime guarantees SP > objabi.StackBig, but
 196                 // framesize is large enough that SP-framesize may
 197                 // underflow, causing a direct comparison with the
 198                 // stack guard to incorrectly succeed. We explicitly
 199                 // guard against underflow.
 200                 //
 201                 //      SUBS    $(framesize-StackSmall), SP, RT2
 202                 //      // On underflow, jump to morestack
 203                 //      BLO     label_of_call_to_morestack
 204                 //      CMP     stackguard, RT2
 205
 206                 p = obj.Appendp(p, c.newprog)
 207                 p.As = ASUBS
 208                 p.From.Type = obj.TYPE_CONST
 209                 p.From.Offset = int64(framesize) - objabi.StackSmall
 210                 p.Reg = REGSP
 211                 p.To.Type = obj.TYPE_REG
 212                 p.To.Reg = REGRT2
 213
 214                 p = obj.Appendp(p, c.newprog)
 215                 q = p
 216                 p.As = ABLO
 217                 p.To.Type = obj.TYPE_BRANCH
 218
 219                 p = obj.Appendp(p, c.newprog)
 220                 p.As = ACMP
 221                 p.From.Type = obj.TYPE_REG
 222                 p.From.Reg = REGRT1
 223                 p.Reg = REGRT2
 224         }
 225
 226         // BLS  do-morestack
 227         bls := obj.Appendp(p, c.newprog)
 228         bls.As = ABLS
 229         bls.To.Type = obj.TYPE_BRANCH
 230
 231         end := c.ctxt.EndUnsafePoint(bls, c.newprog, -1)
 232
 233         var last *obj.Prog
 234         for last = c.cursym.Func().Text; last.Link != nil; last = last.Link {
 235         }
 236
 237         // Now we are at the end of the function, but logically
 238         // we are still in function prologue. We need to fix the
 239         // SP data and PCDATA.
 240         spfix := obj.Appendp(last, c.newprog)
 241         spfix.As = obj.ANOP
 242         spfix.Spadj = -framesize
 243
 244         pcdata := c.ctxt.EmitEntryStackMap(c.cursym, spfix, c.newprog)
 245         pcdata = c.ctxt.StartUnsafePoint(pcdata, c.newprog)
 246
 247         if q != nil {
 248                 q.To.SetTarget(pcdata)
 249         }
 250         bls.To.SetTarget(pcdata)
 251
 252         spill := c.cursym.Func().SpillRegisterArgs(pcdata, c.newprog)
 253
 254         // MOV  LR, R3
 255         movlr := obj.Appendp(spill, c.newprog)
 256         movlr.As = AMOVD
 257         movlr.From.Type = obj.TYPE_REG
 258         movlr.From.Reg = REGLINK
 259         movlr.To.Type = obj.TYPE_REG
 260         movlr.To.Reg = REG_R3
 261
 262         debug := movlr
 263         if false {
 264                 debug = obj.Appendp(debug, c.newprog)
 265                 debug.As = AMOVD
 266                 debug.From.Type = obj.TYPE_CONST
 267                 debug.From.Offset = int64(framesize)
 268                 debug.To.Type = obj.TYPE_REG
 269                 debug.To.Reg = REGTMP
 270         }
 271
 272         // BL   runtime.morestack(SB)
 273         call := obj.Appendp(debug, c.newprog)
 274         call.As = ABL
 275         call.To.Type = obj.TYPE_BRANCH
 276         morestack := "runtime.morestack"
 277         switch {
 278         case c.cursym.CFunc():
 279                 morestack = "runtime.morestackc"
 280         case !c.cursym.Func().Text.From.Sym.NeedCtxt():
 281                 morestack = "runtime.morestack_noctxt"
 282         }
 283         call.To.Sym = c.ctxt.Lookup(morestack)
 284
 285         unspill := c.cursym.Func().UnspillRegisterArgs(call, c.newprog)
 286         pcdata = c.ctxt.EndUnsafePoint(unspill, c.newprog, -1)
 287
 288         // B    start
 289         jmp := obj.Appendp(pcdata, c.newprog)
 290         jmp.As = AB
 291         jmp.To.Type = obj.TYPE_BRANCH
 292         jmp.To.SetTarget(startPred.Link)
 293         jmp.Spadj = +framesize
 294
 295         return end
 296 }
 297
 298 func progedit(ctxt *obj.Link, p *obj.Prog, newprog obj.ProgAlloc) {
 299         c := ctxt7{ctxt: ctxt, newprog: newprog}
 300
 301         p.From.Class = 0
 302         p.To.Class = 0
 303
 304         // $0 results in C_ZCON, which matches both C_REG and various
 305         // C_xCON, however the C_REG cases in asmout don't expect a
 306         // constant, so they will use the register fields and assemble
 307         // a R0. To prevent that, rewrite $0 as ZR.
 308         if p.From.Type == obj.TYPE_CONST && p.From.Offset == 0 {
 309                 p.From.Type = obj.TYPE_REG
 310                 p.From.Reg = REGZERO
 311         }
 312         if p.To.Type == obj.TYPE_CONST && p.To.Offset == 0 && !noZRreplace[p.As] {
 313                 p.To.Type = obj.TYPE_REG
 314                 p.To.Reg = REGZERO
 315         }
 316
 317         // Rewrite BR/BL to symbol as TYPE_BRANCH.
 318         switch p.As {
 319         case AB,
 320                 ABL,
 321                 obj.ARET,
 322                 obj.ADUFFZERO,
 323                 obj.ADUFFCOPY:
 324                 if p.To.Sym != nil {
 325                         p.To.Type = obj.TYPE_BRANCH
 326                 }
 327                 break
 328         }
 329
 330         // Rewrite float constants to values stored in memory.
 331         switch p.As {
 332         case AFMOVS:
 333                 if p.From.Type == obj.TYPE_FCONST {
 334                         f64 := p.From.Val.(float64)
 335                         f32 := float32(f64)
 336                         if c.chipfloat7(f64) > 0 {
 337                                 break
 338                         }
 339                         if math.Float32bits(f32) == 0 {
 340                                 p.From.Type = obj.TYPE_REG
 341                                 p.From.Reg = REGZERO
 342                                 break
 343                         }
 344                         p.From.Type = obj.TYPE_MEM
 345                         p.From.Sym = c.ctxt.Float32Sym(f32)
 346                         p.From.Name = obj.NAME_EXTERN
 347                         p.From.Offset = 0
 348                 }
 349
 350         case AFMOVD:
 351                 if p.From.Type == obj.TYPE_FCONST {
 352                         f64 := p.From.Val.(float64)
 353                         if c.chipfloat7(f64) > 0 {
 354                                 break
 355                         }
 356                         if math.Float64bits(f64) == 0 {
 357                                 p.From.Type = obj.TYPE_REG
 358                                 p.From.Reg = REGZERO
 359                                 break
 360                         }
 361                         p.From.Type = obj.TYPE_MEM
 362                         p.From.Sym = c.ctxt.Float64Sym(f64)
 363                         p.From.Name = obj.NAME_EXTERN
 364                         p.From.Offset = 0
 365                 }
 366
 367                 break
 368         }
 369
 370         // Rewrite negative immediates as positive immediates with
 371         // complementary instruction.
 372         switch p.As {
 373         case AADD, ASUB, ACMP, ACMN:
 374                 if p.From.Type == obj.TYPE_CONST && p.From.Offset < 0 && p.From.Offset != -1<<63 {
 375                         p.From.Offset = -p.From.Offset
 376                         p.As = complements[p.As]
 377                 }
 378         case AADDW, ASUBW, ACMPW, ACMNW:
 379                 if p.From.Type == obj.TYPE_CONST && p.From.Offset < 0 && int32(p.From.Offset) != -1<<31 {
 380                         p.From.Offset = -p.From.Offset
 381                         p.As = complements[p.As]
 382                 }
 383         }
 384
 385         // For 32-bit instruction with constant, rewrite
 386         // the high 32-bit to be a repetition of the low
 387         // 32-bit, so that the BITCON test can be shared
 388         // for both 32-bit and 64-bit. 32-bit ops will
 389         // zero the high 32-bit of the destination register
 390         // anyway.
 391         // For MOVW, the destination register can't be ZR,
 392         // so don't bother rewriting it in this situation.
 393         if (isANDWop(p.As) || isADDWop(p.As) || p.As == AMOVW && p.To.Reg != REGZERO) && p.From.Type == obj.TYPE_CONST {
 394                 v := p.From.Offset & 0xffffffff
 395                 p.From.Offset = v | v<<32
 396         }
 397
 398         if c.ctxt.Flag_dynlink {
 399                 c.rewriteToUseGot(p)
 400         }
 401 }
 402
 403 // Rewrite p, if necessary, to access global data via the global offset table.
 404 func (c *ctxt7) rewriteToUseGot(p *obj.Prog) {
 405         if p.As == obj.ADUFFCOPY || p.As == obj.ADUFFZERO {
 406                 //     ADUFFxxx $offset
 407                 // becomes
 408                 //     MOVD runtime.duffxxx@GOT, REGTMP
 409                 //     ADD $offset, REGTMP
 410                 //     CALL REGTMP
 411                 var sym *obj.LSym
 412                 if p.As == obj.ADUFFZERO {
 413                         sym = c.ctxt.LookupABI("runtime.duffzero", obj.ABIInternal)
 414                 } else {
 415                         sym = c.ctxt.LookupABI("runtime.duffcopy", obj.ABIInternal)
 416                 }
 417                 offset := p.To.Offset
 418                 p.As = AMOVD
 419                 p.From.Type = obj.TYPE_MEM
 420                 p.From.Name = obj.NAME_GOTREF
 421                 p.From.Sym = sym
 422                 p.To.Type = obj.TYPE_REG
 423                 p.To.Reg = REGTMP
 424                 p.To.Name = obj.NAME_NONE
 425                 p.To.Offset = 0
 426                 p.To.Sym = nil
 427                 p1 := obj.Appendp(p, c.newprog)
 428                 p1.As = AADD
 429                 p1.From.Type = obj.TYPE_CONST
 430                 p1.From.Offset = offset
 431                 p1.To.Type = obj.TYPE_REG
 432                 p1.To.Reg = REGTMP
 433                 p2 := obj.Appendp(p1, c.newprog)
 434                 p2.As = obj.ACALL
 435                 p2.To.Type = obj.TYPE_REG
 436                 p2.To.Reg = REGTMP
 437         }
 438
 439         // We only care about global data: NAME_EXTERN means a global
 440         // symbol in the Go sense, and p.Sym.Local is true for a few
 441         // internally defined symbols.
 442         if p.From.Type == obj.TYPE_ADDR && p.From.Name == obj.NAME_EXTERN && !p.From.Sym.Local() {
 443                 // MOVD $sym, Rx becomes MOVD sym@GOT, Rx
 444                 // MOVD $sym+<off>, Rx becomes MOVD sym@GOT, Rx; ADD <off>, Rx
 445                 if p.As != AMOVD {
 446                         c.ctxt.Diag("do not know how to handle TYPE_ADDR in %v with -dynlink", p)
 447                 }
 448                 if p.To.Type != obj.TYPE_REG {
 449                         c.ctxt.Diag("do not know how to handle LEAQ-type insn to non-register in %v with -dynlink", p)
 450                 }
 451                 p.From.Type = obj.TYPE_MEM
 452                 p.From.Name = obj.NAME_GOTREF
 453                 if p.From.Offset != 0 {
 454                         q := obj.Appendp(p, c.newprog)
 455                         q.As = AADD
 456                         q.From.Type = obj.TYPE_CONST
 457                         q.From.Offset = p.From.Offset
 458                         q.To = p.To
 459                         p.From.Offset = 0
 460                 }
 461         }
 462         if p.GetFrom3() != nil && p.GetFrom3().Name == obj.NAME_EXTERN {
 463                 c.ctxt.Diag("don't know how to handle %v with -dynlink", p)
 464         }
 465         var source *obj.Addr
 466         // MOVx sym, Ry becomes MOVD sym@GOT, REGTMP; MOVx (REGTMP), Ry
 467         // MOVx Ry, sym becomes MOVD sym@GOT, REGTMP; MOVD Ry, (REGTMP)
 468         // An addition may be inserted between the two MOVs if there is an offset.
 469         if p.From.Name == obj.NAME_EXTERN && !p.From.Sym.Local() {
 470                 if p.To.Name == obj.NAME_EXTERN && !p.To.Sym.Local() {
 471                         c.ctxt.Diag("cannot handle NAME_EXTERN on both sides in %v with -dynlink", p)
 472                 }
 473                 source = &p.From
 474         } else if p.To.Name == obj.NAME_EXTERN && !p.To.Sym.Local() {
 475                 source = &p.To
 476         } else {
 477                 return
 478         }
 479         if p.As == obj.ATEXT || p.As == obj.AFUNCDATA || p.As == obj.ACALL || p.As == obj.ARET || p.As == obj.AJMP {
 480                 return
 481         }
 482         if source.Sym.Type == objabi.STLSBSS {
 483                 return
 484         }
 485         if source.Type != obj.TYPE_MEM {
 486                 c.ctxt.Diag("don't know how to handle %v with -dynlink", p)
 487         }
 488         p1 := obj.Appendp(p, c.newprog)
 489         p2 := obj.Appendp(p1, c.newprog)
 490         p1.As = AMOVD
 491         p1.From.Type = obj.TYPE_MEM
 492         p1.From.Sym = source.Sym
 493         p1.From.Name = obj.NAME_GOTREF
 494         p1.To.Type = obj.TYPE_REG
 495         p1.To.Reg = REGTMP
 496
 497         p2.As = p.As
 498         p2.From = p.From
 499         p2.To = p.To
 500         if p.From.Name == obj.NAME_EXTERN {
 501                 p2.From.Reg = REGTMP
 502                 p2.From.Name = obj.NAME_NONE
 503                 p2.From.Sym = nil
 504         } else if p.To.Name == obj.NAME_EXTERN {
 505                 p2.To.Reg = REGTMP
 506                 p2.To.Name = obj.NAME_NONE
 507                 p2.To.Sym = nil
 508         } else {
 509                 return
 510         }
 511         obj.Nopout(p)
 512 }
 513
 514 func preprocess(ctxt *obj.Link, cursym *obj.LSym, newprog obj.ProgAlloc) {
 515         if cursym.Func().Text == nil || cursym.Func().Text.Link == nil {
 516                 return
 517         }
 518
 519         c := ctxt7{ctxt: ctxt, newprog: newprog, cursym: cursym}
 520
 521         p := c.cursym.Func().Text
 522         textstksiz := p.To.Offset
 523         if textstksiz == -8 {
 524                 // Historical way to mark NOFRAME.
 525                 p.From.Sym.Set(obj.AttrNoFrame, true)
 526                 textstksiz = 0
 527         }
 528         if textstksiz < 0 {
 529                 c.ctxt.Diag("negative frame size %d - did you mean NOFRAME?", textstksiz)
 530         }
 531         if p.From.Sym.NoFrame() {
 532                 if textstksiz != 0 {
 533                         c.ctxt.Diag("NOFRAME functions must have a frame size of 0, not %d", textstksiz)
 534                 }
 535         }
 536
 537         c.cursym.Func().Args = p.To.Val.(int32)
 538         c.cursym.Func().Locals = int32(textstksiz)
 539
 540         /*
 541          * find leaf subroutines
 542          */
 543         for p := c.cursym.Func().Text; p != nil; p = p.Link {
 544                 switch p.As {
 545                 case obj.ATEXT:
 546                         p.Mark |= LEAF
 547
 548                 case ABL,
 549                         obj.ADUFFZERO,
 550                         obj.ADUFFCOPY:
 551                         c.cursym.Func().Text.Mark &^= LEAF
 552                 }
 553         }
 554
 555         var q *obj.Prog
 556         var q1 *obj.Prog
 557         var retjmp *obj.LSym
 558         for p := c.cursym.Func().Text; p != nil; p = p.Link {
 559                 o := p.As
 560                 switch o {
 561                 case obj.ATEXT:
 562                         c.cursym.Func().Text = p
 563                         c.autosize = int32(textstksiz)
 564
 565                         if p.Mark&LEAF != 0 && c.autosize == 0 {
 566                                 // A leaf function with no locals has no frame.
 567                                 p.From.Sym.Set(obj.AttrNoFrame, true)
 568                         }
 569
 570                         if !p.From.Sym.NoFrame() {
 571                                 // If there is a stack frame at all, it includes
 572                                 // space to save the LR.
 573                                 c.autosize += 8
 574                         }
 575
 576                         if c.autosize != 0 {
 577                                 extrasize := int32(0)
 578                                 if c.autosize%16 == 8 {
 579                                         // Allocate extra 8 bytes on the frame top to save FP
 580                                         extrasize = 8
 581                                 } else if c.autosize&(16-1) == 0 {
 582                                         // Allocate extra 16 bytes to save FP for the old frame whose size is 8 mod 16
 583                                         extrasize = 16
 584                                 } else {
 585                                         c.ctxt.Diag("%v: unaligned frame size %d - must be 16 aligned", p, c.autosize-8)
 586                                 }
 587                                 c.autosize += extrasize
 588                                 c.cursym.Func().Locals += extrasize
 589
 590                                 // low 32 bits for autosize
 591                                 // high 32 bits for extrasize
 592                                 p.To.Offset = int64(c.autosize) | int64(extrasize)<<32
 593                         } else {
 594                                 // NOFRAME
 595                                 p.To.Offset = 0
 596                         }
 597
 598                         if c.autosize == 0 && c.cursym.Func().Text.Mark&LEAF == 0 {
 599                                 if c.ctxt.Debugvlog {
 600                                         c.ctxt.Logf("save suppressed in: %s\n", c.cursym.Func().Text.From.Sym.Name)
 601                                 }
 602                                 c.cursym.Func().Text.Mark |= LEAF
 603                         }
 604
 605                         if cursym.Func().Text.Mark&LEAF != 0 {
 606                                 cursym.Set(obj.AttrLeaf, true)
 607                                 if p.From.Sym.NoFrame() {
 608                                         break
 609                                 }
 610                         }
 611
 612                         if p.Mark&LEAF != 0 && c.autosize < objabi.StackSmall {
 613                                 // A leaf function with a small stack can be marked
 614                                 // NOSPLIT, avoiding a stack check.
 615                                 p.From.Sym.Set(obj.AttrNoSplit, true)
 616                         }
 617
 618                         if !p.From.Sym.NoSplit() {
 619                                 p = c.stacksplit(p, c.autosize) // emit split check
 620                         }
 621
 622                         var prologueEnd *obj.Prog
 623
 624                         aoffset := c.autosize
 625                         if aoffset > 0xF0 {
 626                                 aoffset = 0xF0
 627                         }
 628
 629                         // Frame is non-empty. Make sure to save link register, even if
 630                         // it is a leaf function, so that traceback works.
 631                         q = p
 632                         if c.autosize > aoffset {
 633                                 // Frame size is too large for a MOVD.W instruction.
 634                                 // Store link register before decrementing SP, so if a signal comes
 635                                 // during the execution of the function prologue, the traceback
 636                                 // code will not see a half-updated stack frame.
 637                                 // This sequence is not async preemptible, as if we open a frame
 638                                 // at the current SP, it will clobber the saved LR.
 639                                 q = c.ctxt.StartUnsafePoint(q, c.newprog)
 640
 641                                 q = obj.Appendp(q, c.newprog)
 642                                 q.Pos = p.Pos
 643                                 q.As = ASUB
 644                                 q.From.Type = obj.TYPE_CONST
 645                                 q.From.Offset = int64(c.autosize)
 646                                 q.Reg = REGSP
 647                                 q.To.Type = obj.TYPE_REG
 648                                 q.To.Reg = REGTMP
 649
 650                                 prologueEnd = q
 651
 652                                 q = obj.Appendp(q, c.newprog)
 653                                 q.Pos = p.Pos
 654                                 q.As = AMOVD
 655                                 q.From.Type = obj.TYPE_REG
 656                                 q.From.Reg = REGLINK
 657                                 q.To.Type = obj.TYPE_MEM
 658                                 q.To.Reg = REGTMP
 659
 660                                 q1 = obj.Appendp(q, c.newprog)
 661                                 q1.Pos = p.Pos
 662                                 q1.As = AMOVD
 663                                 q1.From.Type = obj.TYPE_REG
 664                                 q1.From.Reg = REGTMP
 665                                 q1.To.Type = obj.TYPE_REG
 666                                 q1.To.Reg = REGSP
 667                                 q1.Spadj = c.autosize
 668
 669                                 if buildcfg.GOOS == "ios" {
 670                                         // iOS does not support SA_ONSTACK. We will run the signal handler
 671                                         // on the G stack. If we write below SP, it may be clobbered by
 672                                         // the signal handler. So we save LR after decrementing SP.
 673                                         q1 = obj.Appendp(q1, c.newprog)
 674                                         q1.Pos = p.Pos
 675                                         q1.As = AMOVD
 676                                         q1.From.Type = obj.TYPE_REG
 677                                         q1.From.Reg = REGLINK
 678                                         q1.To.Type = obj.TYPE_MEM
 679                                         q1.To.Reg = REGSP
 680                                 }
 681
 682                                 q1 = c.ctxt.EndUnsafePoint(q1, c.newprog, -1)
 683                         } else {
 684                                 // small frame, update SP and save LR in a single MOVD.W instruction
 685                                 q1 = obj.Appendp(q, c.newprog)
 686                                 q1.As = AMOVD
 687                                 q1.Pos = p.Pos
 688                                 q1.From.Type = obj.TYPE_REG
 689                                 q1.From.Reg = REGLINK
 690                                 q1.To.Type = obj.TYPE_MEM
 691                                 q1.Scond = C_XPRE
 692                                 q1.To.Offset = int64(-aoffset)
 693                                 q1.To.Reg = REGSP
 694                                 q1.Spadj = aoffset
 695
 696                                 prologueEnd = q1
 697                         }
 698
 699                         prologueEnd.Pos = prologueEnd.Pos.WithXlogue(src.PosPrologueEnd)
 700
 701                         // Frame pointer.
 702                         q1 = obj.Appendp(q1, c.newprog)
 703                         q1.Pos = p.Pos
 704                         q1.As = AMOVD
 705                         q1.From.Type = obj.TYPE_REG
 706                         q1.From.Reg = REGFP
 707                         q1.To.Type = obj.TYPE_MEM
 708                         q1.To.Reg = REGSP
 709                         q1.To.Offset = -8
 710
 711                         q1 = obj.Appendp(q1, c.newprog)
 712                         q1.Pos = p.Pos
 713                         q1.As = ASUB
 714                         q1.From.Type = obj.TYPE_CONST
 715                         q1.From.Offset = 8
 716                         q1.Reg = REGSP
 717                         q1.To.Type = obj.TYPE_REG
 718                         q1.To.Reg = REGFP
 719
 720                         if c.cursym.Func().Text.From.Sym.Wrapper() {
 721                                 // if(g->panic != nil && g->panic->argp == FP) g->panic->argp = bottom-of-frame
 722                                 //
 723                                 //      MOV  g_panic(g), RT1
 724                                 //      CBNZ checkargp
 725                                 // end:
 726                                 //      NOP
 727                                 // ... function body ...
 728                                 // checkargp:
 729                                 //      MOV  panic_argp(RT1), RT2
 730                                 //      ADD  $(autosize+8), RSP, R20
 731                                 //      CMP  RT2, R20
 732                                 //      BNE  end
 733                                 //      ADD  $8, RSP, R20
 734                                 //      MOVD R20, panic_argp(RT1)
 735                                 //      B    end
 736                                 //
 737                                 // The NOP is needed to give the jumps somewhere to land.
 738                                 // It is a liblink NOP, not an ARM64 NOP: it encodes to 0 instruction bytes.
 739                                 q = q1
 740
 741                                 // MOV g_panic(g), RT1
 742                                 q = obj.Appendp(q, c.newprog)
 743                                 q.As = AMOVD
 744                                 q.From.Type = obj.TYPE_MEM
 745                                 q.From.Reg = REGG
 746                                 q.From.Offset = 4 * int64(c.ctxt.Arch.PtrSize) // G.panic
 747                                 q.To.Type = obj.TYPE_REG
 748                                 q.To.Reg = REGRT1
 749
 750                                 // CBNZ RT1, checkargp
 751                                 cbnz := obj.Appendp(q, c.newprog)
 752                                 cbnz.As = ACBNZ
 753                                 cbnz.From.Type = obj.TYPE_REG
 754                                 cbnz.From.Reg = REGRT1
 755                                 cbnz.To.Type = obj.TYPE_BRANCH
 756
 757                                 // Empty branch target at the top of the function body
 758                                 end := obj.Appendp(cbnz, c.newprog)
 759                                 end.As = obj.ANOP
 760
 761                                 // find the end of the function
 762                                 var last *obj.Prog
 763                                 for last = end; last.Link != nil; last = last.Link {
 764                                 }
 765
 766                                 // MOV panic_argp(RT1), RT2
 767                                 mov := obj.Appendp(last, c.newprog)
 768                                 mov.As = AMOVD
 769                                 mov.From.Type = obj.TYPE_MEM
 770                                 mov.From.Reg = REGRT1
 771                                 mov.From.Offset = 0 // Panic.argp
 772                                 mov.To.Type = obj.TYPE_REG
 773                                 mov.To.Reg = REGRT2
 774
 775                                 // CBNZ branches to the MOV above
 776                                 cbnz.To.SetTarget(mov)
 777
 778                                 // ADD $(autosize+8), SP, R20
 779                                 q = obj.Appendp(mov, c.newprog)
 780                                 q.As = AADD
 781                                 q.From.Type = obj.TYPE_CONST
 782                                 q.From.Offset = int64(c.autosize) + 8
 783                                 q.Reg = REGSP
 784                                 q.To.Type = obj.TYPE_REG
 785                                 q.To.Reg = REG_R20
 786
 787                                 // CMP RT2, R20
 788                                 q = obj.Appendp(q, c.newprog)
 789                                 q.As = ACMP
 790                                 q.From.Type = obj.TYPE_REG
 791                                 q.From.Reg = REGRT2
 792                                 q.Reg = REG_R20
 793
 794                                 // BNE end
 795                                 q = obj.Appendp(q, c.newprog)
 796                                 q.As = ABNE
 797                                 q.To.Type = obj.TYPE_BRANCH
 798                                 q.To.SetTarget(end)
 799
 800                                 // ADD $8, SP, R20
 801                                 q = obj.Appendp(q, c.newprog)
 802                                 q.As = AADD
 803                                 q.From.Type = obj.TYPE_CONST
 804                                 q.From.Offset = 8
 805                                 q.Reg = REGSP
 806                                 q.To.Type = obj.TYPE_REG
 807                                 q.To.Reg = REG_R20
 808
 809                                 // MOV R20, panic_argp(RT1)
 810                                 q = obj.Appendp(q, c.newprog)
 811                                 q.As = AMOVD
 812                                 q.From.Type = obj.TYPE_REG
 813                                 q.From.Reg = REG_R20
 814                                 q.To.Type = obj.TYPE_MEM
 815                                 q.To.Reg = REGRT1
 816                                 q.To.Offset = 0 // Panic.argp
 817
 818                                 // B end
 819                                 q = obj.Appendp(q, c.newprog)
 820                                 q.As = AB
 821                                 q.To.Type = obj.TYPE_BRANCH
 822                                 q.To.SetTarget(end)
 823                         }
 824
 825                 case obj.ARET:
 826                         nocache(p)
 827                         if p.From.Type == obj.TYPE_CONST {
 828                                 c.ctxt.Diag("using BECOME (%v) is not supported!", p)
 829                                 break
 830                         }
 831
 832                         retjmp = p.To.Sym
 833                         p.To = obj.Addr{}
 834                         if c.cursym.Func().Text.Mark&LEAF != 0 {
 835                                 if c.autosize != 0 {
 836                                         p.As = AADD
 837                                         p.From.Type = obj.TYPE_CONST
 838                                         p.From.Offset = int64(c.autosize)
 839                                         p.To.Type = obj.TYPE_REG
 840                                         p.To.Reg = REGSP
 841                                         p.Spadj = -c.autosize
 842
 843                                         // Frame pointer.
 844                                         p = obj.Appendp(p, c.newprog)
 845                                         p.As = ASUB
 846                                         p.From.Type = obj.TYPE_CONST
 847                                         p.From.Offset = 8
 848                                         p.Reg = REGSP
 849                                         p.To.Type = obj.TYPE_REG
 850                                         p.To.Reg = REGFP
 851                                 }
 852                         } else {
 853                                 /* want write-back pre-indexed SP+autosize -> SP, loading REGLINK*/
 854
 855                                 // Frame pointer.
 856                                 p.As = AMOVD
 857                                 p.From.Type = obj.TYPE_MEM
 858                                 p.From.Reg = REGSP
 859                                 p.From.Offset = -8
 860                                 p.To.Type = obj.TYPE_REG
 861                                 p.To.Reg = REGFP
 862                                 p = obj.Appendp(p, c.newprog)
 863
 864                                 aoffset := c.autosize
 865
 866                                 if aoffset <= 0xF0 {
 867                                         p.As = AMOVD
 868                                         p.From.Type = obj.TYPE_MEM
 869                                         p.Scond = C_XPOST
 870                                         p.From.Offset = int64(aoffset)
 871                                         p.From.Reg = REGSP
 872                                         p.To.Type = obj.TYPE_REG
 873                                         p.To.Reg = REGLINK
 874                                         p.Spadj = -aoffset
 875                                 } else {
 876                                         p.As = AMOVD
 877                                         p.From.Type = obj.TYPE_MEM
 878                                         p.From.Offset = 0
 879                                         p.From.Reg = REGSP
 880                                         p.To.Type = obj.TYPE_REG
 881                                         p.To.Reg = REGLINK
 882
 883                                         q = newprog()
 884                                         q.As = AADD
 885                                         q.From.Type = obj.TYPE_CONST
 886                                         q.From.Offset = int64(aoffset)
 887                                         q.To.Type = obj.TYPE_REG
 888                                         q.To.Reg = REGSP
 889                                         q.Link = p.Link
 890                                         q.Spadj = int32(-q.From.Offset)
 891                                         q.Pos = p.Pos
 892                                         p.Link = q
 893                                         p = q
 894                                 }
 895                         }
 896
 897                         // If enabled, this code emits 'MOV PC, R27' before every 'MOV LR, PC',
 898                         // so that if you are debugging a low-level crash where PC and LR are zero,
 899                         // you can look at R27 to see what jumped to the zero.
 900                         // This is useful when bringing up Go on a new system.
 901                         // (There is similar code in ../ppc64/obj9.go:/if.false.)
 902                         const debugRETZERO = false
 903                         if debugRETZERO {
 904                                 if p.As != obj.ARET {
 905                                         q = newprog()
 906                                         q.Pos = p.Pos
 907                                         q.Link = p.Link
 908                                         p.Link = q
 909                                         p = q
 910                                 }
 911                                 p.As = AADR
 912                                 p.From.Type = obj.TYPE_BRANCH
 913                                 p.From.Offset = 0
 914                                 p.To.Type = obj.TYPE_REG
 915                                 p.To.Reg = REGTMP
 916
 917                         }
 918
 919                         if p.As != obj.ARET {
 920                                 q = newprog()
 921                                 q.Pos = p.Pos
 922                                 q.Link = p.Link
 923                                 p.Link = q
 924                                 p = q
 925                         }
 926
 927                         if retjmp != nil { // retjmp
 928                                 p.As = AB
 929                                 p.To.Type = obj.TYPE_BRANCH
 930                                 p.To.Sym = retjmp
 931                                 p.Spadj = +c.autosize
 932                                 break
 933                         }
 934
 935                         p.As = obj.ARET
 936                         p.To.Type = obj.TYPE_MEM
 937                         p.To.Offset = 0
 938                         p.To.Reg = REGLINK
 939                         p.Spadj = +c.autosize
 940
 941                 case AADD, ASUB:
 942                         if p.To.Type == obj.TYPE_REG && p.To.Reg == REGSP && p.From.Type == obj.TYPE_CONST {
 943                                 if p.As == AADD {
 944                                         p.Spadj = int32(-p.From.Offset)
 945                                 } else {
 946                                         p.Spadj = int32(+p.From.Offset)
 947                                 }
 948                         }
 949
 950                 case obj.AGETCALLERPC:
 951                         if cursym.Leaf() {
 952                                 /* MOVD LR, Rd */
 953                                 p.As = AMOVD
 954                                 p.From.Type = obj.TYPE_REG
 955                                 p.From.Reg = REGLINK
 956                         } else {
 957                                 /* MOVD (RSP), Rd */
 958                                 p.As = AMOVD
 959                                 p.From.Type = obj.TYPE_MEM
 960                                 p.From.Reg = REGSP
 961                         }
 962
 963                 case obj.ADUFFCOPY:
 964                         //  ADR ret_addr, R27
 965                         //  STP (FP, R27), -24(SP)
 966                         //  SUB 24, SP, FP
 967                         //  DUFFCOPY
 968                         // ret_addr:
 969                         //  SUB 8, SP, FP
 970
 971                         q1 := p
 972                         // copy DUFFCOPY from q1 to q4
 973                         q4 := obj.Appendp(p, c.newprog)
 974                         q4.Pos = p.Pos
 975                         q4.As = obj.ADUFFCOPY
 976                         q4.To = p.To
 977
 978                         q1.As = AADR
 979                         q1.From.Type = obj.TYPE_BRANCH
 980                         q1.To.Type = obj.TYPE_REG
 981                         q1.To.Reg = REG_R27
 982
 983                         q2 := obj.Appendp(q1, c.newprog)
 984                         q2.Pos = p.Pos
 985                         q2.As = ASTP
 986                         q2.From.Type = obj.TYPE_REGREG
 987                         q2.From.Reg = REGFP
 988                         q2.From.Offset = int64(REG_R27)
 989                         q2.To.Type = obj.TYPE_MEM
 990                         q2.To.Reg = REGSP
 991                         q2.To.Offset = -24
 992
 993                         // maintain FP for DUFFCOPY
 994                         q3 := obj.Appendp(q2, c.newprog)
 995                         q3.Pos = p.Pos
 996                         q3.As = ASUB
 997                         q3.From.Type = obj.TYPE_CONST
 998                         q3.From.Offset = 24
 999                         q3.Reg = REGSP
1000                         q3.To.Type = obj.TYPE_REG
1001                         q3.To.Reg = REGFP
1002
1003                         q5 := obj.Appendp(q4, c.newprog)
1004                         q5.Pos = p.Pos
1005                         q5.As = ASUB
1006                         q5.From.Type = obj.TYPE_CONST
1007                         q5.From.Offset = 8
1008                         q5.Reg = REGSP
1009                         q5.To.Type = obj.TYPE_REG
1010                         q5.To.Reg = REGFP
1011                         q1.From.SetTarget(q5)
1012                         p = q5
1013
1014                 case obj.ADUFFZERO:
1015                         //  ADR ret_addr, R27
1016                         //  STP (FP, R27), -24(SP)
1017                         //  SUB 24, SP, FP
1018                         //  DUFFZERO
1019                         // ret_addr:
1020                         //  SUB 8, SP, FP
1021
1022                         q1 := p
1023                         // copy DUFFZERO from q1 to q4
1024                         q4 := obj.Appendp(p, c.newprog)
1025                         q4.Pos = p.Pos
1026                         q4.As = obj.ADUFFZERO
1027                         q4.To = p.To
1028
1029                         q1.As = AADR
1030                         q1.From.Type = obj.TYPE_BRANCH
1031                         q1.To.Type = obj.TYPE_REG
1032                         q1.To.Reg = REG_R27
1033
1034                         q2 := obj.Appendp(q1, c.newprog)
1035                         q2.Pos = p.Pos
1036                         q2.As = ASTP
1037                         q2.From.Type = obj.TYPE_REGREG
1038                         q2.From.Reg = REGFP
1039                         q2.From.Offset = int64(REG_R27)
1040                         q2.To.Type = obj.TYPE_MEM
1041                         q2.To.Reg = REGSP
1042                         q2.To.Offset = -24
1043
1044                         // maintain FP for DUFFZERO
1045                         q3 := obj.Appendp(q2, c.newprog)
1046                         q3.Pos = p.Pos
1047                         q3.As = ASUB
1048                         q3.From.Type = obj.TYPE_CONST
1049                         q3.From.Offset = 24
1050                         q3.Reg = REGSP
1051                         q3.To.Type = obj.TYPE_REG
1052                         q3.To.Reg = REGFP
1053
1054                         q5 := obj.Appendp(q4, c.newprog)
1055                         q5.Pos = p.Pos
1056                         q5.As = ASUB
1057                         q5.From.Type = obj.TYPE_CONST
1058                         q5.From.Offset = 8
1059                         q5.Reg = REGSP
1060                         q5.To.Type = obj.TYPE_REG
1061                         q5.To.Reg = REGFP
1062                         q1.From.SetTarget(q5)
1063                         p = q5
1064                 }
1065
1066                 if p.To.Type == obj.TYPE_REG && p.To.Reg == REGSP && p.Spadj == 0 {
1067                         f := c.cursym.Func()
1068                         if f.FuncFlag&objabi.FuncFlag_SPWRITE == 0 {
1069                                 c.cursym.Func().FuncFlag |= objabi.FuncFlag_SPWRITE
1070                                 if ctxt.Debugvlog || !ctxt.IsAsm {
1071                                         ctxt.Logf("auto-SPWRITE: %s %v\n", c.cursym.Name, p)
1072                                         if !ctxt.IsAsm {
1073                                                 ctxt.Diag("invalid auto-SPWRITE in non-assembly")
1074                                                 ctxt.DiagFlush()
1075                                                 log.Fatalf("bad SPWRITE")
1076                                         }
1077                                 }
1078                         }
1079                 }
1080         }
1081 }
1082
1083 func nocache(p *obj.Prog) {
1084         p.Optab = 0
1085         p.From.Class = 0
1086         p.To.Class = 0
1087 }
1088
1089 var unaryDst = map[obj.As]bool{
1090         AWORD:  true,
1091         ADWORD: true,
1092         ABL:    true,
1093         AB:     true,
1094         ACLREX: true,
1095 }
1096
1097 var Linkarm64 = obj.LinkArch{
1098         Arch:           sys.ArchARM64,
1099         Init:           buildop,
1100         Preprocess:     preprocess,
1101         Assemble:       span7,
1102         Progedit:       progedit,
1103         UnaryDst:       unaryDst,
1104         DWARFRegisters: ARM64DWARFRegisters,
1105 }