Add NativeClient support to Triple::ParseOS.
[oota-llvm.git] / lib / Support / Triple.cpp
1 //===--- Triple.cpp - Target triple helper class --------------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #include "llvm/ADT/Triple.h"
11 #include "llvm/ADT/SmallString.h"
12 #include "llvm/ADT/STLExtras.h"
13 #include <cstring>
14 using namespace llvm;
15
16 const char *Triple::getArchTypeName(ArchType Kind) {
17   switch (Kind) {
18   case InvalidArch: return "<invalid>";
19   case UnknownArch: return "unknown";
20
21   case alpha:   return "alpha";
22   case arm:     return "arm";
23   case bfin:    return "bfin";
24   case cellspu: return "cellspu";
25   case mips:    return "mips";
26   case mipsel:  return "mipsel";
27   case msp430:  return "msp430";
28   case ppc64:   return "powerpc64";
29   case ppc:     return "powerpc";
30   case sparc:   return "sparc";
31   case sparcv9: return "sparcv9";
32   case systemz: return "s390x";
33   case tce:     return "tce";
34   case thumb:   return "thumb";
35   case x86:     return "i386";
36   case x86_64:  return "x86_64";
37   case xcore:   return "xcore";
38   case mblaze:  return "mblaze";
39   case ptx32:   return "ptx32";
40   case ptx64:   return "ptx64";
41   }
42
43   return "<invalid>";
44 }
45
46 const char *Triple::getArchTypePrefix(ArchType Kind) {
47   switch (Kind) {
48   default:
49     return 0;
50
51   case alpha:   return "alpha";
52
53   case arm:
54   case thumb:   return "arm";
55
56   case bfin:    return "bfin";
57
58   case cellspu: return "spu";
59
60   case ppc64:
61   case ppc:     return "ppc";
62
63   case mblaze:  return "mblaze";
64
65   case sparcv9:
66   case sparc:   return "sparc";
67
68   case x86:
69   case x86_64:  return "x86";
70
71   case xcore:   return "xcore";
72
73   case ptx32:   return "ptx";
74   case ptx64:   return "ptx";
75   }
76 }
77
78 const char *Triple::getVendorTypeName(VendorType Kind) {
79   switch (Kind) {
80   case UnknownVendor: return "unknown";
81
82   case Apple: return "apple";
83   case PC: return "pc";
84   case SCEI: return "scei";
85   }
86
87   return "<invalid>";
88 }
89
90 const char *Triple::getOSTypeName(OSType Kind) {
91   switch (Kind) {
92   case UnknownOS: return "unknown";
93
94   case AuroraUX: return "auroraux";
95   case Cygwin: return "cygwin";
96   case Darwin: return "darwin";
97   case DragonFly: return "dragonfly";
98   case FreeBSD: return "freebsd";
99   case IOS: return "ios";
100   case KFreeBSD: return "kfreebsd";
101   case Linux: return "linux";
102   case Lv2: return "lv2";
103   case MacOSX: return "macosx";
104   case MinGW32: return "mingw32";
105   case NetBSD: return "netbsd";
106   case OpenBSD: return "openbsd";
107   case Psp: return "psp";
108   case Solaris: return "solaris";
109   case Win32: return "win32";
110   case Haiku: return "haiku";
111   case Minix: return "minix";
112   case RTEMS: return "rtems";
113   case NativeClient: return "nacl";
114   }
115
116   return "<invalid>";
117 }
118
119 const char *Triple::getEnvironmentTypeName(EnvironmentType Kind) {
120   switch (Kind) {
121   case UnknownEnvironment: return "unknown";
122   case GNU: return "gnu";
123   case GNUEABI: return "gnueabi";
124   case EABI: return "eabi";
125   case MachO: return "macho";
126   }
127
128   return "<invalid>";
129 }
130
131 Triple::ArchType Triple::getArchTypeForLLVMName(StringRef Name) {
132   if (Name == "alpha")
133     return alpha;
134   if (Name == "arm")
135     return arm;
136   if (Name == "bfin")
137     return bfin;
138   if (Name == "cellspu")
139     return cellspu;
140   if (Name == "mips")
141     return mips;
142   if (Name == "mipsel")
143     return mipsel;
144   if (Name == "msp430")
145     return msp430;
146   if (Name == "ppc64")
147     return ppc64;
148   if (Name == "ppc32")
149     return ppc;
150   if (Name == "ppc")
151     return ppc;
152   if (Name == "mblaze")
153     return mblaze;
154   if (Name == "sparc")
155     return sparc;
156   if (Name == "sparcv9")
157     return sparcv9;
158   if (Name == "systemz")
159     return systemz;
160   if (Name == "tce")
161     return tce;
162   if (Name == "thumb")
163     return thumb;
164   if (Name == "x86")
165     return x86;
166   if (Name == "x86-64")
167     return x86_64;
168   if (Name == "xcore")
169     return xcore;
170   if (Name == "ptx32")
171     return ptx32;
172   if (Name == "ptx64")
173     return ptx64;
174
175   return UnknownArch;
176 }
177
178 Triple::ArchType Triple::getArchTypeForDarwinArchName(StringRef Str) {
179   // See arch(3) and llvm-gcc's driver-driver.c. We don't implement support for
180   // archs which Darwin doesn't use.
181
182   // The matching this routine does is fairly pointless, since it is neither the
183   // complete architecture list, nor a reasonable subset. The problem is that
184   // historically the driver driver accepts this and also ties its -march=
185   // handling to the architecture name, so we need to be careful before removing
186   // support for it.
187
188   // This code must be kept in sync with Clang's Darwin specific argument
189   // translation.
190
191   if (Str == "ppc" || Str == "ppc601" || Str == "ppc603" || Str == "ppc604" ||
192       Str == "ppc604e" || Str == "ppc750" || Str == "ppc7400" ||
193       Str == "ppc7450" || Str == "ppc970")
194     return Triple::ppc;
195
196   if (Str == "ppc64")
197     return Triple::ppc64;
198
199   if (Str == "i386" || Str == "i486" || Str == "i486SX" || Str == "pentium" ||
200       Str == "i586" || Str == "pentpro" || Str == "i686" || Str == "pentIIm3" ||
201       Str == "pentIIm5" || Str == "pentium4")
202     return Triple::x86;
203
204   if (Str == "x86_64")
205     return Triple::x86_64;
206
207   // This is derived from the driver driver.
208   if (Str == "arm" || Str == "armv4t" || Str == "armv5" || Str == "xscale" ||
209       Str == "armv6" || Str == "armv7" || Str == "armv7f" || Str == "armv7k" ||
210       Str == "armv7s")
211     return Triple::arm;
212
213   if (Str == "ptx32")
214     return Triple::ptx32;
215   if (Str == "ptx64")
216     return Triple::ptx64;
217
218   return Triple::UnknownArch;
219 }
220
221 // Returns architecture name that is understood by the target assembler.
222 const char *Triple::getArchNameForAssembler() {
223   if (!isOSDarwin() && getVendor() != Triple::Apple)
224     return NULL;
225
226   StringRef Str = getArchName();
227   if (Str == "i386")
228     return "i386";
229   if (Str == "x86_64")
230     return "x86_64";
231   if (Str == "powerpc")
232     return "ppc";
233   if (Str == "powerpc64")
234     return "ppc64";
235   if (Str == "mblaze" || Str == "microblaze")
236     return "mblaze";
237   if (Str == "arm")
238     return "arm";
239   if (Str == "armv4t" || Str == "thumbv4t")
240     return "armv4t";
241   if (Str == "armv5" || Str == "armv5e" || Str == "thumbv5"
242       || Str == "thumbv5e")
243     return "armv5";
244   if (Str == "armv6" || Str == "thumbv6")
245     return "armv6";
246   if (Str == "armv7" || Str == "thumbv7")
247     return "armv7";
248   if (Str == "ptx32")
249     return "ptx32";
250   if (Str == "ptx64")
251     return "ptx64";
252   return NULL;
253 }
254
255 //
256
257 Triple::ArchType Triple::ParseArch(StringRef ArchName) {
258   if (ArchName.size() == 4 && ArchName[0] == 'i' &&
259       ArchName[2] == '8' && ArchName[3] == '6' &&
260       ArchName[1] - '3' < 6) // i[3-9]86
261     return x86;
262   else if (ArchName == "amd64" || ArchName == "x86_64")
263     return x86_64;
264   else if (ArchName == "bfin")
265     return bfin;
266   else if (ArchName == "powerpc")
267     return ppc;
268   else if ((ArchName == "powerpc64") || (ArchName == "ppu"))
269     return ppc64;
270   else if (ArchName == "mblaze")
271     return mblaze;
272   else if (ArchName == "arm" ||
273            ArchName.startswith("armv") ||
274            ArchName == "xscale")
275     return arm;
276   else if (ArchName == "thumb" ||
277            ArchName.startswith("thumbv"))
278     return thumb;
279   else if (ArchName.startswith("alpha"))
280     return alpha;
281   else if (ArchName == "spu" || ArchName == "cellspu")
282     return cellspu;
283   else if (ArchName == "msp430")
284     return msp430;
285   else if (ArchName == "mips" || ArchName == "mipseb" ||
286            ArchName == "mipsallegrex")
287     return mips;
288   else if (ArchName == "mipsel" || ArchName == "mipsallegrexel" ||
289            ArchName == "psp")
290     return mipsel;
291   else if (ArchName == "sparc")
292     return sparc;
293   else if (ArchName == "sparcv9")
294     return sparcv9;
295   else if (ArchName == "s390x")
296     return systemz;
297   else if (ArchName == "tce")
298     return tce;
299   else if (ArchName == "xcore")
300     return xcore;
301   else if (ArchName == "ptx32")
302     return ptx32;
303   else if (ArchName == "ptx64")
304     return ptx64;
305   else
306     return UnknownArch;
307 }
308
309 Triple::VendorType Triple::ParseVendor(StringRef VendorName) {
310   if (VendorName == "apple")
311     return Apple;
312   else if (VendorName == "pc")
313     return PC;
314   else if (VendorName == "scei")
315     return SCEI;
316   else
317     return UnknownVendor;
318 }
319
320 Triple::OSType Triple::ParseOS(StringRef OSName) {
321   if (OSName.startswith("auroraux"))
322     return AuroraUX;
323   else if (OSName.startswith("cygwin"))
324     return Cygwin;
325   else if (OSName.startswith("darwin"))
326     return Darwin;
327   else if (OSName.startswith("dragonfly"))
328     return DragonFly;
329   else if (OSName.startswith("freebsd"))
330     return FreeBSD;
331   else if (OSName.startswith("ios"))
332     return IOS;
333   else if (OSName.startswith("kfreebsd"))
334     return KFreeBSD;
335   else if (OSName.startswith("linux"))
336     return Linux;
337   else if (OSName.startswith("lv2"))
338     return Lv2;
339   else if (OSName.startswith("macosx"))
340     return MacOSX;
341   else if (OSName.startswith("mingw32"))
342     return MinGW32;
343   else if (OSName.startswith("netbsd"))
344     return NetBSD;
345   else if (OSName.startswith("openbsd"))
346     return OpenBSD;
347   else if (OSName.startswith("psp"))
348     return Psp;
349   else if (OSName.startswith("solaris"))
350     return Solaris;
351   else if (OSName.startswith("win32"))
352     return Win32;
353   else if (OSName.startswith("haiku"))
354     return Haiku;
355   else if (OSName.startswith("minix"))
356     return Minix;
357   else if (OSName.startswith("rtems"))
358     return RTEMS;
359   else if (OSName.startswith("nacl"))
360     return NativeClient;
361   else
362     return UnknownOS;
363 }
364
365 Triple::EnvironmentType Triple::ParseEnvironment(StringRef EnvironmentName) {
366   if (EnvironmentName.startswith("eabi"))
367     return EABI;
368   else if (EnvironmentName.startswith("gnueabi"))
369     return GNUEABI;
370   else if (EnvironmentName.startswith("gnu"))
371     return GNU;
372   else if (EnvironmentName.startswith("macho"))
373     return MachO;
374   else
375     return UnknownEnvironment;
376 }
377
378 void Triple::Parse() const {
379   assert(!isInitialized() && "Invalid parse call.");
380
381   Arch = ParseArch(getArchName());
382   Vendor = ParseVendor(getVendorName());
383   OS = ParseOS(getOSName());
384   Environment = ParseEnvironment(getEnvironmentName());
385
386   assert(isInitialized() && "Failed to initialize!");
387 }
388
389 std::string Triple::normalize(StringRef Str) {
390   // Parse into components.
391   SmallVector<StringRef, 4> Components;
392   for (size_t First = 0, Last = 0; Last != StringRef::npos; First = Last + 1) {
393     Last = Str.find('-', First);
394     Components.push_back(Str.slice(First, Last));
395   }
396
397   // If the first component corresponds to a known architecture, preferentially
398   // use it for the architecture.  If the second component corresponds to a
399   // known vendor, preferentially use it for the vendor, etc.  This avoids silly
400   // component movement when a component parses as (eg) both a valid arch and a
401   // valid os.
402   ArchType Arch = UnknownArch;
403   if (Components.size() > 0)
404     Arch = ParseArch(Components[0]);
405   VendorType Vendor = UnknownVendor;
406   if (Components.size() > 1)
407     Vendor = ParseVendor(Components[1]);
408   OSType OS = UnknownOS;
409   if (Components.size() > 2)
410     OS = ParseOS(Components[2]);
411   EnvironmentType Environment = UnknownEnvironment;
412   if (Components.size() > 3)
413     Environment = ParseEnvironment(Components[3]);
414
415   // Note which components are already in their final position.  These will not
416   // be moved.
417   bool Found[4];
418   Found[0] = Arch != UnknownArch;
419   Found[1] = Vendor != UnknownVendor;
420   Found[2] = OS != UnknownOS;
421   Found[3] = Environment != UnknownEnvironment;
422
423   // If they are not there already, permute the components into their canonical
424   // positions by seeing if they parse as a valid architecture, and if so moving
425   // the component to the architecture position etc.
426   for (unsigned Pos = 0; Pos != array_lengthof(Found); ++Pos) {
427     if (Found[Pos])
428       continue; // Already in the canonical position.
429
430     for (unsigned Idx = 0; Idx != Components.size(); ++Idx) {
431       // Do not reparse any components that already matched.
432       if (Idx < array_lengthof(Found) && Found[Idx])
433         continue;
434
435       // Does this component parse as valid for the target position?
436       bool Valid = false;
437       StringRef Comp = Components[Idx];
438       switch (Pos) {
439       default:
440         assert(false && "unexpected component type!");
441       case 0:
442         Arch = ParseArch(Comp);
443         Valid = Arch != UnknownArch;
444         break;
445       case 1:
446         Vendor = ParseVendor(Comp);
447         Valid = Vendor != UnknownVendor;
448         break;
449       case 2:
450         OS = ParseOS(Comp);
451         Valid = OS != UnknownOS;
452         break;
453       case 3:
454         Environment = ParseEnvironment(Comp);
455         Valid = Environment != UnknownEnvironment;
456         break;
457       }
458       if (!Valid)
459         continue; // Nope, try the next component.
460
461       // Move the component to the target position, pushing any non-fixed
462       // components that are in the way to the right.  This tends to give
463       // good results in the common cases of a forgotten vendor component
464       // or a wrongly positioned environment.
465       if (Pos < Idx) {
466         // Insert left, pushing the existing components to the right.  For
467         // example, a-b-i386 -> i386-a-b when moving i386 to the front.
468         StringRef CurrentComponent(""); // The empty component.
469         // Replace the component we are moving with an empty component.
470         std::swap(CurrentComponent, Components[Idx]);
471         // Insert the component being moved at Pos, displacing any existing
472         // components to the right.
473         for (unsigned i = Pos; !CurrentComponent.empty(); ++i) {
474           // Skip over any fixed components.
475           while (i < array_lengthof(Found) && Found[i]) ++i;
476           // Place the component at the new position, getting the component
477           // that was at this position - it will be moved right.
478           std::swap(CurrentComponent, Components[i]);
479         }
480       } else if (Pos > Idx) {
481         // Push right by inserting empty components until the component at Idx
482         // reaches the target position Pos.  For example, pc-a -> -pc-a when
483         // moving pc to the second position.
484         do {
485           // Insert one empty component at Idx.
486           StringRef CurrentComponent(""); // The empty component.
487           for (unsigned i = Idx; i < Components.size();) {
488             // Place the component at the new position, getting the component
489             // that was at this position - it will be moved right.
490             std::swap(CurrentComponent, Components[i]);
491             // If it was placed on top of an empty component then we are done.
492             if (CurrentComponent.empty())
493               break;
494             // Advance to the next component, skipping any fixed components.
495             while (++i < array_lengthof(Found) && Found[i])
496               ;
497           }
498           // The last component was pushed off the end - append it.
499           if (!CurrentComponent.empty())
500             Components.push_back(CurrentComponent);
501
502           // Advance Idx to the component's new position.
503           while (++Idx < array_lengthof(Found) && Found[Idx]) {}
504         } while (Idx < Pos); // Add more until the final position is reached.
505       }
506       assert(Pos < Components.size() && Components[Pos] == Comp &&
507              "Component moved wrong!");
508       Found[Pos] = true;
509       break;
510     }
511   }
512
513   // Special case logic goes here.  At this point Arch, Vendor and OS have the
514   // correct values for the computed components.
515
516   // Stick the corrected components back together to form the normalized string.
517   std::string Normalized;
518   for (unsigned i = 0, e = Components.size(); i != e; ++i) {
519     if (i) Normalized += '-';
520     Normalized += Components[i];
521   }
522   return Normalized;
523 }
524
525 StringRef Triple::getArchName() const {
526   return StringRef(Data).split('-').first;           // Isolate first component
527 }
528
529 StringRef Triple::getVendorName() const {
530   StringRef Tmp = StringRef(Data).split('-').second; // Strip first component
531   return Tmp.split('-').first;                       // Isolate second component
532 }
533
534 StringRef Triple::getOSName() const {
535   StringRef Tmp = StringRef(Data).split('-').second; // Strip first component
536   Tmp = Tmp.split('-').second;                       // Strip second component
537   return Tmp.split('-').first;                       // Isolate third component
538 }
539
540 StringRef Triple::getEnvironmentName() const {
541   StringRef Tmp = StringRef(Data).split('-').second; // Strip first component
542   Tmp = Tmp.split('-').second;                       // Strip second component
543   return Tmp.split('-').second;                      // Strip third component
544 }
545
546 StringRef Triple::getOSAndEnvironmentName() const {
547   StringRef Tmp = StringRef(Data).split('-').second; // Strip first component
548   return Tmp.split('-').second;                      // Strip second component
549 }
550
551 static unsigned EatNumber(StringRef &Str) {
552   assert(!Str.empty() && Str[0] >= '0' && Str[0] <= '9' && "Not a number");
553   unsigned Result = 0;
554
555   do {
556     // Consume the leading digit.
557     Result = Result*10 + (Str[0] - '0');
558
559     // Eat the digit.
560     Str = Str.substr(1);
561   } while (!Str.empty() && Str[0] >= '0' && Str[0] <= '9');
562
563   return Result;
564 }
565
566 void Triple::getOSVersion(unsigned &Major, unsigned &Minor,
567                           unsigned &Micro) const {
568   StringRef OSName = getOSName();
569
570   // Assume that the OS portion of the triple starts with the canonical name.
571   StringRef OSTypeName = getOSTypeName(getOS());
572   if (OSName.startswith(OSTypeName))
573     OSName = OSName.substr(OSTypeName.size());
574
575   // Any unset version defaults to 0.
576   Major = Minor = Micro = 0;
577
578   // Parse up to three components.
579   unsigned *Components[3] = { &Major, &Minor, &Micro };
580   for (unsigned i = 0; i != 3; ++i) {
581     if (OSName.empty() || OSName[0] < '0' || OSName[0] > '9')
582       break;
583
584     // Consume the leading number.
585     *Components[i] = EatNumber(OSName);
586
587     // Consume the separator, if present.
588     if (OSName.startswith("."))
589       OSName = OSName.substr(1);
590   }
591 }
592
593 void Triple::setTriple(const Twine &Str) {
594   Data = Str.str();
595   Arch = InvalidArch;
596 }
597
598 void Triple::setArch(ArchType Kind) {
599   setArchName(getArchTypeName(Kind));
600 }
601
602 void Triple::setVendor(VendorType Kind) {
603   setVendorName(getVendorTypeName(Kind));
604 }
605
606 void Triple::setOS(OSType Kind) {
607   setOSName(getOSTypeName(Kind));
608 }
609
610 void Triple::setEnvironment(EnvironmentType Kind) {
611   setEnvironmentName(getEnvironmentTypeName(Kind));
612 }
613
614 void Triple::setArchName(StringRef Str) {
615   // Work around a miscompilation bug for Twines in gcc 4.0.3.
616   SmallString<64> Triple;
617   Triple += Str;
618   Triple += "-";
619   Triple += getVendorName();
620   Triple += "-";
621   Triple += getOSAndEnvironmentName();
622   setTriple(Triple.str());
623 }
624
625 void Triple::setVendorName(StringRef Str) {
626   setTriple(getArchName() + "-" + Str + "-" + getOSAndEnvironmentName());
627 }
628
629 void Triple::setOSName(StringRef Str) {
630   if (hasEnvironment())
631     setTriple(getArchName() + "-" + getVendorName() + "-" + Str +
632               "-" + getEnvironmentName());
633   else
634     setTriple(getArchName() + "-" + getVendorName() + "-" + Str);
635 }
636
637 void Triple::setEnvironmentName(StringRef Str) {
638   setTriple(getArchName() + "-" + getVendorName() + "-" + getOSName() +
639             "-" + Str);
640 }
641
642 void Triple::setOSAndEnvironmentName(StringRef Str) {
643   setTriple(getArchName() + "-" + getVendorName() + "-" + Str);
644 }