#include "stdio.h"
#include "stdlib.h"
#include "string.h"
#include "libc.h"
#include <sys/file.h>
#include <mach/mach.h>
#include "stuff/arch.h"
#include "stuff/errors.h"
#include "stuff/execute.h"
#include "stuff/allocate.h"
#include <mach-o/dyld.h>
char *progname = NULL;
int
main(
int argc,
char **argv,
char **envp)
{
const char *LIB = "../libexec/as/";
const char *LOCALLIB = "../local/libexec/as/";
const char *AS = "/as";
int i, j;
uint32_t count, verbose, run_clang;
char *p, c, *arch_name, *as, *as_local;
char **new_argv;
const char *CLANG = "clang";
char *prefix, buf[MAXPATHLEN], resolved_name[PATH_MAX];
uint32_t bufsize;
struct arch_flag arch_flag;
const struct arch_flag *arch_flags, *family_arch_flag;
enum bool oflag_specified, qflag, Qflag;
progname = argv[0];
arch_name = NULL;
verbose = 0;
run_clang = 0;
oflag_specified = FALSE;
qflag = FALSE;
Qflag = FALSE;
bufsize = MAXPATHLEN;
p = buf;
i = _NSGetExecutablePath(p, &bufsize);
if(i == -1){
p = allocate(bufsize);
_NSGetExecutablePath(p, &bufsize);
}
prefix = realpath(p, resolved_name);
if(realpath == NULL)
system_fatal("realpath(3) for %s failed", p);
p = rindex(prefix, '/');
if(p != NULL)
p[1] = '\0';
for(i = 1; i < argc; i++){
if(argv[i][0] == '-' &&
!(argv[i][1] == '-' && argv[i][2] == '0')){
for(p = &(argv[i][1]); (c = *p); p++){
c &= 0x7F;
switch(c){
case 'o':
oflag_specified = TRUE;
case 'I':
case 'm':
case 'N':
if(p[1] == '\0')
i++;
p = " ";
break;
case 'g':
if(strcmp(p, "gstabs") == 0 ||
strcmp(p, "gdwarf2") == 0 ||
strcmp(p, "gdwarf-2") == 0){
p = " ";
}
break;
case 'd':
if(strcmp(p, "dynamic") == 0){
p = " ";
}
break;
case 's':
if(strcmp(p, "static") == 0){
p = " ";
}
break;
case 'a':
if(strcmp(p, "arch_multiple") == 0){
p = " ";
}
if(strcmp(p, "arch") == 0){
if(i + 1 >= argc)
fatal("missing argument to %s option", argv[i]);
if(arch_name != NULL)
fatal("more than one %s option (not allowed, "
"use cc(1) instead)", argv[i]);
arch_name = argv[i+1];
p = " ";
i++;
break;
}
case 'f':
if(strcmp(p, "force_cpusubtype_ALL") == 0){
p = " ";
break;
}
case 'k':
case 'v':
case 'W':
case 'L':
case 'l':
default:
break;
case 'q':
qflag = TRUE;
break;
case 'Q':
Qflag = TRUE;
break;
case 'V':
verbose = 1;
break;
}
}
}
}
if(arch_name == NULL){
if(get_arch_from_host(&arch_flag, NULL)){
#if __LP64__
if(arch_flag.cputype == CPU_TYPE_I386)
arch_flag = *get_arch_family_from_cputype(CPU_TYPE_X86_64);
#endif
arch_name = arch_flag.name;
}
else
fatal("unknown host architecture (can't determine which "
"assembler to run)");
}
else{
if(get_arch_from_flag(arch_name, &arch_flag) != 0){
family_arch_flag =
get_arch_family_from_cputype(arch_flag.cputype);
if(family_arch_flag != NULL)
arch_name = (char *)(family_arch_flag->name);
}
}
if(qflag == TRUE && Qflag == TRUE){
printf("%s: can't specifiy both -q and -Q\n", progname);
exit(1);
}
if(Qflag == FALSE &&
getenv("AS_INTEGRATED_ASSEMBLER") != NULL &&
(arch_flag.cputype == CPU_TYPE_X86_64 ||
arch_flag.cputype == CPU_TYPE_I386 ||
arch_flag.cputype == CPU_TYPE_ARM)){
qflag = TRUE;
}
if(qflag == TRUE &&
(arch_flag.cputype != CPU_TYPE_X86_64 &&
arch_flag.cputype != CPU_TYPE_I386 &&
arch_flag.cputype != CPU_TYPE_ARM)){
printf("%s: can't specifiy -q with -arch %s\n", progname,
arch_flag.name);
exit(1);
}
#ifdef notyet
if(arch_flag.cputype == CPU_TYPE_X86_64 ||
arch_flag.cputype == CPU_TYPE_I386)
run_clang = 1;
#endif
if((run_clang || qflag) && !Qflag &&
(arch_flag.cputype == CPU_TYPE_X86_64 ||
arch_flag.cputype == CPU_TYPE_I386 ||
arch_flag.cputype == CPU_TYPE_ARM)){
as = makestr(prefix, CLANG, NULL);
if(access(as, F_OK) != 0){
printf("%s: assembler (%s) not installed\n", progname, as);
exit(1);
}
new_argv = allocate((argc + 5) * sizeof(char *));
new_argv[0] = as;
j = 1;
for(i = 1; i < argc; i++){
if(strcmp(argv[i], "-V") != 0 &&
strcmp(argv[i], "-q") != 0 &&
strcmp(argv[i], "-Q") != 0){
new_argv[j] = argv[i];
j++;
}
}
if(oflag_specified == FALSE){
new_argv[j] = "-o";
j++;
new_argv[j] = "a.out";
j++;
}
new_argv[j] = "-integrated-as";
j++;
new_argv[j] = "-c";
j++;
new_argv[j] = NULL;
if(execute(new_argv, verbose))
exit(0);
else
exit(1);
}
as = makestr(prefix, LIB, arch_name, AS, NULL);
new_argv = allocate((argc + 1) * sizeof(char *));
new_argv[0] = as;
j = 1;
for(i = 1; i < argc; i++){
if(strcmp(argv[i], "-q") != 0 &&
strcmp(argv[i], "-Q") != 0){
new_argv[j] = argv[i];
j++;
}
}
new_argv[j] = NULL;
if(access(as, F_OK) == 0){
argv[0] = as;
if(execute(new_argv, verbose))
exit(0);
else
exit(1);
}
as_local = makestr(prefix, LOCALLIB, arch_name, AS, NULL);
new_argv[0] = as_local;
if(access(as_local, F_OK) == 0){
argv[0] = as_local;
if(execute(new_argv, verbose))
exit(0);
else
exit(1);
}
printf("%s: assembler (%s or %s) for architecture %s not installed\n",
progname, as, as_local, arch_name);
arch_flags = get_arch_flags();
count = 0;
for(i = 0; arch_flags[i].name != NULL; i++){
as = makestr(prefix, LIB, arch_flags[i].name, AS, NULL);
if(access(as, F_OK) == 0){
if(count == 0)
printf("Installed assemblers are:\n");
printf("%s for architecture %s\n", as, arch_flags[i].name);
count++;
}
else{
as_local = makestr(prefix, LOCALLIB, arch_flags[i].name, AS,
NULL);
if(access(as_local, F_OK) == 0){
if(count == 0)
printf("Installed assemblers are:\n");
printf("%s for architecture %s\n", as_local,
arch_flags[i].name);
count++;
}
}
}
if(count == 0)
printf("%s: no assemblers installed\n", progname);
exit(1);
}