mirror of
https://github.com/danog/gllvm.git
synced 2025-01-22 21:51:39 +01:00
Move file types out to their own file.
This commit is contained in:
Ian A Mason
parent
d153584bba
commit
ec053c08e2
@ -113,10 +113,10 @@ func attachBitcodePathToObject(bcFile, objFile string) {
|
||||
var attachCmdArgs []string
|
||||
if runtime.GOOS == "darwin" {
|
||||
attachCmd = "ld"
|
||||
attachCmdArgs = []string{"-r", "-keep_private_externs", objFile, "-sectcreate", darwinSEGMENTNAME, darwinSECTIONNAME, tmpFile.Name(), "-o", objFile}
|
||||
attachCmdArgs = []string{"-r", "-keep_private_externs", objFile, "-sectcreate", DarwinSegmentName, DarwinSectionName, tmpFile.Name(), "-o", objFile}
|
||||
} else {
|
||||
attachCmd = "objcopy"
|
||||
attachCmdArgs = []string{"--add-section", elfSECTIONNAME + "=" + tmpFile.Name(), objFile}
|
||||
attachCmdArgs = []string{"--add-section", ELFSectionName + "=" + tmpFile.Name(), objFile}
|
||||
}
|
||||
|
||||
// Run the attach command and ignore errors
|
||||
|
||||
53
constants.go
53
constants.go
@ -1,46 +1,33 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"os"
|
||||
"os"
|
||||
)
|
||||
|
||||
const (
|
||||
elfSECTIONNAME = ".llvm_bc"
|
||||
darwinSEGMENTNAME = "__WLLVM"
|
||||
darwinSECTIONNAME = "__llvm_bc"
|
||||
ELFSectionName = ".llvm_bc"
|
||||
DarwinSegmentName = "__WLLVM"
|
||||
DarwinSectionName = "__llvm_bc"
|
||||
)
|
||||
|
||||
var LLVMToolChainBinDir = ""
|
||||
var LLVMCCName = ""
|
||||
var LLVMCXXName = ""
|
||||
var LLVMARName = ""
|
||||
var LLVMLINKName = ""
|
||||
var LLVMToolChainBinDir string
|
||||
var LLVMCCName string
|
||||
var LLVMCXXName string
|
||||
var LLVMARName string
|
||||
var LLVMLINKName string
|
||||
|
||||
var ConfigureOnly = ""
|
||||
var BitcodeStorePath = ""
|
||||
var ConfigureOnly string
|
||||
var BitcodeStorePath string
|
||||
|
||||
func init(){
|
||||
func init() {
|
||||
|
||||
LLVMToolChainBinDir = os.Getenv("GLLVM_TOOLS_PATH")
|
||||
LLVMCCName = os.Getenv("GLLVM_CC_NAME")
|
||||
LLVMCXXName = os.Getenv("GLLVM_CXX_NAME")
|
||||
LLVMARName = os.Getenv("GLLVM_AR_NAME")
|
||||
LLVMLINKName = os.Getenv("GLLVM_LINK_NAME")
|
||||
|
||||
ConfigureOnly = os.Getenv("GLLVM_CONFIGURE_ONLY")
|
||||
BitcodeStorePath = os.Getenv("GLLVM_BC_STORE")
|
||||
LLVMToolChainBinDir = os.Getenv("GLLVM_TOOLS_PATH")
|
||||
LLVMCCName = os.Getenv("GLLVM_CC_NAME")
|
||||
LLVMCXXName = os.Getenv("GLLVM_CXX_NAME")
|
||||
LLVMARName = os.Getenv("GLLVM_AR_NAME")
|
||||
LLVMLINKName = os.Getenv("GLLVM_LINK_NAME")
|
||||
|
||||
ConfigureOnly = os.Getenv("GLLVM_CONFIGURE_ONLY")
|
||||
BitcodeStorePath = os.Getenv("GLLVM_BC_STORE")
|
||||
}
|
||||
|
||||
|
||||
|
||||
const (
|
||||
// File types
|
||||
ftUNDEFINED = iota
|
||||
ftELFEXECUTABLE
|
||||
ftELFOBJECT
|
||||
ftELFSHARED
|
||||
ftMACHEXECUTABLE
|
||||
ftMACHOBJECT
|
||||
ftMACHSHARED
|
||||
ftARCHIVE
|
||||
)
|
||||
|
||||
62
extractor.go
62
extractor.go
@ -7,7 +7,6 @@ import (
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"os"
|
||||
"os/exec"
|
||||
"path"
|
||||
"path/filepath"
|
||||
"runtime"
|
||||
@ -32,14 +31,14 @@ func extract(args []string) {
|
||||
ea := parseExtractionArgs(args)
|
||||
|
||||
switch ea.InputType {
|
||||
case ftELFEXECUTABLE,
|
||||
ftELFSHARED,
|
||||
ftELFOBJECT,
|
||||
ftMACHEXECUTABLE,
|
||||
ftMACHSHARED,
|
||||
ftMACHOBJECT:
|
||||
case fileTypeELFEXECUTABLE,
|
||||
fileTypeELFSHARED,
|
||||
fileTypeELFOBJECT,
|
||||
fileTypeMACHEXECUTABLE,
|
||||
fileTypeMACHSHARED,
|
||||
fileTypeMACHOBJECT:
|
||||
handleExecutable(ea)
|
||||
case ftARCHIVE:
|
||||
case fileTypeARCHIVE:
|
||||
handleArchive(ea)
|
||||
default:
|
||||
log.Fatal("Incorrect input file type.")
|
||||
@ -119,21 +118,21 @@ func parseExtractionArgs(args []string) extractionArgs {
|
||||
} else {
|
||||
ea.ArArgs = append(ea.ArArgs, "x")
|
||||
}
|
||||
ea.ObjectTypeInArchive = ftELFOBJECT
|
||||
ea.ObjectTypeInArchive = fileTypeELFOBJECT
|
||||
case "darwin":
|
||||
ea.Extractor = extractSectionDarwin
|
||||
ea.ArArgs = append(ea.ArArgs, "-x")
|
||||
if ea.IsVerbose {
|
||||
ea.ArArgs = append(ea.ArArgs, "-v")
|
||||
}
|
||||
ea.ObjectTypeInArchive = ftMACHOBJECT
|
||||
ea.ObjectTypeInArchive = fileTypeMACHOBJECT
|
||||
default:
|
||||
log.Fatal("Unsupported platform: ", platform)
|
||||
}
|
||||
|
||||
// Create output filename if not given
|
||||
if ea.OutputFile == "" {
|
||||
if ea.InputType == ftARCHIVE {
|
||||
if ea.InputType == fileTypeARCHIVE {
|
||||
var ext string
|
||||
if ea.IsBuildBitcodeArchive {
|
||||
ext = ".a.bc"
|
||||
@ -187,8 +186,8 @@ func handleArchive(ea extractionArgs) {
|
||||
// Define object file handling closure
|
||||
var walkHandlingFunc = func(path string, info os.FileInfo, err error) error {
|
||||
if err == nil && !info.IsDir() {
|
||||
ft := getFileType(path)
|
||||
if ft == ea.ObjectTypeInArchive {
|
||||
fileType := getFileType(path)
|
||||
if fileType == ea.ObjectTypeInArchive {
|
||||
artifactPaths := ea.Extractor(path)
|
||||
for _, artPath := range artifactPaths {
|
||||
bcPath := resolveBitcodePath(artPath)
|
||||
@ -259,10 +258,10 @@ func extractSectionDarwin(inputFile string) (contents []string) {
|
||||
if err != nil {
|
||||
log.Fatal("Mach-O file ", inputFile, " could not be read.")
|
||||
}
|
||||
section := machoFile.Section(darwinSECTIONNAME)
|
||||
section := machoFile.Section(DarwinSectionName)
|
||||
sectionContents, errContents := section.Data()
|
||||
if errContents != nil {
|
||||
log.Fatal("Error reading the ", darwinSECTIONNAME, " section of Mach-O file ", inputFile, ".")
|
||||
log.Fatal("Error reading the ", DarwinSectionName, " section of Mach-O file ", inputFile, ".")
|
||||
}
|
||||
contents = strings.Split(strings.TrimSuffix(string(sectionContents), "\n"), "\n")
|
||||
return
|
||||
@ -273,10 +272,10 @@ func extractSectionUnix(inputFile string) (contents []string) {
|
||||
if err != nil {
|
||||
log.Fatal("ELF file ", inputFile, " could not be read.")
|
||||
}
|
||||
section := elfFile.Section(elfSECTIONNAME)
|
||||
section := elfFile.Section(ELFSectionName)
|
||||
sectionContents, errContents := section.Data()
|
||||
if errContents != nil {
|
||||
log.Fatal("Error reading the ", elfSECTIONNAME, " section of ELF file ", inputFile, ".")
|
||||
log.Fatal("Error reading the ", ELFSectionName, " section of ELF file ", inputFile, ".")
|
||||
}
|
||||
contents = strings.Split(strings.TrimSuffix(string(sectionContents), "\n"), "\n")
|
||||
return
|
||||
@ -300,35 +299,6 @@ func resolveBitcodePath(bcPath string) string {
|
||||
return bcPath
|
||||
}
|
||||
|
||||
func getFileType(realPath string) (fileType int) {
|
||||
// We need the file command to guess the file type
|
||||
cmd := exec.Command("file", realPath)
|
||||
out, err := cmd.Output()
|
||||
if err != nil {
|
||||
log.Fatal("There was an error getting the type of ", realPath, ". Make sure that the 'file' command is installed.")
|
||||
}
|
||||
|
||||
// Test the output
|
||||
if fo := string(out); strings.Contains(fo, "ELF") && strings.Contains(fo, "executable") {
|
||||
fileType = ftELFEXECUTABLE
|
||||
} else if strings.Contains(fo, "Mach-O") && strings.Contains(fo, "executable") {
|
||||
fileType = ftMACHEXECUTABLE
|
||||
} else if strings.Contains(fo, "ELF") && strings.Contains(fo, "shared") {
|
||||
fileType = ftELFSHARED
|
||||
} else if strings.Contains(fo, "Mach-O") && strings.Contains(fo, "dynamically linked shared") {
|
||||
fileType = ftMACHSHARED
|
||||
} else if strings.Contains(fo, "current ar archive") {
|
||||
fileType = ftARCHIVE
|
||||
} else if strings.Contains(fo, "ELF") && strings.Contains(fo, "relocatable") {
|
||||
fileType = ftELFOBJECT
|
||||
} else if strings.Contains(fo, "Mach-O") && strings.Contains(fo, "object") {
|
||||
fileType = ftMACHOBJECT
|
||||
} else {
|
||||
fileType = ftUNDEFINED
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func writeManifest(ea extractionArgs, bcFiles []string, artifactFiles []string) {
|
||||
section1 := "Physical location of extracted files:\n" + strings.Join(bcFiles, "\n") + "\n\n"
|
||||
|
||||
62
filetypes.go
Normal file
62
filetypes.go
Normal file
@ -0,0 +1,62 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"os"
|
||||
"os/exec"
|
||||
"log"
|
||||
"strings"
|
||||
)
|
||||
|
||||
const (
|
||||
// File types
|
||||
fileTypeUNDEFINED = iota
|
||||
fileTypeELFEXECUTABLE
|
||||
fileTypeELFOBJECT
|
||||
fileTypeELFSHARED
|
||||
fileTypeMACHEXECUTABLE
|
||||
fileTypeMACHOBJECT
|
||||
fileTypeMACHSHARED
|
||||
fileTypeARCHIVE
|
||||
)
|
||||
|
||||
func init() {
|
||||
|
||||
LLVMToolChainBinDir = os.Getenv("GLLVM_TOOLS_PATH")
|
||||
LLVMCCName = os.Getenv("GLLVM_CC_NAME")
|
||||
LLVMCXXName = os.Getenv("GLLVM_CXX_NAME")
|
||||
LLVMARName = os.Getenv("GLLVM_AR_NAME")
|
||||
LLVMLINKName = os.Getenv("GLLVM_LINK_NAME")
|
||||
|
||||
ConfigureOnly = os.Getenv("GLLVM_CONFIGURE_ONLY")
|
||||
BitcodeStorePath = os.Getenv("GLLVM_BC_STORE")
|
||||
}
|
||||
|
||||
func getFileType(realPath string) (fileType int) {
|
||||
// We need the file command to guess the file type
|
||||
cmd := exec.Command("file", realPath)
|
||||
out, err := cmd.Output()
|
||||
if err != nil {
|
||||
log.Fatal("There was an error getting the type of ", realPath, ". Make sure that the 'file' command is installed.")
|
||||
}
|
||||
|
||||
// Test the output
|
||||
if fo := string(out); strings.Contains(fo, "ELF") && strings.Contains(fo, "executable") {
|
||||
fileType = fileTypeELFEXECUTABLE
|
||||
} else if strings.Contains(fo, "Mach-O") && strings.Contains(fo, "executable") {
|
||||
fileType = fileTypeMACHEXECUTABLE
|
||||
} else if strings.Contains(fo, "ELF") && strings.Contains(fo, "shared") {
|
||||
fileType = fileTypeELFSHARED
|
||||
} else if strings.Contains(fo, "Mach-O") && strings.Contains(fo, "dynamically linked shared") {
|
||||
fileType = fileTypeMACHSHARED
|
||||
} else if strings.Contains(fo, "current ar archive") {
|
||||
fileType = fileTypeARCHIVE
|
||||
} else if strings.Contains(fo, "ELF") && strings.Contains(fo, "relocatable") {
|
||||
fileType = fileTypeELFOBJECT
|
||||
} else if strings.Contains(fo, "Mach-O") && strings.Contains(fo, "object") {
|
||||
fileType = fileTypeMACHOBJECT
|
||||
} else {
|
||||
fileType = fileTypeUNDEFINED
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user