Exemplo n.º 1
0
func main() {
	llvm.InitializeAllTargets()
	llvm.InitializeAllTargetMCs()
	llvm.InitializeAllTargetInfos()
	flag.Parse()
	if *version {
		displayVersion()
	}

	module, err := compileFiles(flag.Args())
	if err == nil {
		defer module.Dispose()
		if exitCode == 0 {
			if *dump {
				module.Dump()
			} else {
				err := llvm.WriteBitcodeToFile(module.Module, os.Stdout)
				if err != nil {
					fmt.Println(err)
				}
			}
		}
	} else {
		//fmt.Printf("llg.Compile(%v) failed: %v", file.Name, err)
		report(err)
	}
	os.Exit(exitCode)
}
Exemplo n.º 2
0
func link(files []string) error {
	tempf, err := ioutil.TempFile("", "llgolink")
	if err != nil {
		return err
	}
	defer os.Remove(tempf.Name())
	defer tempf.Close()

	// TODO? Fix order of modules to be based on package dependency.
	args := append([]string{"-o", tempf.Name()}, files...)
	err = exec.Command(llvmlink, args...).Run()
	if err != nil {
		return err
	}

	module, err := llvm.ParseBitcodeFile(tempf.Name())
	if err != nil {
		return err
	}
	defer module.Dispose()

	f := os.Stdout
	if outfile != "-" {
		f, err = os.Create(outfile)
		if err != nil {
			return err
		}
		defer f.Close()
	}
	return llvm.WriteBitcodeToFile(module, f)
}
Exemplo n.º 3
0
func link(files []string) error {
	tempf, err := ioutil.TempFile("", "llgolink")
	if err != nil {
		return err
	}
	defer os.Remove(tempf.Name())
	defer tempf.Close()

	args := append([]string{"-o", tempf.Name()}, files...)
	err = exec.Command(llvmlink, args...).Run()
	if err != nil {
		return err
	}

	module, err := llvm.ParseBitcodeFile(tempf.Name())
	if err != nil {
		return err
	}
	defer module.Dispose()

	err = reorderGlobalConstructors(module)
	if err != nil {
		return err
	}

	f := os.Stdout
	if outfile != "-" {
		f, err = os.Create(outfile)
		if err != nil {
			return err
		}
		defer f.Close()
	}
	return llvm.WriteBitcodeToFile(module, f)
}
Exemplo n.º 4
0
func getRuntimeModuleFile() (string, error) {
	if runtimemodulefile != "" {
		return runtimemodulefile, nil
	}

	gofiles, llfiles, cfiles, err := getRuntimeFiles()
	if err != nil {
		return "", err
	}

	var runtimeModule *llgo.Module
	runtimeModule, err = compileFiles(testCompiler, gofiles, "runtime")
	defer runtimeModule.Dispose()
	if err != nil {
		return "", err
	}

	outfile := filepath.Join(tempdir, "runtime.bc")
	f, err := os.Create(outfile)
	if err != nil {
		return "", err
	}
	err = llvm.WriteBitcodeToFile(runtimeModule.Module, f)
	if err != nil {
		f.Close()
		return "", err
	}
	f.Close()

	for i, cfile := range cfiles {
		bcfile := filepath.Join(tempdir, fmt.Sprintf("%d.bc", i))
		args := []string{"-c", "-emit-llvm", "-o", bcfile, cfile}
		if runtime.GOOS != "darwin" {
			// TODO(q): -g breaks badly on my system at the moment,
			// so is not enabled on darwin for now
			args = append([]string{"-g"}, args...)
		}
		cmd := exec.Command("clang", args...)
		cmd.Stderr = os.Stderr
		if err := cmd.Run(); err != nil {
			return "", fmt.Errorf("clang failed: %s", err)
		}
		llfiles = append(llfiles, bcfile)
	}

	if llfiles != nil {
		args := append([]string{"-o", outfile, outfile}, llfiles...)
		cmd := exec.Command("llvm-link", args...)
		output, err := cmd.CombinedOutput()
		if err != nil {
			err = fmt.Errorf("llvm-link failed: %s", err)
			fmt.Fprintf(os.Stderr, string(output))
			return "", err
		}
	}

	runtimemodulefile = outfile
	return runtimemodulefile, nil
}
Exemplo n.º 5
0
Arquivo: llgo.go Projeto: spate/llgo
func writeObjectFile(m *llgo.Module) error {
	var outfile *os.File
	switch *outputFile {
	case "-":
		outfile = os.Stdout
	default:
		var err error
		outfile, err = os.Create(*outputFile)
		if err != nil {
			return err
		}
	}
	return llvm.WriteBitcodeToFile(m.Module, outfile)
}
Exemplo n.º 6
0
func getRuntimeModule() (llvm.Module, error) {
	gofiles, llfiles, err := getRuntimeFiles()
	if err != nil {
		return llvm.Module{}, err
	}

	var runtimeModule *llgo.Module
	runtimeModule, err = compileFiles(testCompiler, gofiles, "runtime")
	if err != nil {
		return llvm.Module{}, err
	}

	if llfiles != nil {
		outfile, err := ioutil.TempFile("", "runtime.ll")
		if err != nil {
			runtimeModule.Dispose()
			return llvm.Module{}, err
		}
		defer func() {
			outfile.Close()
			os.Remove(outfile.Name())
		}()
		err = llvm.WriteBitcodeToFile(runtimeModule.Module, outfile)
		if err != nil {
			runtimeModule.Dispose()
			return llvm.Module{}, err
		}
		runtimeModule.Dispose()
		o := outfile.Name()
		for _, llfile := range llfiles {
			cmd := exec.Command("llvm-link", "-o", o, o, llfile)
			output, err := cmd.CombinedOutput()
			if err != nil {
				err = fmt.Errorf("llvm-link failed: %s", err)
				fmt.Fprintf(os.Stderr, string(output))
				return llvm.Module{}, err
			}
		}
		runtimeModule.Module, err = llvm.ParseBitcodeFile(o)
		if err != nil {
			return llvm.Module{}, err
		}
	}

	return runtimeModule.Module, nil
}
Exemplo n.º 7
0
func runMainFunction(m *llgo.Module) (output []string, err error) {
	runtimeModule, err := getRuntimeModuleFile()
	if err != nil {
		return
	}

	err = llvm.VerifyModule(m.Module, llvm.ReturnStatusAction)
	if err != nil {
		return nil, fmt.Errorf("Verification failed: %v", err)
	}

	bcpath := filepath.Join(tempdir, "test.bc")
	bcfile, err := os.Create(bcpath)
	if err != nil {
		return nil, err
	}
	llvm.WriteBitcodeToFile(m.Module, bcfile)
	bcfile.Close()

	cmd := exec.Command("llvm-link", "-o", bcpath, bcpath, runtimeModule)
	data, err := cmd.CombinedOutput()
	if err != nil {
		output = strings.Split(strings.TrimSpace(string(data)), "\n")
		return output, fmt.Errorf("llvm-link failed: %v", err)
	}

	exepath := filepath.Join(tempdir, "test")
	args := []string{"-pthread", "-o", exepath, bcpath}
	if runtime.GOOS != "darwin" {
		// TODO(q): -g breaks badly on my system at the moment, so is not enabled on darwin for now
		args = append([]string{"-g"}, args...)
	}

	cmd = exec.Command("clang++", args...)
	data, err = cmd.CombinedOutput()
	if err != nil {
		output = strings.Split(strings.TrimSpace(string(data)), "\n")
		return output, fmt.Errorf("clang++ failed: %v", err)
	}

	cmd = exec.Command(exepath)
	data, err = cmd.CombinedOutput()
	output = strings.Split(strings.TrimSpace(string(data)), "\n")
	return output, err
}
Exemplo n.º 8
0
Arquivo: llgo.go Projeto: pcc/llgo
func writeObjectFile(m *llgo.Module) error {
	var outfile *os.File
	switch *outputFile {
	case "-":
		outfile = os.Stdout
	default:
		var err error
		outfile, err = os.Create(*outputFile)
		if err != nil {
			return err
		}
	}
	err := llvm.VerifyModule(m.Module, llvm.ReturnStatusAction)
	if err != nil {
		return fmt.Errorf("Verification failed: %v", err)
	}
	return llvm.WriteBitcodeToFile(m.Module, outfile)
}
Exemplo n.º 9
0
func runMainFunction(m *llgo.Module) (output []string, err error) {
	runtimeModule, err := getRuntimeModuleFile()
	if err != nil {
		return
	}

	err = llvm.VerifyModule(m.Module, llvm.ReturnStatusAction)
	if err != nil {
		err = fmt.Errorf("Verification failed: %v", err)
		return
	}

	bcpath := filepath.Join(tempdir, "test.bc")
	bcfile, err := os.Create(bcpath)
	if err != nil {
		return nil, err
	}
	llvm.WriteBitcodeToFile(m.Module, bcfile)
	bcfile.Close()

	cmd := exec.Command("llvm-link", "-o", bcpath, bcpath, runtimeModule)
	cmd.Stderr = os.Stderr
	err = cmd.Run()
	if err != nil {
		return
	}

	exepath := filepath.Join(tempdir, "test")
	cmd = exec.Command("clang++", "-pthread", "-g", "-o", exepath, bcpath)
	cmd.Stderr = os.Stderr
	err = cmd.Run()
	if err != nil {
		return
	}

	cmd = exec.Command(exepath)
	data, err := cmd.Output()
	if err != nil {
		return
	}
	output = strings.Split(strings.TrimSpace(string(data)), "\n")
	return
}