clang/lib/Driver/ToolChains/AVR.cpp

   1 //===--- AVR.cpp - AVR ToolChain Implementations ----------------*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "AVR.h"
  10 #include "CommonArgs.h"
  11 #include "clang/Driver/Compilation.h"
  12 #include "clang/Driver/DriverDiagnostic.h"
  13 #include "clang/Driver/InputInfo.h"
  14 #include "clang/Driver/Options.h"
  15 #include "llvm/ADT/Optional.h"
  16 #include "llvm/ADT/StringExtras.h"
  17 #include "llvm/ADT/StringSwitch.h"
  18 #include "llvm/MC/MCSubtargetInfo.h"
  19 #include "llvm/MC/SubtargetFeature.h"
  20 #include "llvm/Option/ArgList.h"
  21 #include "llvm/Support/FileSystem.h"
  22 #include "llvm/Support/Path.h"
  23
  24 using namespace clang::driver;
  25 using namespace clang::driver::toolchains;
  26 using namespace clang::driver::tools;
  27 using namespace clang;
  28 using namespace llvm::opt;
  29
  30 namespace {
  31
  32 // NOTE: This list has been synchronized with gcc-avr 7.3.0 and avr-libc 2.0.0.
  33 constexpr struct {
  34   StringRef Name;
  35   StringRef SubPath;
  36   StringRef Family;
  37   unsigned DataAddr;
  38 } MCUInfo[] = {
  39     {"at90s1200", "", "avr1", 0},
  40     {"attiny11", "", "avr1", 0},
  41     {"attiny12", "", "avr1", 0},
  42     {"attiny15", "", "avr1", 0},
  43     {"attiny28", "", "avr1", 0},
  44     {"at90s2313", "tiny-stack", "avr2", 0x800060},
  45     {"at90s2323", "tiny-stack", "avr2", 0x800060},
  46     {"at90s2333", "tiny-stack", "avr2", 0x800060},
  47     {"at90s2343", "tiny-stack", "avr2", 0x800060},
  48     {"at90s4433", "tiny-stack", "avr2", 0x800060},
  49     {"attiny22", "tiny-stack", "avr2", 0x800060},
  50     {"attiny26", "tiny-stack", "avr2", 0x800060},
  51     {"at90s4414", "", "avr2", 0x800060},
  52     {"at90s4434", "", "avr2", 0x800060},
  53     {"at90s8515", "", "avr2", 0x800060},
  54     {"at90c8534", "", "avr2", 0x800060},
  55     {"at90s8535", "", "avr2", 0x800060},
  56     {"attiny13", "avr25/tiny-stack", "avr25", 0x800060},
  57     {"attiny13a", "avr25/tiny-stack", "avr25", 0x800060},
  58     {"attiny2313", "avr25/tiny-stack", "avr25", 0x800060},
  59     {"attiny2313a", "avr25/tiny-stack", "avr25", 0x800060},
  60     {"attiny24", "avr25/tiny-stack", "avr25", 0x800060},
  61     {"attiny24a", "avr25/tiny-stack", "avr25", 0x800060},
  62     {"attiny25", "avr25/tiny-stack", "avr25", 0x800060},
  63     {"attiny261", "avr25/tiny-stack", "avr25", 0x800060},
  64     {"attiny261a", "avr25/tiny-stack", "avr25", 0x800060},
  65     {"at86rf401", "avr25", "avr25", 0x800060},
  66     {"ata5272", "avr25", "avr25", 0x800100},
  67     {"ata6616c", "avr25", "avr25", 0x800100},
  68     {"attiny4313", "avr25", "avr25", 0x800060},
  69     {"attiny44", "avr25", "avr25", 0x800060},
  70     {"attiny44a", "avr25", "avr25", 0x800060},
  71     {"attiny84", "avr25", "avr25", 0x800060},
  72     {"attiny84a", "avr25", "avr25", 0x800060},
  73     {"attiny45", "avr25", "avr25", 0x800060},
  74     {"attiny85", "avr25", "avr25", 0x800060},
  75     {"attiny441", "avr25", "avr25", 0x800100},
  76     {"attiny461", "avr25", "avr25", 0x800060},
  77     {"attiny461a", "avr25", "avr25", 0x800060},
  78     {"attiny841", "avr25", "avr25", 0x800100},
  79     {"attiny861", "avr25", "avr25", 0x800060},
  80     {"attiny861a", "avr25", "avr25", 0x800060},
  81     {"attiny87", "avr25", "avr25", 0x800100},
  82     {"attiny43u", "avr25", "avr25", 0x800060},
  83     {"attiny48", "avr25", "avr25", 0x800100},
  84     {"attiny88", "avr25", "avr25", 0x800100},
  85     {"attiny828", "avr25", "avr25", 0x800100},
  86     {"at43usb355", "avr3", "avr3", 0x800100},
  87     {"at76c711", "avr3", "avr3", 0x800060},
  88     {"atmega103", "avr31", "avr31", 0x800060},
  89     {"at43usb320", "avr31", "avr31", 0x800060},
  90     {"attiny167", "avr35", "avr35", 0x800100},
  91     {"at90usb82", "avr35", "avr35", 0x800100},
  92     {"at90usb162", "avr35", "avr35", 0x800100},
  93     {"ata5505", "avr35", "avr35", 0x800100},
  94     {"ata6617c", "avr35", "avr35", 0x800100},
  95     {"ata664251", "avr35", "avr35", 0x800100},
  96     {"atmega8u2", "avr35", "avr35", 0x800100},
  97     {"atmega16u2", "avr35", "avr35", 0x800100},
  98     {"atmega32u2", "avr35", "avr35", 0x800100},
  99     {"attiny1634", "avr35", "avr35", 0x800100},
 100     {"atmega8", "avr4", "avr4", 0x800060},
 101     {"ata6289", "avr4", "avr4", 0x800100},
 102     {"atmega8a", "avr4", "avr4", 0x800060},
 103     {"ata6285", "avr4", "avr4", 0x800100},
 104     {"ata6286", "avr4", "avr4", 0x800100},
 105     {"ata6612c", "avr4", "avr4", 0x800100},
 106     {"atmega48", "avr4", "avr4", 0x800100},
 107     {"atmega48a", "avr4", "avr4", 0x800100},
 108     {"atmega48pa", "avr4", "avr4", 0x800100},
 109     {"atmega48pb", "avr4", "avr4", 0x800100},
 110     {"atmega48p", "avr4", "avr4", 0x800100},
 111     {"atmega88", "avr4", "avr4", 0x800100},
 112     {"atmega88a", "avr4", "avr4", 0x800100},
 113     {"atmega88p", "avr4", "avr4", 0x800100},
 114     {"atmega88pa", "avr4", "avr4", 0x800100},
 115     {"atmega88pb", "avr4", "avr4", 0x800100},
 116     {"atmega8515", "avr4", "avr4", 0x800060},
 117     {"atmega8535", "avr4", "avr4", 0x800060},
 118     {"atmega8hva", "avr4", "avr4", 0x800100},
 119     {"at90pwm1", "avr4", "avr4", 0x800100},
 120     {"at90pwm2", "avr4", "avr4", 0x800100},
 121     {"at90pwm2b", "avr4", "avr4", 0x800100},
 122     {"at90pwm3", "avr4", "avr4", 0x800100},
 123     {"at90pwm3b", "avr4", "avr4", 0x800100},
 124     {"at90pwm81", "avr4", "avr4", 0x800100},
 125     {"ata5702m322", "avr5", "avr5", 0x800200},
 126     {"ata5782", "avr5", "avr5", 0x800200},
 127     {"ata5790", "avr5", "avr5", 0x800100},
 128     {"ata5790n", "avr5", "avr5", 0x800100},
 129     {"ata5791", "avr5", "avr5", 0x800100},
 130     {"ata5795", "avr5", "avr5", 0x800100},
 131     {"ata5831", "avr5", "avr5", 0x800200},
 132     {"ata6613c", "avr5", "avr5", 0x800100},
 133     {"ata6614q", "avr5", "avr5", 0x800100},
 134     {"ata8210", "avr5", "avr5", 0x800200},
 135     {"ata8510", "avr5", "avr5", 0x800200},
 136     {"atmega16", "avr5", "avr5", 0x800060},
 137     {"atmega16a", "avr5", "avr5", 0x800060},
 138     {"atmega161", "avr5", "avr5", 0x800060},
 139     {"atmega162", "avr5", "avr5", 0x800100},
 140     {"atmega163", "avr5", "avr5", 0x800060},
 141     {"atmega164a", "avr5", "avr5", 0x800100},
 142     {"atmega164p", "avr5", "avr5", 0x800100},
 143     {"atmega164pa", "avr5", "avr5", 0x800100},
 144     {"atmega165", "avr5", "avr5", 0x800100},
 145     {"atmega165a", "avr5", "avr5", 0x800100},
 146     {"atmega165p", "avr5", "avr5", 0x800100},
 147     {"atmega165pa", "avr5", "avr5", 0x800100},
 148     {"atmega168", "avr5", "avr5", 0x800100},
 149     {"atmega168a", "avr5", "avr5", 0x800100},
 150     {"atmega168p", "avr5", "avr5", 0x800100},
 151     {"atmega168pa", "avr5", "avr5", 0x800100},
 152     {"atmega168pb", "avr5", "avr5", 0x800100},
 153     {"atmega169", "avr5", "avr5", 0x800100},
 154     {"atmega169a", "avr5", "avr5", 0x800100},
 155     {"atmega169p", "avr5", "avr5", 0x800100},
 156     {"atmega169pa", "avr5", "avr5", 0x800100},
 157     {"atmega32", "avr5", "avr5", 0x800060},
 158     {"atmega32a", "avr5", "avr5", 0x800060},
 159     {"atmega323", "avr5", "avr5", 0x800060},
 160     {"atmega324a", "avr5", "avr5", 0x800100},
 161     {"atmega324p", "avr5", "avr5", 0x800100},
 162     {"atmega324pa", "avr5", "avr5", 0x800100},
 163     {"atmega324pb", "avr5", "avr5", 0x800100},
 164     {"atmega325", "avr5", "avr5", 0x800100},
 165     {"atmega325a", "avr5", "avr5", 0x800100},
 166     {"atmega325p", "avr5", "avr5", 0x800100},
 167     {"atmega325pa", "avr5", "avr5", 0x800100},
 168     {"atmega3250", "avr5", "avr5", 0x800100},
 169     {"atmega3250a", "avr5", "avr5", 0x800100},
 170     {"atmega3250p", "avr5", "avr5", 0x800100},
 171     {"atmega3250pa", "avr5", "avr5", 0x800100},
 172     {"atmega328", "avr5", "avr5", 0x800100},
 173     {"atmega328p", "avr5", "avr5", 0x800100},
 174     {"atmega328pb", "avr5", "avr5", 0x800100},
 175     {"atmega329", "avr5", "avr5", 0x800100},
 176     {"atmega329a", "avr5", "avr5", 0x800100},
 177     {"atmega329p", "avr5", "avr5", 0x800100},
 178     {"atmega329pa", "avr5", "avr5", 0x800100},
 179     {"atmega3290", "avr5", "avr5", 0x800100},
 180     {"atmega3290a", "avr5", "avr5", 0x800100},
 181     {"atmega3290p", "avr5", "avr5", 0x800100},
 182     {"atmega3290pa", "avr5", "avr5", 0x800100},
 183     {"atmega406", "avr5", "avr5", 0x800100},
 184     {"atmega64", "avr5", "avr5", 0x800100},
 185     {"atmega64a", "avr5", "avr5", 0x800100},
 186     {"atmega640", "avr5", "avr5", 0x800200},
 187     {"atmega644", "avr5", "avr5", 0x800100},
 188     {"atmega644a", "avr5", "avr5", 0x800100},
 189     {"atmega644p", "avr5", "avr5", 0x800100},
 190     {"atmega644pa", "avr5", "avr5", 0x800100},
 191     {"atmega645", "avr5", "avr5", 0x800100},
 192     {"atmega645a", "avr5", "avr5", 0x800100},
 193     {"atmega645p", "avr5", "avr5", 0x800100},
 194     {"atmega649", "avr5", "avr5", 0x800100},
 195     {"atmega649a", "avr5", "avr5", 0x800100},
 196     {"atmega649p", "avr5", "avr5", 0x800100},
 197     {"atmega6450", "avr5", "avr5", 0x800100},
 198     {"atmega6450a", "avr5", "avr5", 0x800100},
 199     {"atmega6450p", "avr5", "avr5", 0x800100},
 200     {"atmega6490", "avr5", "avr5", 0x800100},
 201     {"atmega6490a", "avr5", "avr5", 0x800100},
 202     {"atmega6490p", "avr5", "avr5", 0x800100},
 203     {"atmega64rfr2", "avr5", "avr5", 0x800200},
 204     {"atmega644rfr2", "avr5", "avr5", 0x800200},
 205     {"atmega16hva", "avr5", "avr5", 0x800100},
 206     {"atmega16hva2", "avr5", "avr5", 0x800100},
 207     {"atmega16hvb", "avr5", "avr5", 0x800100},
 208     {"atmega16hvbrevb", "avr5", "avr5", 0x800100},
 209     {"atmega32hvb", "avr5", "avr5", 0x800100},
 210     {"atmega32hvbrevb", "avr5", "avr5", 0x800100},
 211     {"atmega64hve", "avr5", "avr5", 0x800100},
 212     {"atmega64hve2", "avr5", "avr5", 0x800100},
 213     {"at90can32", "avr5", "avr5", 0x800100},
 214     {"at90can64", "avr5", "avr5", 0x800100},
 215     {"at90pwm161", "avr5", "avr5", 0x800100},
 216     {"at90pwm216", "avr5", "avr5", 0x800100},
 217     {"at90pwm316", "avr5", "avr5", 0x800100},
 218     {"atmega32c1", "avr5", "avr5", 0x800100},
 219     {"atmega64c1", "avr5", "avr5", 0x800100},
 220     {"atmega16m1", "avr5", "avr5", 0x800100},
 221     {"atmega32m1", "avr5", "avr5", 0x800100},
 222     {"atmega64m1", "avr5", "avr5", 0x800100},
 223     {"atmega16u4", "avr5", "avr5", 0x800100},
 224     {"atmega32u4", "avr5", "avr5", 0x800100},
 225     {"atmega32u6", "avr5", "avr5", 0x800100},
 226     {"at90usb646", "avr5", "avr5", 0x800100},
 227     {"at90usb647", "avr5", "avr5", 0x800100},
 228     {"at90scr100", "avr5", "avr5", 0x800100},
 229     {"at94k", "avr5", "avr5", 0x800060},
 230     {"m3000", "avr5", "avr5", 0x800060},
 231     {"atmega128", "avr51", "avr51", 0x800100},
 232     {"atmega128a", "avr51", "avr51", 0x800100},
 233     {"atmega1280", "avr51", "avr51", 0x800200},
 234     {"atmega1281", "avr51", "avr51", 0x800200},
 235     {"atmega1284", "avr51", "avr51", 0x800100},
 236     {"atmega1284p", "avr51", "avr51", 0x800100},
 237     {"atmega128rfa1", "avr51", "avr51", 0x800200},
 238     {"atmega128rfr2", "avr51", "avr51", 0x800200},
 239     {"atmega1284rfr2", "avr51", "avr51", 0x800200},
 240     {"at90can128", "avr51", "avr51", 0x800200},
 241     {"at90usb1286", "avr51", "avr51", 0x800200},
 242     {"at90usb1287", "avr51", "avr51", 0x800200},
 243     {"atmega2560", "avr6", "avr6", 0x800200},
 244     {"atmega2561", "avr6", "avr6", 0x800200},
 245     {"atmega256rfr2", "avr6", "avr6", 0x800200},
 246     {"atmega2564rfr2", "avr6", "avr6", 0x800200},
 247     {"attiny4", "avrtiny", "avrtiny", 0x800040},
 248     {"attiny5", "avrtiny", "avrtiny", 0x800040},
 249     {"attiny9", "avrtiny", "avrtiny", 0x800040},
 250     {"attiny10", "avrtiny", "avrtiny", 0x800040},
 251     {"attiny20", "avrtiny", "avrtiny", 0x800040},
 252     {"attiny40", "avrtiny", "avrtiny", 0x800040},
 253     {"attiny102", "avrtiny", "avrtiny", 0x800040},
 254     {"attiny104", "avrtiny", "avrtiny", 0x800040},
 255     {"atxmega16a4", "avrxmega2", "avrxmega2", 0x802000},
 256     {"atxmega16a4u", "avrxmega2", "avrxmega2", 0x802000},
 257     {"atxmega16c4", "avrxmega2", "avrxmega2", 0x802000},
 258     {"atxmega16d4", "avrxmega2", "avrxmega2", 0x802000},
 259     {"atxmega32a4", "avrxmega2", "avrxmega2", 0x802000},
 260     {"atxmega32a4u", "avrxmega2", "avrxmega2", 0x802000},
 261     {"atxmega32c3", "avrxmega2", "avrxmega2", 0x802000},
 262     {"atxmega32c4", "avrxmega2", "avrxmega2", 0x802000},
 263     {"atxmega32d3", "avrxmega2", "avrxmega2", 0x802000},
 264     {"atxmega32d4", "avrxmega2", "avrxmega2", 0x802000},
 265     {"atxmega32e5", "avrxmega2", "avrxmega2", 0x802000},
 266     {"atxmega16e5", "avrxmega2", "avrxmega2", 0x802000},
 267     {"atxmega8e5", "avrxmega2", "avrxmega2", 0x802000},
 268     {"atxmega64a3", "avrxmega4", "avrxmega4", 0x802000},
 269     {"atxmega64a3u", "avrxmega4", "avrxmega4", 0x802000},
 270     {"atxmega64a4u", "avrxmega4", "avrxmega4", 0x802000},
 271     {"atxmega64b1", "avrxmega4", "avrxmega4", 0x802000},
 272     {"atxmega64b3", "avrxmega4", "avrxmega4", 0x802000},
 273     {"atxmega64c3", "avrxmega4", "avrxmega4", 0x802000},
 274     {"atxmega64d3", "avrxmega4", "avrxmega4", 0x802000},
 275     {"atxmega64d4", "avrxmega4", "avrxmega4", 0x802000},
 276     {"atxmega64a1", "avrxmega5", "avrxmega5", 0x802000},
 277     {"atxmega64a1u", "avrxmega5", "avrxmega5", 0x802000},
 278     {"atxmega128a3", "avrxmega6", "avrxmega6", 0x802000},
 279     {"atxmega128a3u", "avrxmega6", "avrxmega6", 0x802000},
 280     {"atxmega128b1", "avrxmega6", "avrxmega6", 0x802000},
 281     {"atxmega128b3", "avrxmega6", "avrxmega6", 0x802000},
 282     {"atxmega128c3", "avrxmega6", "avrxmega6", 0x802000},
 283     {"atxmega128d3", "avrxmega6", "avrxmega6", 0x802000},
 284     {"atxmega128d4", "avrxmega6", "avrxmega6", 0x802000},
 285     {"atxmega192a3", "avrxmega6", "avrxmega6", 0x802000},
 286     {"atxmega192a3u", "avrxmega6", "avrxmega6", 0x802000},
 287     {"atxmega192c3", "avrxmega6", "avrxmega6", 0x802000},
 288     {"atxmega192d3", "avrxmega6", "avrxmega6", 0x802000},
 289     {"atxmega256a3", "avrxmega6", "avrxmega6", 0x802000},
 290     {"atxmega256a3u", "avrxmega6", "avrxmega6", 0x802000},
 291     {"atxmega256a3b", "avrxmega6", "avrxmega6", 0x802000},
 292     {"atxmega256a3bu", "avrxmega6", "avrxmega6", 0x802000},
 293     {"atxmega256c3", "avrxmega6", "avrxmega6", 0x802000},
 294     {"atxmega256d3", "avrxmega6", "avrxmega6", 0x802000},
 295     {"atxmega384c3", "avrxmega6", "avrxmega6", 0x802000},
 296     {"atxmega384d3", "avrxmega6", "avrxmega6", 0x802000},
 297     {"atxmega128a1", "avrxmega7", "avrxmega7", 0x802000},
 298     {"atxmega128a1u", "avrxmega7", "avrxmega7", 0x802000},
 299     {"atxmega128a4u", "avrxmega7", "avrxmega7", 0x802000},
 300     {"attiny202", "avrxmega3/short-calls", "avrxmega3", 0x803F80},
 301     {"attiny204", "avrxmega3/short-calls", "avrxmega3", 0x803F80},
 302     {"attiny212", "avrxmega3/short-calls", "avrxmega3", 0x803F80},
 303     {"attiny214", "avrxmega3/short-calls", "avrxmega3", 0x803F80},
 304     {"attiny402", "avrxmega3/short-calls", "avrxmega3", 0x803F00},
 305     {"attiny404", "avrxmega3/short-calls", "avrxmega3", 0x803F00},
 306     {"attiny406", "avrxmega3/short-calls", "avrxmega3", 0x803F00},
 307     {"attiny412", "avrxmega3/short-calls", "avrxmega3", 0x803F00},
 308     {"attiny414", "avrxmega3/short-calls", "avrxmega3", 0x803F00},
 309     {"attiny416", "avrxmega3/short-calls", "avrxmega3", 0x803F00},
 310     {"attiny417", "avrxmega3/short-calls", "avrxmega3", 0x803F00},
 311     {"attiny804", "avrxmega3/short-calls", "avrxmega3", 0x803E00},
 312     {"attiny806", "avrxmega3/short-calls", "avrxmega3", 0x803E00},
 313     {"attiny807", "avrxmega3/short-calls", "avrxmega3", 0x803E00},
 314     {"attiny814", "avrxmega3/short-calls", "avrxmega3", 0x803E00},
 315     {"attiny816", "avrxmega3/short-calls", "avrxmega3", 0x803E00},
 316     {"attiny817", "avrxmega3/short-calls", "avrxmega3", 0x803E00},
 317     {"atmega808", "avrxmega3/short-calls", "avrxmega3", 0x803C00},
 318     {"atmega809", "avrxmega3/short-calls", "avrxmega3", 0x803C00},
 319     {"atmega1608", "avrxmega3", "avrxmega3", 0x803800},
 320     {"atmega1609", "avrxmega3", "avrxmega3", 0x803800},
 321     {"atmega3208", "avrxmega3", "avrxmega3", 0x803000},
 322     {"atmega3209", "avrxmega3", "avrxmega3", 0x803000},
 323     {"atmega4808", "avrxmega3", "avrxmega3", 0x802800},
 324     {"atmega4809", "avrxmega3", "avrxmega3", 0x802800},
 325     {"attiny1604", "avrxmega3", "avrxmega3", 0x803C00},
 326     {"attiny1606", "avrxmega3", "avrxmega3", 0x803C00},
 327     {"attiny1607", "avrxmega3", "avrxmega3", 0x803C00},
 328     {"attiny1614", "avrxmega3", "avrxmega3", 0x803800},
 329     {"attiny1616", "avrxmega3", "avrxmega3", 0x803800},
 330     {"attiny1617", "avrxmega3", "avrxmega3", 0x803800},
 331     {"attiny1624", "avrxmega3", "avrxmega3", 0x803800},
 332     {"attiny1626", "avrxmega3", "avrxmega3", 0x803800},
 333     {"attiny1627", "avrxmega3", "avrxmega3", 0x803800},
 334     {"attiny3216", "avrxmega3", "avrxmega3", 0x803800},
 335     {"attiny3217", "avrxmega3", "avrxmega3", 0x803800},
 336 };
 337
 338 std::string GetMCUSubPath(StringRef MCUName) {
 339   for (const auto &MCU : MCUInfo)
 340     if (MCU.Name == MCUName)
 341       return std::string(MCU.SubPath);
 342   return "";
 343 }
 344
 345 llvm::Optional<StringRef> GetMCUFamilyName(StringRef MCUName) {
 346   for (const auto &MCU : MCUInfo)
 347     if (MCU.Name == MCUName)
 348       return Optional<StringRef>(MCU.Family);
 349   return Optional<StringRef>();
 350 }
 351
 352 llvm::Optional<unsigned> GetMCUSectionAddressData(StringRef MCUName) {
 353   for (const auto &MCU : MCUInfo)
 354     if (MCU.Name == MCUName && MCU.DataAddr > 0)
 355       return Optional<unsigned>(MCU.DataAddr);
 356   return Optional<unsigned>();
 357 }
 358
 359 const StringRef PossibleAVRLibcLocations[] = {
 360     "/avr",
 361     "/usr/avr",
 362     "/usr/lib/avr",
 363 };
 364
 365 } // end anonymous namespace
 366
 367 /// AVR Toolchain
 368 AVRToolChain::AVRToolChain(const Driver &D, const llvm::Triple &Triple,
 369                            const ArgList &Args)
 370     : Generic_ELF(D, Triple, Args) {
 371   GCCInstallation.init(Triple, Args);
 372
 373   if (getCPUName(D, Args, Triple).empty())
 374     D.Diag(diag::warn_drv_avr_mcu_not_specified);
 375
 376   // Only add default libraries if the user hasn't explicitly opted out.
 377   if (!Args.hasArg(options::OPT_nostdlib) &&
 378       !Args.hasArg(options::OPT_nodefaultlibs) && GCCInstallation.isValid()) {
 379     GCCInstallPath = GCCInstallation.getInstallPath();
 380     std::string GCCParentPath(GCCInstallation.getParentLibPath());
 381     getProgramPaths().push_back(GCCParentPath + "/../bin");
 382   }
 383 }
 384
 385 void AVRToolChain::AddClangSystemIncludeArgs(const ArgList &DriverArgs,
 386                                              ArgStringList &CC1Args) const {
 387   if (DriverArgs.hasArg(options::OPT_nostdinc) ||
 388       DriverArgs.hasArg(options::OPT_nostdlibinc))
 389     return;
 390
 391   // Omit if there is no avr-libc installed.
 392   Optional<std::string> AVRLibcRoot = findAVRLibcInstallation();
 393   if (!AVRLibcRoot)
 394     return;
 395
 396   // Add 'avr-libc/include' to clang system include paths if applicable.
 397   std::string AVRInc = *AVRLibcRoot + "/include";
 398   if (llvm::sys::fs::is_directory(AVRInc))
 399     addSystemInclude(DriverArgs, CC1Args, AVRInc);
 400 }
 401
 402 void AVRToolChain::addClangTargetOptions(
 403     const llvm::opt::ArgList &DriverArgs, llvm::opt::ArgStringList &CC1Args,
 404     Action::OffloadKind DeviceOffloadKind) const {
 405   // By default, use `.ctors` (not `.init_array`), as required by libgcc, which
 406   // runs constructors/destructors on AVR.
 407   if (!DriverArgs.hasFlag(options::OPT_fuse_init_array,
 408                           options::OPT_fno_use_init_array, false))
 409     CC1Args.push_back("-fno-use-init-array");
 410   // Use `-fno-use-cxa-atexit` as default, since avr-libc does not support
 411   // `__cxa_atexit()`.
 412   if (!DriverArgs.hasFlag(options::OPT_fuse_cxa_atexit,
 413                           options::OPT_fno_use_cxa_atexit, false))
 414     CC1Args.push_back("-fno-use-cxa-atexit");
 415 }
 416
 417 Tool *AVRToolChain::buildLinker() const {
 418   return new tools::AVR::Linker(getTriple(), *this);
 419 }
 420
 421 std::string
 422 AVRToolChain::getCompilerRT(const llvm::opt::ArgList &Args, StringRef Component,
 423                             FileType Type = ToolChain::FT_Static) const {
 424   assert(Type == ToolChain::FT_Static && "AVR only supports static libraries");
 425   // Since AVR can never be a host environment, its compiler-rt library files
 426   // should always have ".a" suffix, even on windows.
 427   SmallString<32> File("/libclang_rt.");
 428   File += Component.str();
 429   File += ".a";
 430   // Return the default compiler-rt path appended with
 431   // "avr/libclang_rt.$COMPONENT.a".
 432   SmallString<256> Path(ToolChain::getCompilerRTPath());
 433   llvm::sys::path::append(Path, "avr");
 434   llvm::sys::path::append(Path, File.str());
 435   return std::string(Path.str());
 436 }
 437
 438 void AVR::Linker::ConstructJob(Compilation &C, const JobAction &JA,
 439                                const InputInfo &Output,
 440                                const InputInfoList &Inputs, const ArgList &Args,
 441                                const char *LinkingOutput) const {
 442   const auto &TC = static_cast<const AVRToolChain &>(getToolChain());
 443   const Driver &D = getToolChain().getDriver();
 444
 445   // Compute information about the target AVR.
 446   std::string CPU = getCPUName(D, Args, getToolChain().getTriple());
 447   llvm::Optional<StringRef> FamilyName = GetMCUFamilyName(CPU);
 448   llvm::Optional<std::string> AVRLibcRoot = TC.findAVRLibcInstallation();
 449   llvm::Optional<unsigned> SectionAddressData = GetMCUSectionAddressData(CPU);
 450
 451   // Compute the linker program path, and use GNU "avr-ld" as default.
 452   const Arg *A = Args.getLastArg(options::OPT_fuse_ld_EQ);
 453   std::string Linker = A ? getToolChain().GetLinkerPath(nullptr)
 454                          : getToolChain().GetProgramPath(getShortName());
 455
 456   ArgStringList CmdArgs;
 457
 458   CmdArgs.push_back("-o");
 459   CmdArgs.push_back(Output.getFilename());
 460
 461   // Enable garbage collection of unused sections.
 462   CmdArgs.push_back("--gc-sections");
 463
 464   // Add library search paths before we specify libraries.
 465   Args.AddAllArgs(CmdArgs, options::OPT_L);
 466   getToolChain().AddFilePathLibArgs(Args, CmdArgs);
 467
 468   // Currently we only support libgcc and compiler-rt.
 469   auto RtLib = TC.GetRuntimeLibType(Args);
 470   assert(
 471       (RtLib == ToolChain::RLT_Libgcc || RtLib == ToolChain::RLT_CompilerRT) &&
 472       "unknown runtime library");
 473
 474   // Only add default libraries if the user hasn't explicitly opted out.
 475   bool LinkStdlib = false;
 476   if (!Args.hasArg(options::OPT_nostdlib) &&
 477       !Args.hasArg(options::OPT_nodefaultlibs)) {
 478     if (!CPU.empty()) {
 479       if (!FamilyName) {
 480         // We do not have an entry for this CPU in the family
 481         // mapping table yet.
 482         D.Diag(diag::warn_drv_avr_family_linking_stdlibs_not_implemented)
 483             << CPU;
 484       } else if (!AVRLibcRoot) {
 485         // No avr-libc found and so no runtime linked.
 486         D.Diag(diag::warn_drv_avr_libc_not_found);
 487       } else {
 488         std::string SubPath = GetMCUSubPath(CPU);
 489         // Add path of avr-libc.
 490         CmdArgs.push_back(
 491             Args.MakeArgString(Twine("-L") + *AVRLibcRoot + "/lib/" + SubPath));
 492         if (RtLib == ToolChain::RLT_Libgcc)
 493           CmdArgs.push_back(Args.MakeArgString("-L" + TC.getGCCInstallPath() +
 494                                                "/" + SubPath));
 495         LinkStdlib = true;
 496       }
 497     }
 498     if (!LinkStdlib)
 499       D.Diag(diag::warn_drv_avr_stdlib_not_linked);
 500   }
 501
 502   if (SectionAddressData) {
 503     std::string DataSectionArg =
 504         std::string("-Tdata=0x") + llvm::utohexstr(SectionAddressData.value());
 505     CmdArgs.push_back(Args.MakeArgString(DataSectionArg));
 506   } else {
 507     // We do not have an entry for this CPU in the address mapping table yet.
 508     D.Diag(diag::warn_drv_avr_linker_section_addresses_not_implemented) << CPU;
 509   }
 510
 511   // If the family name is known, we can link with the device-specific libgcc.
 512   // Without it, libgcc will simply not be linked. This matches avr-gcc
 513   // behavior.
 514   if (LinkStdlib) {
 515     assert(!CPU.empty() && "CPU name must be known in order to link stdlibs");
 516
 517     CmdArgs.push_back("--start-group");
 518
 519     // Add the object file for the CRT.
 520     std::string CrtFileName = std::string("-l:crt") + CPU + std::string(".o");
 521     CmdArgs.push_back(Args.MakeArgString(CrtFileName));
 522
 523     // Link to libgcc.
 524     if (RtLib == ToolChain::RLT_Libgcc)
 525       CmdArgs.push_back("-lgcc");
 526
 527     // Link to generic libraries of avr-libc.
 528     CmdArgs.push_back("-lm");
 529     CmdArgs.push_back("-lc");
 530
 531     // Add the link library specific to the MCU.
 532     CmdArgs.push_back(Args.MakeArgString(std::string("-l") + CPU));
 533
 534     // Add the relocatable inputs.
 535     AddLinkerInputs(getToolChain(), Inputs, Args, CmdArgs, JA);
 536
 537     // We directly use libclang_rt.builtins.a as input file, instead of using
 538     // '-lclang_rt.builtins'.
 539     if (RtLib == ToolChain::RLT_CompilerRT) {
 540       std::string RtLib =
 541           getToolChain().getCompilerRT(Args, "builtins", ToolChain::FT_Static);
 542       if (llvm::sys::fs::exists(RtLib))
 543         CmdArgs.push_back(Args.MakeArgString(RtLib));
 544     }
 545
 546     CmdArgs.push_back("--end-group");
 547
 548     // Add user specified linker script.
 549     Args.AddAllArgs(CmdArgs, options::OPT_T);
 550
 551     // Specify the family name as the emulation mode to use.
 552     // This is almost always required because otherwise avr-ld
 553     // will assume 'avr2' and warn about the program being larger
 554     // than the bare minimum supports.
 555     if (Linker.find("avr-ld") != std::string::npos)
 556       CmdArgs.push_back(Args.MakeArgString(std::string("-m") + *FamilyName));
 557   } else {
 558     AddLinkerInputs(getToolChain(), Inputs, Args, CmdArgs, JA);
 559   }
 560
 561   C.addCommand(std::make_unique<Command>(
 562       JA, *this, ResponseFileSupport::AtFileCurCP(), Args.MakeArgString(Linker),
 563       CmdArgs, Inputs, Output));
 564 }
 565
 566 llvm::Optional<std::string> AVRToolChain::findAVRLibcInstallation() const {
 567   // Search avr-libc installation according to avr-gcc installation.
 568   std::string GCCParent(GCCInstallation.getParentLibPath());
 569   std::string Path(GCCParent + "/avr");
 570   if (llvm::sys::fs::is_directory(Path))
 571     return Path;
 572   Path = GCCParent + "/../avr";
 573   if (llvm::sys::fs::is_directory(Path))
 574     return Path;
 575
 576   // Search avr-libc installation from possible locations, and return the first
 577   // one that exists, if there is no avr-gcc installed.
 578   for (StringRef PossiblePath : PossibleAVRLibcLocations) {
 579     std::string Path = getDriver().SysRoot + PossiblePath.str();
 580     if (llvm::sys::fs::is_directory(Path))
 581       return Path;
 582   }
 583
 584   return llvm::None;
 585 }