1//===- Object.cpp - C bindings to the object file library--------*- C++ -*-===// 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// This file defines the C bindings to the file-format-independent object 11// library. 12// 13//===----------------------------------------------------------------------===// 14 15#include "llvm/ADT/SmallVector.h" 16#include "llvm-c/Object.h" 17#include "llvm/Object/ObjectFile.h" 18 19using namespace llvm; 20using namespace object; 21 22inline OwningBinary<ObjectFile> *unwrap(LLVMObjectFileRef OF) { 23 return reinterpret_cast<OwningBinary<ObjectFile> *>(OF); 24} 25 26inline LLVMObjectFileRef wrap(const OwningBinary<ObjectFile> *OF) { 27 return reinterpret_cast<LLVMObjectFileRef>( 28 const_cast<OwningBinary<ObjectFile> *>(OF)); 29} 30 31inline section_iterator *unwrap(LLVMSectionIteratorRef SI) { 32 return reinterpret_cast<section_iterator*>(SI); 33} 34 35inline LLVMSectionIteratorRef 36wrap(const section_iterator *SI) { 37 return reinterpret_cast<LLVMSectionIteratorRef> 38 (const_cast<section_iterator*>(SI)); 39} 40 41inline symbol_iterator *unwrap(LLVMSymbolIteratorRef SI) { 42 return reinterpret_cast<symbol_iterator*>(SI); 43} 44 45inline LLVMSymbolIteratorRef 46wrap(const symbol_iterator *SI) { 47 return reinterpret_cast<LLVMSymbolIteratorRef> 48 (const_cast<symbol_iterator*>(SI)); 49} 50 51inline relocation_iterator *unwrap(LLVMRelocationIteratorRef SI) { 52 return reinterpret_cast<relocation_iterator*>(SI); 53} 54 55inline LLVMRelocationIteratorRef 56wrap(const relocation_iterator *SI) { 57 return reinterpret_cast<LLVMRelocationIteratorRef> 58 (const_cast<relocation_iterator*>(SI)); 59} 60 61// ObjectFile creation 62LLVMObjectFileRef LLVMCreateObjectFile(LLVMMemoryBufferRef MemBuf) { 63 std::unique_ptr<MemoryBuffer> Buf(unwrap(MemBuf)); 64 ErrorOr<std::unique_ptr<ObjectFile>> ObjOrErr( 65 ObjectFile::createObjectFile(Buf->getMemBufferRef())); 66 std::unique_ptr<ObjectFile> Obj; 67 if (!ObjOrErr) 68 return nullptr; 69 70 auto *Ret = new OwningBinary<ObjectFile>(std::move(ObjOrErr.get()), std::move(Buf)); 71 return wrap(Ret); 72} 73 74void LLVMDisposeObjectFile(LLVMObjectFileRef ObjectFile) { 75 delete unwrap(ObjectFile); 76} 77 78// ObjectFile Section iterators 79LLVMSectionIteratorRef LLVMGetSections(LLVMObjectFileRef OF) { 80 OwningBinary<ObjectFile> *OB = unwrap(OF); 81 section_iterator SI = OB->getBinary()->section_begin(); 82 return wrap(new section_iterator(SI)); 83} 84 85void LLVMDisposeSectionIterator(LLVMSectionIteratorRef SI) { 86 delete unwrap(SI); 87} 88 89LLVMBool LLVMIsSectionIteratorAtEnd(LLVMObjectFileRef OF, 90 LLVMSectionIteratorRef SI) { 91 OwningBinary<ObjectFile> *OB = unwrap(OF); 92 return (*unwrap(SI) == OB->getBinary()->section_end()) ? 1 : 0; 93} 94 95void LLVMMoveToNextSection(LLVMSectionIteratorRef SI) { 96 ++(*unwrap(SI)); 97} 98 99void LLVMMoveToContainingSection(LLVMSectionIteratorRef Sect, 100 LLVMSymbolIteratorRef Sym) { 101 if (std::error_code ec = (*unwrap(Sym))->getSection(*unwrap(Sect))) 102 report_fatal_error(ec.message()); 103} 104 105// ObjectFile Symbol iterators 106LLVMSymbolIteratorRef LLVMGetSymbols(LLVMObjectFileRef OF) { 107 OwningBinary<ObjectFile> *OB = unwrap(OF); 108 symbol_iterator SI = OB->getBinary()->symbol_begin(); 109 return wrap(new symbol_iterator(SI)); 110} 111 112void LLVMDisposeSymbolIterator(LLVMSymbolIteratorRef SI) { 113 delete unwrap(SI); 114} 115 116LLVMBool LLVMIsSymbolIteratorAtEnd(LLVMObjectFileRef OF, 117 LLVMSymbolIteratorRef SI) { 118 OwningBinary<ObjectFile> *OB = unwrap(OF); 119 return (*unwrap(SI) == OB->getBinary()->symbol_end()) ? 1 : 0; 120} 121 122void LLVMMoveToNextSymbol(LLVMSymbolIteratorRef SI) { 123 ++(*unwrap(SI)); 124} 125 126// SectionRef accessors 127const char *LLVMGetSectionName(LLVMSectionIteratorRef SI) { 128 StringRef ret; 129 if (std::error_code ec = (*unwrap(SI))->getName(ret)) 130 report_fatal_error(ec.message()); 131 return ret.data(); 132} 133 134uint64_t LLVMGetSectionSize(LLVMSectionIteratorRef SI) { 135 return (*unwrap(SI))->getSize(); 136} 137 138const char *LLVMGetSectionContents(LLVMSectionIteratorRef SI) { 139 StringRef ret; 140 if (std::error_code ec = (*unwrap(SI))->getContents(ret)) 141 report_fatal_error(ec.message()); 142 return ret.data(); 143} 144 145uint64_t LLVMGetSectionAddress(LLVMSectionIteratorRef SI) { 146 return (*unwrap(SI))->getAddress(); 147} 148 149LLVMBool LLVMGetSectionContainsSymbol(LLVMSectionIteratorRef SI, 150 LLVMSymbolIteratorRef Sym) { 151 return (*unwrap(SI))->containsSymbol(**unwrap(Sym)); 152} 153 154// Section Relocation iterators 155LLVMRelocationIteratorRef LLVMGetRelocations(LLVMSectionIteratorRef Section) { 156 relocation_iterator SI = (*unwrap(Section))->relocation_begin(); 157 return wrap(new relocation_iterator(SI)); 158} 159 160void LLVMDisposeRelocationIterator(LLVMRelocationIteratorRef SI) { 161 delete unwrap(SI); 162} 163 164LLVMBool LLVMIsRelocationIteratorAtEnd(LLVMSectionIteratorRef Section, 165 LLVMRelocationIteratorRef SI) { 166 return (*unwrap(SI) == (*unwrap(Section))->relocation_end()) ? 1 : 0; 167} 168 169void LLVMMoveToNextRelocation(LLVMRelocationIteratorRef SI) { 170 ++(*unwrap(SI)); 171} 172 173 174// SymbolRef accessors 175const char *LLVMGetSymbolName(LLVMSymbolIteratorRef SI) { 176 StringRef ret; 177 if (std::error_code ec = (*unwrap(SI))->getName(ret)) 178 report_fatal_error(ec.message()); 179 return ret.data(); 180} 181 182uint64_t LLVMGetSymbolAddress(LLVMSymbolIteratorRef SI) { 183 uint64_t ret; 184 if (std::error_code ec = (*unwrap(SI))->getAddress(ret)) 185 report_fatal_error(ec.message()); 186 return ret; 187} 188 189uint64_t LLVMGetSymbolSize(LLVMSymbolIteratorRef SI) { 190 uint64_t ret; 191 if (std::error_code ec = (*unwrap(SI))->getSize(ret)) 192 report_fatal_error(ec.message()); 193 return ret; 194} 195 196// RelocationRef accessors 197uint64_t LLVMGetRelocationAddress(LLVMRelocationIteratorRef RI) { 198 uint64_t ret; 199 if (std::error_code ec = (*unwrap(RI))->getAddress(ret)) 200 report_fatal_error(ec.message()); 201 return ret; 202} 203 204uint64_t LLVMGetRelocationOffset(LLVMRelocationIteratorRef RI) { 205 uint64_t ret; 206 if (std::error_code ec = (*unwrap(RI))->getOffset(ret)) 207 report_fatal_error(ec.message()); 208 return ret; 209} 210 211LLVMSymbolIteratorRef LLVMGetRelocationSymbol(LLVMRelocationIteratorRef RI) { 212 symbol_iterator ret = (*unwrap(RI))->getSymbol(); 213 return wrap(new symbol_iterator(ret)); 214} 215 216uint64_t LLVMGetRelocationType(LLVMRelocationIteratorRef RI) { 217 uint64_t ret; 218 if (std::error_code ec = (*unwrap(RI))->getType(ret)) 219 report_fatal_error(ec.message()); 220 return ret; 221} 222 223// NOTE: Caller takes ownership of returned string. 224const char *LLVMGetRelocationTypeName(LLVMRelocationIteratorRef RI) { 225 SmallVector<char, 0> ret; 226 if (std::error_code ec = (*unwrap(RI))->getTypeName(ret)) 227 report_fatal_error(ec.message()); 228 229 char *str = static_cast<char*>(malloc(ret.size())); 230 std::copy(ret.begin(), ret.end(), str); 231 return str; 232} 233 234// NOTE: Caller takes ownership of returned string. 235const char *LLVMGetRelocationValueString(LLVMRelocationIteratorRef RI) { 236 SmallVector<char, 0> ret; 237 if (std::error_code ec = (*unwrap(RI))->getValueString(ret)) 238 report_fatal_error(ec.message()); 239 240 char *str = static_cast<char*>(malloc(ret.size())); 241 std::copy(ret.begin(), ret.end(), str); 242 return str; 243} 244 245