Carl Shapiro | 1fb8620 | 2011-06-27 17:43:13 -0700 | [diff] [blame] | 1 | // Copyright 2011 Google Inc. All Rights Reserved. |
| 2 | |
Brian Carlstrom | 578bbdc | 2011-07-21 14:07:47 -0700 | [diff] [blame] | 3 | #include "dex_file.h" |
Brian Carlstrom | 7e49dca | 2011-07-22 18:07:34 -0700 | [diff] [blame^] | 4 | |
| 5 | #include <fcntl.h> |
| 6 | #include <string.h> |
| 7 | #include <sys/mman.h> |
| 8 | #include <sys/stat.h> |
| 9 | #include <sys/types.h> |
| 10 | #include <map> |
| 11 | |
Brian Carlstrom | 578bbdc | 2011-07-21 14:07:47 -0700 | [diff] [blame] | 12 | #include "globals.h" |
| 13 | #include "logging.h" |
| 14 | #include "object.h" |
Brian Carlstrom | 7e49dca | 2011-07-22 18:07:34 -0700 | [diff] [blame^] | 15 | #include "scoped_ptr.h" |
| 16 | #include "utils.h" |
Carl Shapiro | 1fb8620 | 2011-06-27 17:43:13 -0700 | [diff] [blame] | 17 | |
| 18 | namespace art { |
| 19 | |
Brian Carlstrom | 7e49dca | 2011-07-22 18:07:34 -0700 | [diff] [blame^] | 20 | const byte RawDexFile::kDexMagic[] = { 'd', 'e', 'x', '\n' }; |
| 21 | const byte RawDexFile::kDexMagicVersion[] = { '0', '3', '5', '\0' }; |
| 22 | |
| 23 | RawDexFile::Closer::~Closer() {} |
| 24 | |
| 25 | RawDexFile::MmapCloser::MmapCloser(void* addr, size_t length) : addr_(addr), length_(length) { |
| 26 | CHECK(addr != NULL); |
| 27 | } |
| 28 | RawDexFile::MmapCloser::~MmapCloser() { |
| 29 | if (munmap(addr_, length_) == -1) { |
| 30 | PLOG(INFO) << "munmap failed"; |
| 31 | } |
| 32 | } |
| 33 | |
| 34 | RawDexFile::PtrCloser::PtrCloser(byte* addr) : addr_(addr) {} |
| 35 | RawDexFile::PtrCloser::~PtrCloser() { delete[] addr_; } |
| 36 | |
| 37 | RawDexFile* RawDexFile::OpenFile(const char* filename) { |
| 38 | CHECK(filename != NULL); |
| 39 | int fd = open(filename, O_RDONLY); // TODO: scoped_fd |
| 40 | if (fd == -1) { |
| 41 | PLOG(ERROR) << "open(\"" << filename << "\", O_RDONLY) failed"; |
| 42 | return NULL; |
| 43 | } |
| 44 | struct stat sbuf; |
| 45 | memset(&sbuf, 0, sizeof(sbuf)); |
| 46 | if (fstat(fd, &sbuf) == -1) { |
| 47 | PLOG(ERROR) << "fstat \"" << filename << "\" failed"; |
| 48 | close(fd); |
| 49 | return NULL; |
| 50 | } |
| 51 | size_t length = sbuf.st_size; |
| 52 | void* addr = mmap(NULL, length, PROT_READ, MAP_SHARED, fd, 0); |
| 53 | if (addr == MAP_FAILED) { |
| 54 | PLOG(ERROR) << "mmap \"" << filename << "\" failed"; |
| 55 | close(fd); |
| 56 | return NULL; |
| 57 | } |
| 58 | close(fd); |
| 59 | byte* dex_file = reinterpret_cast<byte*>(addr); |
| 60 | Closer* closer = new MmapCloser(addr, length); |
| 61 | return Open(dex_file, length, closer); |
| 62 | } |
| 63 | |
| 64 | RawDexFile* RawDexFile::OpenPtr(byte* ptr, size_t length) { |
| 65 | CHECK(ptr != NULL); |
| 66 | RawDexFile::Closer* closer = new PtrCloser(ptr); |
| 67 | return Open(ptr, length, closer); |
| 68 | } |
| 69 | |
| 70 | RawDexFile* RawDexFile::Open(const byte* dex_file, size_t length, |
| 71 | Closer* closer) { |
| 72 | scoped_ptr<RawDexFile> raw(new RawDexFile(dex_file, length, closer)); |
| 73 | if (!raw->Init()) { |
| 74 | return NULL; |
| 75 | } else { |
| 76 | return raw.release(); |
| 77 | } |
| 78 | } |
| 79 | |
| 80 | RawDexFile::~RawDexFile() {} |
| 81 | |
| 82 | bool RawDexFile::Init() { |
| 83 | InitMembers(); |
| 84 | if (!IsMagicValid()) { |
| 85 | return false; |
| 86 | } |
| 87 | InitIndex(); |
| 88 | return true; |
| 89 | } |
| 90 | |
| 91 | void RawDexFile::InitMembers() { |
| 92 | const byte* b = base_; |
| 93 | header_ = reinterpret_cast<const Header*>(b); |
| 94 | const Header* h = header_; |
| 95 | string_ids_ = reinterpret_cast<const StringId*>(b + h->string_ids_off_); |
| 96 | type_ids_ = reinterpret_cast<const TypeId*>(b + h->type_ids_off_); |
| 97 | field_ids_ = reinterpret_cast<const FieldId*>(b + h->field_ids_off_); |
| 98 | method_ids_ = reinterpret_cast<const MethodId*>(b + h->method_ids_off_); |
| 99 | proto_ids_ = reinterpret_cast<const ProtoId*>(b + h->proto_ids_off_); |
| 100 | class_defs_ = reinterpret_cast<const ClassDef*>(b + h->class_defs_off_); |
| 101 | } |
| 102 | |
| 103 | bool RawDexFile::IsMagicValid() { |
| 104 | return CheckMagic(header_->magic_); |
| 105 | } |
| 106 | |
| 107 | bool RawDexFile::CheckMagic(const byte* magic) { |
| 108 | CHECK(magic != NULL); |
| 109 | if (memcmp(magic, kDexMagic, sizeof(kDexMagic)) != 0) { |
| 110 | LOG(WARNING) << "Unrecognized magic number:" |
| 111 | << " " << magic[0] |
| 112 | << " " << magic[1] |
| 113 | << " " << magic[2] |
| 114 | << " " << magic[3]; |
| 115 | return false; |
| 116 | } |
| 117 | const byte* version = &magic[sizeof(kDexMagic)]; |
| 118 | if (memcmp(version, kDexMagicVersion, sizeof(kDexMagicVersion)) != 0) { |
| 119 | LOG(WARNING) << "Unrecognized version number:" |
| 120 | << " " << version[0] |
| 121 | << " " << version[1] |
| 122 | << " " << version[2] |
| 123 | << " " << version[3]; |
| 124 | return false; |
| 125 | } |
| 126 | return true; |
| 127 | } |
| 128 | |
| 129 | void RawDexFile::InitIndex() { |
| 130 | CHECK_EQ(index_.size(), 0U); |
| 131 | for (size_t i = 0; i < NumClassDefs(); ++i) { |
| 132 | const ClassDef& class_def = GetClassDef(i); |
| 133 | const char* descriptor = GetClassDescriptor(class_def); |
| 134 | index_[descriptor] = &class_def; |
| 135 | } |
| 136 | } |
| 137 | |
| 138 | const RawDexFile::ClassDef* RawDexFile::FindClassDef(const StringPiece& descriptor) const { |
| 139 | CHECK(descriptor != NULL); |
| 140 | Index::const_iterator it = index_.find(descriptor); |
| 141 | if (it == index_.end()) { |
| 142 | return NULL; |
| 143 | } else { |
| 144 | return it->second; |
| 145 | } |
| 146 | } |
| 147 | |
| 148 | // Read a signed integer. "zwidth" is the zero-based byte count. |
| 149 | static int32_t ReadSignedInt(const byte* ptr, int zwidth) |
| 150 | { |
| 151 | int32_t val = 0; |
| 152 | for (int i = zwidth; i >= 0; --i) { |
| 153 | val = ((uint32_t)val >> 8) | (((int32_t)*ptr++) << 24); |
| 154 | } |
| 155 | val >>= (3 - zwidth) * 8; |
| 156 | return val; |
| 157 | } |
| 158 | |
| 159 | // Read an unsigned integer. "zwidth" is the zero-based byte count, |
| 160 | // "fill_on_right" indicates which side we want to zero-fill from. |
| 161 | static uint32_t ReadUnsignedInt(const byte* ptr, int zwidth, |
| 162 | bool fill_on_right) { |
| 163 | uint32_t val = 0; |
| 164 | if (!fill_on_right) { |
| 165 | for (int i = zwidth; i >= 0; --i) { |
| 166 | val = (val >> 8) | (((uint32_t)*ptr++) << 24); |
| 167 | } |
| 168 | val >>= (3 - zwidth) * 8; |
| 169 | } else { |
| 170 | for (int i = zwidth; i >= 0; --i) { |
| 171 | val = (val >> 8) | (((uint32_t)*ptr++) << 24); |
| 172 | } |
| 173 | } |
| 174 | return val; |
| 175 | } |
| 176 | |
| 177 | // Read a signed long. "zwidth" is the zero-based byte count. |
| 178 | static int64_t ReadSignedLong(const byte* ptr, int zwidth) { |
| 179 | int64_t val = 0; |
| 180 | for (int i = zwidth; i >= 0; --i) { |
| 181 | val = ((uint64_t)val >> 8) | (((int64_t)*ptr++) << 56); |
| 182 | } |
| 183 | val >>= (7 - zwidth) * 8; |
| 184 | return val; |
| 185 | } |
| 186 | |
| 187 | // Read an unsigned long. "zwidth" is the zero-based byte count, |
| 188 | // "fill_on_right" indicates which side we want to zero-fill from. |
| 189 | static uint64_t ReadUnsignedLong(const byte* ptr, int zwidth, |
| 190 | bool fill_on_right) { |
| 191 | uint64_t val = 0; |
| 192 | if (!fill_on_right) { |
| 193 | for (int i = zwidth; i >= 0; --i) { |
| 194 | val = (val >> 8) | (((uint64_t)*ptr++) << 56); |
| 195 | } |
| 196 | val >>= (7 - zwidth) * 8; |
| 197 | } else { |
| 198 | for (int i = zwidth; i >= 0; --i) { |
| 199 | val = (val >> 8) | (((uint64_t)*ptr++) << 56); |
| 200 | } |
| 201 | } |
| 202 | return val; |
| 203 | } |
| 204 | |
| 205 | RawDexFile::ValueType RawDexFile::ReadEncodedValue(const byte** stream, |
| 206 | JValue* value) const { |
| 207 | const byte* ptr = *stream; |
| 208 | byte value_type = *ptr++; |
| 209 | byte value_arg = value_type >> kEncodedValueArgShift; |
| 210 | size_t width = value_arg + 1; // assume and correct later |
| 211 | int type = value_type & kEncodedValueTypeMask; |
| 212 | switch (type) { |
| 213 | case RawDexFile::kByte: { |
| 214 | int32_t b = ReadSignedInt(ptr, value_arg); |
| 215 | CHECK(IsInt(8, b)); |
| 216 | value->i = b; |
| 217 | break; |
| 218 | } |
| 219 | case RawDexFile::kShort: { |
| 220 | int32_t s = ReadSignedInt(ptr, value_arg); |
| 221 | CHECK(IsInt(16, s)); |
| 222 | value->i = s; |
| 223 | break; |
| 224 | } |
| 225 | case RawDexFile::kChar: { |
| 226 | uint32_t c = ReadUnsignedInt(ptr, value_arg, false); |
| 227 | CHECK(IsUint(16, c)); |
| 228 | value->i = c; |
| 229 | break; |
| 230 | } |
| 231 | case RawDexFile::kInt: |
| 232 | value->i = ReadSignedInt(ptr, value_arg); |
| 233 | break; |
| 234 | case RawDexFile::kLong: |
| 235 | value->j = ReadSignedLong(ptr, value_arg); |
| 236 | break; |
| 237 | case RawDexFile::kFloat: |
| 238 | value->i = ReadUnsignedInt(ptr, value_arg, true); |
| 239 | break; |
| 240 | case RawDexFile::kDouble: |
| 241 | value->j = ReadUnsignedLong(ptr, value_arg, true); |
| 242 | break; |
| 243 | case RawDexFile::kBoolean: |
| 244 | value->i = (value_arg != 0); |
| 245 | width = 0; |
| 246 | break; |
| 247 | case RawDexFile::kString: |
| 248 | case RawDexFile::kType: |
| 249 | case RawDexFile::kMethod: |
| 250 | case RawDexFile::kEnum: |
| 251 | value->i = ReadUnsignedInt(ptr, value_arg, false); |
| 252 | break; |
| 253 | case RawDexFile::kField: |
| 254 | case RawDexFile::kArray: |
| 255 | case RawDexFile::kAnnotation: |
| 256 | LOG(FATAL) << "Unimplemented"; |
| 257 | break; |
| 258 | case RawDexFile::kNull: |
| 259 | value->i = 0; |
| 260 | width = 0; |
| 261 | break; |
| 262 | default: |
| 263 | LOG(FATAL) << "Unreached"; |
| 264 | } |
| 265 | ptr += width; |
| 266 | *stream = ptr; |
| 267 | return static_cast<ValueType>(type); |
Carl Shapiro | 1fb8620 | 2011-06-27 17:43:13 -0700 | [diff] [blame] | 268 | } |
| 269 | |
Carl Shapiro | 1fb8620 | 2011-06-27 17:43:13 -0700 | [diff] [blame] | 270 | } // namespace art |