1 /*
2  * Copyright (C) 2011 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #include "dex_file_verifier.h"
18 
19 #include <zlib.h>
20 #include <memory>
21 
22 #include "base/stringprintf.h"
23 #include "dex_file-inl.h"
24 #include "leb128.h"
25 #include "safe_map.h"
26 #include "utf-inl.h"
27 #include "utils.h"
28 
29 namespace art {
30 
MapTypeToBitMask(uint32_t map_type)31 static uint32_t MapTypeToBitMask(uint32_t map_type) {
32   switch (map_type) {
33     case DexFile::kDexTypeHeaderItem:               return 1 << 0;
34     case DexFile::kDexTypeStringIdItem:             return 1 << 1;
35     case DexFile::kDexTypeTypeIdItem:               return 1 << 2;
36     case DexFile::kDexTypeProtoIdItem:              return 1 << 3;
37     case DexFile::kDexTypeFieldIdItem:              return 1 << 4;
38     case DexFile::kDexTypeMethodIdItem:             return 1 << 5;
39     case DexFile::kDexTypeClassDefItem:             return 1 << 6;
40     case DexFile::kDexTypeMapList:                  return 1 << 7;
41     case DexFile::kDexTypeTypeList:                 return 1 << 8;
42     case DexFile::kDexTypeAnnotationSetRefList:     return 1 << 9;
43     case DexFile::kDexTypeAnnotationSetItem:        return 1 << 10;
44     case DexFile::kDexTypeClassDataItem:            return 1 << 11;
45     case DexFile::kDexTypeCodeItem:                 return 1 << 12;
46     case DexFile::kDexTypeStringDataItem:           return 1 << 13;
47     case DexFile::kDexTypeDebugInfoItem:            return 1 << 14;
48     case DexFile::kDexTypeAnnotationItem:           return 1 << 15;
49     case DexFile::kDexTypeEncodedArrayItem:         return 1 << 16;
50     case DexFile::kDexTypeAnnotationsDirectoryItem: return 1 << 17;
51   }
52   return 0;
53 }
54 
IsDataSectionType(uint32_t map_type)55 static bool IsDataSectionType(uint32_t map_type) {
56   switch (map_type) {
57     case DexFile::kDexTypeHeaderItem:
58     case DexFile::kDexTypeStringIdItem:
59     case DexFile::kDexTypeTypeIdItem:
60     case DexFile::kDexTypeProtoIdItem:
61     case DexFile::kDexTypeFieldIdItem:
62     case DexFile::kDexTypeMethodIdItem:
63     case DexFile::kDexTypeClassDefItem:
64       return false;
65   }
66   return true;
67 }
68 
CheckLoadStringByIdx(uint32_t idx,const char * error_string)69 const char* DexFileVerifier::CheckLoadStringByIdx(uint32_t idx, const char* error_string) {
70   if (UNLIKELY(!CheckIndex(idx, dex_file_->NumStringIds(), error_string))) {
71     return nullptr;
72   }
73   return dex_file_->StringDataByIdx(idx);
74 }
75 
CheckLoadStringByTypeIdx(uint32_t type_idx,const char * error_string)76 const char* DexFileVerifier::CheckLoadStringByTypeIdx(uint32_t type_idx, const char* error_string) {
77   if (UNLIKELY(!CheckIndex(type_idx, dex_file_->NumTypeIds(), error_string))) {
78     return nullptr;
79   }
80   const DexFile::TypeId& type_id = dex_file_->GetTypeId(type_idx);
81   uint32_t idx = type_id.descriptor_idx_;
82   return CheckLoadStringByIdx(idx, error_string);
83 }
84 
CheckLoadFieldId(uint32_t idx,const char * error_string)85 const DexFile::FieldId* DexFileVerifier::CheckLoadFieldId(uint32_t idx, const char* error_string) {
86   if (UNLIKELY(!CheckIndex(idx, dex_file_->NumFieldIds(), error_string))) {
87     return nullptr;
88   }
89   return &dex_file_->GetFieldId(idx);
90 }
91 
CheckLoadMethodId(uint32_t idx,const char * err_string)92 const DexFile::MethodId* DexFileVerifier::CheckLoadMethodId(uint32_t idx, const char* err_string) {
93   if (UNLIKELY(!CheckIndex(idx, dex_file_->NumMethodIds(), err_string))) {
94     return nullptr;
95   }
96   return &dex_file_->GetMethodId(idx);
97 }
98 
99 // Helper macro to load string and return false on error.
100 #define LOAD_STRING(var, idx, error)                  \
101   const char* var = CheckLoadStringByIdx(idx, error); \
102   if (UNLIKELY(var == nullptr)) {                     \
103     return false;                                     \
104   }
105 
106 // Helper macro to load string by type idx and return false on error.
107 #define LOAD_STRING_BY_TYPE(var, type_idx, error)              \
108   const char* var = CheckLoadStringByTypeIdx(type_idx, error); \
109   if (UNLIKELY(var == nullptr)) {                              \
110     return false;                                              \
111   }
112 
113 // Helper macro to load method id. Return last parameter on error.
114 #define LOAD_METHOD(var, idx, error_string, error_stmt)                 \
115   const DexFile::MethodId* var  = CheckLoadMethodId(idx, error_string); \
116   if (UNLIKELY(var == nullptr)) {                                       \
117     error_stmt;                                                         \
118   }
119 
120 // Helper macro to load method id. Return last parameter on error.
121 #define LOAD_FIELD(var, idx, fmt, error_stmt)               \
122   const DexFile::FieldId* var = CheckLoadFieldId(idx, fmt); \
123   if (UNLIKELY(var == nullptr)) {                           \
124     error_stmt;                                             \
125   }
126 
Verify(const DexFile * dex_file,const uint8_t * begin,size_t size,const char * location,std::string * error_msg)127 bool DexFileVerifier::Verify(const DexFile* dex_file, const uint8_t* begin, size_t size,
128                              const char* location, std::string* error_msg) {
129   std::unique_ptr<DexFileVerifier> verifier(new DexFileVerifier(dex_file, begin, size, location));
130   if (!verifier->Verify()) {
131     *error_msg = verifier->FailureReason();
132     return false;
133   }
134   return true;
135 }
136 
CheckShortyDescriptorMatch(char shorty_char,const char * descriptor,bool is_return_type)137 bool DexFileVerifier::CheckShortyDescriptorMatch(char shorty_char, const char* descriptor,
138                                                 bool is_return_type) {
139   switch (shorty_char) {
140     case 'V':
141       if (UNLIKELY(!is_return_type)) {
142         ErrorStringPrintf("Invalid use of void");
143         return false;
144       }
145       FALLTHROUGH_INTENDED;
146     case 'B':
147     case 'C':
148     case 'D':
149     case 'F':
150     case 'I':
151     case 'J':
152     case 'S':
153     case 'Z':
154       if (UNLIKELY((descriptor[0] != shorty_char) || (descriptor[1] != '\0'))) {
155         ErrorStringPrintf("Shorty vs. primitive type mismatch: '%c', '%s'",
156                           shorty_char, descriptor);
157         return false;
158       }
159       break;
160     case 'L':
161       if (UNLIKELY((descriptor[0] != 'L') && (descriptor[0] != '['))) {
162         ErrorStringPrintf("Shorty vs. type mismatch: '%c', '%s'", shorty_char, descriptor);
163         return false;
164       }
165       break;
166     default:
167       ErrorStringPrintf("Bad shorty character: '%c'", shorty_char);
168       return false;
169   }
170   return true;
171 }
172 
CheckListSize(const void * start,size_t count,size_t elem_size,const char * label)173 bool DexFileVerifier::CheckListSize(const void* start, size_t count, size_t elem_size,
174                                     const char* label) {
175   // Check that size is not 0.
176   CHECK_NE(elem_size, 0U);
177 
178   const uint8_t* range_start = reinterpret_cast<const uint8_t*>(start);
179   const uint8_t* file_start = reinterpret_cast<const uint8_t*>(begin_);
180 
181   // Check for overflow.
182   uintptr_t max = 0 - 1;
183   size_t available_bytes_till_end_of_mem = max - reinterpret_cast<uintptr_t>(start);
184   size_t max_count = available_bytes_till_end_of_mem / elem_size;
185   if (max_count < count) {
186     ErrorStringPrintf("Overflow in range for %s: %zx for %zu@%zu", label,
187                       static_cast<size_t>(range_start - file_start),
188                       count, elem_size);
189     return false;
190   }
191 
192   const uint8_t* range_end = range_start + count * elem_size;
193   const uint8_t* file_end = file_start + size_;
194   if (UNLIKELY((range_start < file_start) || (range_end > file_end))) {
195     // Note: these two tests are enough as we make sure above that there's no overflow.
196     ErrorStringPrintf("Bad range for %s: %zx to %zx", label,
197                       static_cast<size_t>(range_start - file_start),
198                       static_cast<size_t>(range_end - file_start));
199     return false;
200   }
201   return true;
202 }
203 
CheckList(size_t element_size,const char * label,const uint8_t ** ptr)204 bool DexFileVerifier::CheckList(size_t element_size, const char* label, const uint8_t* *ptr) {
205   // Check that the list is available. The first 4B are the count.
206   if (!CheckListSize(*ptr, 1, 4U, label)) {
207     return false;
208   }
209 
210   uint32_t count = *reinterpret_cast<const uint32_t*>(*ptr);
211   if (count > 0) {
212     if (!CheckListSize(*ptr + 4, count, element_size, label)) {
213       return false;
214     }
215   }
216 
217   *ptr += 4 + count * element_size;
218   return true;
219 }
220 
CheckIndex(uint32_t field,uint32_t limit,const char * label)221 bool DexFileVerifier::CheckIndex(uint32_t field, uint32_t limit, const char* label) {
222   if (UNLIKELY(field >= limit)) {
223     ErrorStringPrintf("Bad index for %s: %x >= %x", label, field, limit);
224     return false;
225   }
226   return true;
227 }
228 
CheckValidOffsetAndSize(uint32_t offset,uint32_t size,const char * label)229 bool DexFileVerifier::CheckValidOffsetAndSize(uint32_t offset, uint32_t size, const char* label) {
230   if (size == 0) {
231     if (offset != 0) {
232       ErrorStringPrintf("Offset(%d) should be zero when size is zero for %s.", offset, label);
233       return false;
234     }
235   }
236   if (size_ <= offset) {
237     ErrorStringPrintf("Offset(%d) should be within file size(%zu) for %s.", offset, size_, label);
238     return false;
239   }
240   return true;
241 }
242 
CheckHeader()243 bool DexFileVerifier::CheckHeader() {
244   // Check file size from the header.
245   uint32_t expected_size = header_->file_size_;
246   if (size_ != expected_size) {
247     ErrorStringPrintf("Bad file size (%zd, expected %ud)", size_, expected_size);
248     return false;
249   }
250 
251   // Compute and verify the checksum in the header.
252   uint32_t adler_checksum = adler32(0L, Z_NULL, 0);
253   const uint32_t non_sum = sizeof(header_->magic_) + sizeof(header_->checksum_);
254   const uint8_t* non_sum_ptr = reinterpret_cast<const uint8_t*>(header_) + non_sum;
255   adler_checksum = adler32(adler_checksum, non_sum_ptr, expected_size - non_sum);
256   if (adler_checksum != header_->checksum_) {
257     ErrorStringPrintf("Bad checksum (%08x, expected %08x)", adler_checksum, header_->checksum_);
258     return false;
259   }
260 
261   // Check the contents of the header.
262   if (header_->endian_tag_ != DexFile::kDexEndianConstant) {
263     ErrorStringPrintf("Unexpected endian_tag: %x", header_->endian_tag_);
264     return false;
265   }
266 
267   if (header_->header_size_ != sizeof(DexFile::Header)) {
268     ErrorStringPrintf("Bad header size: %ud", header_->header_size_);
269     return false;
270   }
271 
272   // Check that all offsets are inside the file.
273   bool result =
274       CheckValidOffsetAndSize(header_->link_off_, header_->link_size_, "link") &&
275       CheckValidOffsetAndSize(header_->map_off_, header_->map_off_, "map") &&
276       CheckValidOffsetAndSize(header_->string_ids_off_, header_->string_ids_size_, "string-ids") &&
277       CheckValidOffsetAndSize(header_->type_ids_off_, header_->type_ids_size_, "type-ids") &&
278       CheckValidOffsetAndSize(header_->proto_ids_off_, header_->proto_ids_size_, "proto-ids") &&
279       CheckValidOffsetAndSize(header_->field_ids_off_, header_->field_ids_size_, "field-ids") &&
280       CheckValidOffsetAndSize(header_->method_ids_off_, header_->method_ids_size_, "method-ids") &&
281       CheckValidOffsetAndSize(header_->class_defs_off_, header_->class_defs_size_, "class-defs") &&
282       CheckValidOffsetAndSize(header_->data_off_, header_->data_size_, "data");
283 
284   return result;
285 }
286 
CheckMap()287 bool DexFileVerifier::CheckMap() {
288   const DexFile::MapList* map = reinterpret_cast<const DexFile::MapList*>(begin_ +
289                                                                           header_->map_off_);
290   // Check that map list content is available.
291   if (!CheckListSize(map, 1, sizeof(DexFile::MapList), "maplist content")) {
292     return false;
293   }
294 
295   const DexFile::MapItem* item = map->list_;
296 
297   uint32_t count = map->size_;
298   uint32_t last_offset = 0;
299   uint32_t data_item_count = 0;
300   uint32_t data_items_left = header_->data_size_;
301   uint32_t used_bits = 0;
302 
303   // Sanity check the size of the map list.
304   if (!CheckListSize(item, count, sizeof(DexFile::MapItem), "map size")) {
305     return false;
306   }
307 
308   // Check the items listed in the map.
309   for (uint32_t i = 0; i < count; i++) {
310     if (UNLIKELY(last_offset >= item->offset_ && i != 0)) {
311       ErrorStringPrintf("Out of order map item: %x then %x", last_offset, item->offset_);
312       return false;
313     }
314     if (UNLIKELY(item->offset_ >= header_->file_size_)) {
315       ErrorStringPrintf("Map item after end of file: %x, size %x",
316                         item->offset_, header_->file_size_);
317       return false;
318     }
319 
320     if (IsDataSectionType(item->type_)) {
321       uint32_t icount = item->size_;
322       if (UNLIKELY(icount > data_items_left)) {
323         ErrorStringPrintf("Too many items in data section: %ud", data_item_count + icount);
324         return false;
325       }
326       data_items_left -= icount;
327       data_item_count += icount;
328     }
329 
330     uint32_t bit = MapTypeToBitMask(item->type_);
331 
332     if (UNLIKELY(bit == 0)) {
333       ErrorStringPrintf("Unknown map section type %x", item->type_);
334       return false;
335     }
336 
337     if (UNLIKELY((used_bits & bit) != 0)) {
338       ErrorStringPrintf("Duplicate map section of type %x", item->type_);
339       return false;
340     }
341 
342     used_bits |= bit;
343     last_offset = item->offset_;
344     item++;
345   }
346 
347   // Check for missing sections in the map.
348   if (UNLIKELY((used_bits & MapTypeToBitMask(DexFile::kDexTypeHeaderItem)) == 0)) {
349     ErrorStringPrintf("Map is missing header entry");
350     return false;
351   }
352   if (UNLIKELY((used_bits & MapTypeToBitMask(DexFile::kDexTypeMapList)) == 0)) {
353     ErrorStringPrintf("Map is missing map_list entry");
354     return false;
355   }
356   if (UNLIKELY((used_bits & MapTypeToBitMask(DexFile::kDexTypeStringIdItem)) == 0 &&
357                ((header_->string_ids_off_ != 0) || (header_->string_ids_size_ != 0)))) {
358     ErrorStringPrintf("Map is missing string_ids entry");
359     return false;
360   }
361   if (UNLIKELY((used_bits & MapTypeToBitMask(DexFile::kDexTypeTypeIdItem)) == 0 &&
362                ((header_->type_ids_off_ != 0) || (header_->type_ids_size_ != 0)))) {
363     ErrorStringPrintf("Map is missing type_ids entry");
364     return false;
365   }
366   if (UNLIKELY((used_bits & MapTypeToBitMask(DexFile::kDexTypeProtoIdItem)) == 0 &&
367                ((header_->proto_ids_off_ != 0) || (header_->proto_ids_size_ != 0)))) {
368     ErrorStringPrintf("Map is missing proto_ids entry");
369     return false;
370   }
371   if (UNLIKELY((used_bits & MapTypeToBitMask(DexFile::kDexTypeFieldIdItem)) == 0 &&
372                ((header_->field_ids_off_ != 0) || (header_->field_ids_size_ != 0)))) {
373     ErrorStringPrintf("Map is missing field_ids entry");
374     return false;
375   }
376   if (UNLIKELY((used_bits & MapTypeToBitMask(DexFile::kDexTypeMethodIdItem)) == 0 &&
377                ((header_->method_ids_off_ != 0) || (header_->method_ids_size_ != 0)))) {
378     ErrorStringPrintf("Map is missing method_ids entry");
379     return false;
380   }
381   if (UNLIKELY((used_bits & MapTypeToBitMask(DexFile::kDexTypeClassDefItem)) == 0 &&
382                ((header_->class_defs_off_ != 0) || (header_->class_defs_size_ != 0)))) {
383     ErrorStringPrintf("Map is missing class_defs entry");
384     return false;
385   }
386   return true;
387 }
388 
ReadUnsignedLittleEndian(uint32_t size)389 uint32_t DexFileVerifier::ReadUnsignedLittleEndian(uint32_t size) {
390   uint32_t result = 0;
391   if (LIKELY(CheckListSize(ptr_, size, sizeof(uint8_t), "encoded_value"))) {
392     for (uint32_t i = 0; i < size; i++) {
393       result |= ((uint32_t) *(ptr_++)) << (i * 8);
394     }
395   }
396   return result;
397 }
398 
CheckAndGetHandlerOffsets(const DexFile::CodeItem * code_item,uint32_t * handler_offsets,uint32_t handlers_size)399 bool DexFileVerifier::CheckAndGetHandlerOffsets(const DexFile::CodeItem* code_item,
400                                                 uint32_t* handler_offsets, uint32_t handlers_size) {
401   const uint8_t* handlers_base = DexFile::GetCatchHandlerData(*code_item, 0);
402 
403   for (uint32_t i = 0; i < handlers_size; i++) {
404     bool catch_all;
405     size_t offset = ptr_ - handlers_base;
406     int32_t size = DecodeSignedLeb128(&ptr_);
407 
408     if (UNLIKELY((size < -65536) || (size > 65536))) {
409       ErrorStringPrintf("Invalid exception handler size: %d", size);
410       return false;
411     }
412 
413     if (size <= 0) {
414       catch_all = true;
415       size = -size;
416     } else {
417       catch_all = false;
418     }
419 
420     handler_offsets[i] = static_cast<uint32_t>(offset);
421 
422     while (size-- > 0) {
423       uint32_t type_idx = DecodeUnsignedLeb128(&ptr_);
424       if (!CheckIndex(type_idx, header_->type_ids_size_, "handler type_idx")) {
425         return false;
426       }
427 
428       uint32_t addr = DecodeUnsignedLeb128(&ptr_);
429       if (UNLIKELY(addr >= code_item->insns_size_in_code_units_)) {
430         ErrorStringPrintf("Invalid handler addr: %x", addr);
431         return false;
432       }
433     }
434 
435     if (catch_all) {
436       uint32_t addr = DecodeUnsignedLeb128(&ptr_);
437       if (UNLIKELY(addr >= code_item->insns_size_in_code_units_)) {
438         ErrorStringPrintf("Invalid handler catch_all_addr: %x", addr);
439         return false;
440       }
441     }
442   }
443 
444   return true;
445 }
446 
CheckClassDataItemField(uint32_t idx,uint32_t access_flags,bool expect_static)447 bool DexFileVerifier::CheckClassDataItemField(uint32_t idx, uint32_t access_flags,
448                                               bool expect_static) {
449   if (!CheckIndex(idx, header_->field_ids_size_, "class_data_item field_idx")) {
450     return false;
451   }
452 
453   bool is_static = (access_flags & kAccStatic) != 0;
454   if (UNLIKELY(is_static != expect_static)) {
455     ErrorStringPrintf("Static/instance field not in expected list");
456     return false;
457   }
458 
459   if (UNLIKELY((access_flags & ~kAccJavaFlagsMask) != 0)) {
460     ErrorStringPrintf("Bad class_data_item field access_flags %x", access_flags);
461     return false;
462   }
463 
464   return true;
465 }
466 
CheckClassDataItemMethod(uint32_t idx,uint32_t access_flags,uint32_t code_offset,bool expect_direct)467 bool DexFileVerifier::CheckClassDataItemMethod(uint32_t idx, uint32_t access_flags,
468                                                uint32_t code_offset, bool expect_direct) {
469   if (!CheckIndex(idx, header_->method_ids_size_, "class_data_item method_idx")) {
470     return false;
471   }
472 
473   bool is_direct = (access_flags & (kAccStatic | kAccPrivate | kAccConstructor)) != 0;
474   bool expect_code = (access_flags & (kAccNative | kAccAbstract)) == 0;
475   bool is_synchronized = (access_flags & kAccSynchronized) != 0;
476   bool allow_synchronized = (access_flags & kAccNative) != 0;
477 
478   if (UNLIKELY(is_direct != expect_direct)) {
479     ErrorStringPrintf("Direct/virtual method not in expected list");
480     return false;
481   }
482 
483   constexpr uint32_t access_method_mask = kAccJavaFlagsMask | kAccConstructor |
484       kAccDeclaredSynchronized;
485   if (UNLIKELY(((access_flags & ~access_method_mask) != 0) ||
486                (is_synchronized && !allow_synchronized))) {
487     ErrorStringPrintf("Bad class_data_item method access_flags %x", access_flags);
488     return false;
489   }
490 
491   if (UNLIKELY(expect_code && (code_offset == 0))) {
492     ErrorStringPrintf("Unexpected zero value for class_data_item method code_off with access "
493                       "flags %x", access_flags);
494     return false;
495   } else if (UNLIKELY(!expect_code && (code_offset != 0))) {
496     ErrorStringPrintf("Unexpected non-zero value %x for class_data_item method code_off"
497                       " with access flags %x", code_offset, access_flags);
498     return false;
499   }
500 
501   return true;
502 }
503 
CheckPadding(size_t offset,uint32_t aligned_offset)504 bool DexFileVerifier::CheckPadding(size_t offset, uint32_t aligned_offset) {
505   if (offset < aligned_offset) {
506     if (!CheckListSize(begin_ + offset, aligned_offset - offset, sizeof(uint8_t), "section")) {
507       return false;
508     }
509     while (offset < aligned_offset) {
510       if (UNLIKELY(*ptr_ != '\0')) {
511         ErrorStringPrintf("Non-zero padding %x before section start at %zx", *ptr_, offset);
512         return false;
513       }
514       ptr_++;
515       offset++;
516     }
517   }
518   return true;
519 }
520 
CheckEncodedValue()521 bool DexFileVerifier::CheckEncodedValue() {
522   if (!CheckListSize(ptr_, 1, sizeof(uint8_t), "encoded_value header")) {
523     return false;
524   }
525 
526   uint8_t header_byte = *(ptr_++);
527   uint32_t value_type = header_byte & DexFile::kDexAnnotationValueTypeMask;
528   uint32_t value_arg = header_byte >> DexFile::kDexAnnotationValueArgShift;
529 
530   switch (value_type) {
531     case DexFile::kDexAnnotationByte:
532       if (UNLIKELY(value_arg != 0)) {
533         ErrorStringPrintf("Bad encoded_value byte size %x", value_arg);
534         return false;
535       }
536       ptr_++;
537       break;
538     case DexFile::kDexAnnotationShort:
539     case DexFile::kDexAnnotationChar:
540       if (UNLIKELY(value_arg > 1)) {
541         ErrorStringPrintf("Bad encoded_value char/short size %x", value_arg);
542         return false;
543       }
544       ptr_ += value_arg + 1;
545       break;
546     case DexFile::kDexAnnotationInt:
547     case DexFile::kDexAnnotationFloat:
548       if (UNLIKELY(value_arg > 3)) {
549         ErrorStringPrintf("Bad encoded_value int/float size %x", value_arg);
550         return false;
551       }
552       ptr_ += value_arg + 1;
553       break;
554     case DexFile::kDexAnnotationLong:
555     case DexFile::kDexAnnotationDouble:
556       ptr_ += value_arg + 1;
557       break;
558     case DexFile::kDexAnnotationString: {
559       if (UNLIKELY(value_arg > 3)) {
560         ErrorStringPrintf("Bad encoded_value string size %x", value_arg);
561         return false;
562       }
563       uint32_t idx = ReadUnsignedLittleEndian(value_arg + 1);
564       if (!CheckIndex(idx, header_->string_ids_size_, "encoded_value string")) {
565         return false;
566       }
567       break;
568     }
569     case DexFile::kDexAnnotationType: {
570       if (UNLIKELY(value_arg > 3)) {
571         ErrorStringPrintf("Bad encoded_value type size %x", value_arg);
572         return false;
573       }
574       uint32_t idx = ReadUnsignedLittleEndian(value_arg + 1);
575       if (!CheckIndex(idx, header_->type_ids_size_, "encoded_value type")) {
576         return false;
577       }
578       break;
579     }
580     case DexFile::kDexAnnotationField:
581     case DexFile::kDexAnnotationEnum: {
582       if (UNLIKELY(value_arg > 3)) {
583         ErrorStringPrintf("Bad encoded_value field/enum size %x", value_arg);
584         return false;
585       }
586       uint32_t idx = ReadUnsignedLittleEndian(value_arg + 1);
587       if (!CheckIndex(idx, header_->field_ids_size_, "encoded_value field")) {
588         return false;
589       }
590       break;
591     }
592     case DexFile::kDexAnnotationMethod: {
593       if (UNLIKELY(value_arg > 3)) {
594         ErrorStringPrintf("Bad encoded_value method size %x", value_arg);
595         return false;
596       }
597       uint32_t idx = ReadUnsignedLittleEndian(value_arg + 1);
598       if (!CheckIndex(idx, header_->method_ids_size_, "encoded_value method")) {
599         return false;
600       }
601       break;
602     }
603     case DexFile::kDexAnnotationArray:
604       if (UNLIKELY(value_arg != 0)) {
605         ErrorStringPrintf("Bad encoded_value array value_arg %x", value_arg);
606         return false;
607       }
608       if (!CheckEncodedArray()) {
609         return false;
610       }
611       break;
612     case DexFile::kDexAnnotationAnnotation:
613       if (UNLIKELY(value_arg != 0)) {
614         ErrorStringPrintf("Bad encoded_value annotation value_arg %x", value_arg);
615         return false;
616       }
617       if (!CheckEncodedAnnotation()) {
618         return false;
619       }
620       break;
621     case DexFile::kDexAnnotationNull:
622       if (UNLIKELY(value_arg != 0)) {
623         ErrorStringPrintf("Bad encoded_value null value_arg %x", value_arg);
624         return false;
625       }
626       break;
627     case DexFile::kDexAnnotationBoolean:
628       if (UNLIKELY(value_arg > 1)) {
629         ErrorStringPrintf("Bad encoded_value boolean size %x", value_arg);
630         return false;
631       }
632       break;
633     default:
634       ErrorStringPrintf("Bogus encoded_value value_type %x", value_type);
635       return false;
636   }
637 
638   return true;
639 }
640 
CheckEncodedArray()641 bool DexFileVerifier::CheckEncodedArray() {
642   uint32_t size = DecodeUnsignedLeb128(&ptr_);
643 
644   while (size--) {
645     if (!CheckEncodedValue()) {
646       failure_reason_ = StringPrintf("Bad encoded_array value: %s", failure_reason_.c_str());
647       return false;
648     }
649   }
650   return true;
651 }
652 
CheckEncodedAnnotation()653 bool DexFileVerifier::CheckEncodedAnnotation() {
654   uint32_t idx = DecodeUnsignedLeb128(&ptr_);
655   if (!CheckIndex(idx, header_->type_ids_size_, "encoded_annotation type_idx")) {
656     return false;
657   }
658 
659   uint32_t size = DecodeUnsignedLeb128(&ptr_);
660   uint32_t last_idx = 0;
661 
662   for (uint32_t i = 0; i < size; i++) {
663     idx = DecodeUnsignedLeb128(&ptr_);
664     if (!CheckIndex(idx, header_->string_ids_size_, "annotation_element name_idx")) {
665       return false;
666     }
667 
668     if (UNLIKELY(last_idx >= idx && i != 0)) {
669       ErrorStringPrintf("Out-of-order annotation_element name_idx: %x then %x",
670                         last_idx, idx);
671       return false;
672     }
673 
674     if (!CheckEncodedValue()) {
675       return false;
676     }
677 
678     last_idx = idx;
679   }
680   return true;
681 }
682 
CheckIntraClassDataItem()683 bool DexFileVerifier::CheckIntraClassDataItem() {
684   ClassDataItemIterator it(*dex_file_, ptr_);
685 
686   // These calls use the raw access flags to check whether the whole dex field is valid.
687   uint32_t prev_index = 0;
688   for (; it.HasNextStaticField(); it.Next()) {
689     uint32_t curr_index = it.GetMemberIndex();
690     if (curr_index < prev_index) {
691       ErrorStringPrintf("out-of-order static field indexes %d and %d", prev_index, curr_index);
692       return false;
693     }
694     prev_index = curr_index;
695     if (!CheckClassDataItemField(curr_index, it.GetRawMemberAccessFlags(), true)) {
696       return false;
697     }
698   }
699   prev_index = 0;
700   for (; it.HasNextInstanceField(); it.Next()) {
701     uint32_t curr_index = it.GetMemberIndex();
702     if (curr_index < prev_index) {
703       ErrorStringPrintf("out-of-order instance field indexes %d and %d", prev_index, curr_index);
704       return false;
705     }
706     prev_index = curr_index;
707     if (!CheckClassDataItemField(curr_index, it.GetRawMemberAccessFlags(), false)) {
708       return false;
709     }
710   }
711   prev_index = 0;
712   for (; it.HasNextDirectMethod(); it.Next()) {
713     uint32_t curr_index = it.GetMemberIndex();
714     if (curr_index < prev_index) {
715       ErrorStringPrintf("out-of-order direct method indexes %d and %d", prev_index, curr_index);
716       return false;
717     }
718     prev_index = curr_index;
719     if (!CheckClassDataItemMethod(curr_index, it.GetRawMemberAccessFlags(),
720         it.GetMethodCodeItemOffset(), true)) {
721       return false;
722     }
723   }
724   prev_index = 0;
725   for (; it.HasNextVirtualMethod(); it.Next()) {
726     uint32_t curr_index = it.GetMemberIndex();
727     if (curr_index < prev_index) {
728       ErrorStringPrintf("out-of-order virtual method indexes %d and %d", prev_index, curr_index);
729       return false;
730     }
731     prev_index = curr_index;
732     if (!CheckClassDataItemMethod(curr_index, it.GetRawMemberAccessFlags(),
733         it.GetMethodCodeItemOffset(), false)) {
734       return false;
735     }
736   }
737 
738   ptr_ = it.EndDataPointer();
739   return true;
740 }
741 
CheckIntraCodeItem()742 bool DexFileVerifier::CheckIntraCodeItem() {
743   const DexFile::CodeItem* code_item = reinterpret_cast<const DexFile::CodeItem*>(ptr_);
744   if (!CheckListSize(code_item, 1, sizeof(DexFile::CodeItem), "code")) {
745     return false;
746   }
747 
748   if (UNLIKELY(code_item->ins_size_ > code_item->registers_size_)) {
749     ErrorStringPrintf("ins_size (%ud) > registers_size (%ud)",
750                       code_item->ins_size_, code_item->registers_size_);
751     return false;
752   }
753 
754   if (UNLIKELY((code_item->outs_size_ > 5) &&
755                (code_item->outs_size_ > code_item->registers_size_))) {
756     /*
757      * outs_size can be up to 5, even if registers_size is smaller, since the
758      * short forms of method invocation allow repetitions of a register multiple
759      * times within a single parameter list. However, longer parameter lists
760      * need to be represented in-order in the register file.
761      */
762     ErrorStringPrintf("outs_size (%ud) > registers_size (%ud)",
763                       code_item->outs_size_, code_item->registers_size_);
764     return false;
765   }
766 
767   const uint16_t* insns = code_item->insns_;
768   uint32_t insns_size = code_item->insns_size_in_code_units_;
769   if (!CheckListSize(insns, insns_size, sizeof(uint16_t), "insns size")) {
770     return false;
771   }
772 
773   // Grab the end of the insns if there are no try_items.
774   uint32_t try_items_size = code_item->tries_size_;
775   if (try_items_size == 0) {
776     ptr_ = reinterpret_cast<const uint8_t*>(&insns[insns_size]);
777     return true;
778   }
779 
780   // try_items are 4-byte aligned. Verify the spacer is 0.
781   if (((reinterpret_cast<uintptr_t>(&insns[insns_size]) & 3) != 0) && (insns[insns_size] != 0)) {
782     ErrorStringPrintf("Non-zero padding: %x", insns[insns_size]);
783     return false;
784   }
785 
786   const DexFile::TryItem* try_items = DexFile::GetTryItems(*code_item, 0);
787   ptr_ = DexFile::GetCatchHandlerData(*code_item, 0);
788   uint32_t handlers_size = DecodeUnsignedLeb128(&ptr_);
789 
790   if (!CheckListSize(try_items, try_items_size, sizeof(DexFile::TryItem), "try_items size")) {
791     return false;
792   }
793 
794   if (UNLIKELY((handlers_size == 0) || (handlers_size >= 65536))) {
795     ErrorStringPrintf("Invalid handlers_size: %ud", handlers_size);
796     return false;
797   }
798 
799   std::unique_ptr<uint32_t[]> handler_offsets(new uint32_t[handlers_size]);
800   if (!CheckAndGetHandlerOffsets(code_item, &handler_offsets[0], handlers_size)) {
801     return false;
802   }
803 
804   uint32_t last_addr = 0;
805   while (try_items_size--) {
806     if (UNLIKELY(try_items->start_addr_ < last_addr)) {
807       ErrorStringPrintf("Out-of_order try_item with start_addr: %x", try_items->start_addr_);
808       return false;
809     }
810 
811     if (UNLIKELY(try_items->start_addr_ >= insns_size)) {
812       ErrorStringPrintf("Invalid try_item start_addr: %x", try_items->start_addr_);
813       return false;
814     }
815 
816     uint32_t i;
817     for (i = 0; i < handlers_size; i++) {
818       if (try_items->handler_off_ == handler_offsets[i]) {
819         break;
820       }
821     }
822 
823     if (UNLIKELY(i == handlers_size)) {
824       ErrorStringPrintf("Bogus handler offset: %x", try_items->handler_off_);
825       return false;
826     }
827 
828     last_addr = try_items->start_addr_ + try_items->insn_count_;
829     if (UNLIKELY(last_addr > insns_size)) {
830       ErrorStringPrintf("Invalid try_item insn_count: %x", try_items->insn_count_);
831       return false;
832     }
833 
834     try_items++;
835   }
836 
837   return true;
838 }
839 
CheckIntraStringDataItem()840 bool DexFileVerifier::CheckIntraStringDataItem() {
841   uint32_t size = DecodeUnsignedLeb128(&ptr_);
842   const uint8_t* file_end = begin_ + size_;
843 
844   for (uint32_t i = 0; i < size; i++) {
845     CHECK_LT(i, size);  // b/15014252 Prevents hitting the impossible case below
846     if (UNLIKELY(ptr_ >= file_end)) {
847       ErrorStringPrintf("String data would go beyond end-of-file");
848       return false;
849     }
850 
851     uint8_t byte = *(ptr_++);
852 
853     // Switch on the high 4 bits.
854     switch (byte >> 4) {
855       case 0x00:
856         // Special case of bit pattern 0xxx.
857         if (UNLIKELY(byte == 0)) {
858           CHECK_LT(i, size);  // b/15014252 Actually hit this impossible case with clang
859           ErrorStringPrintf("String data shorter than indicated utf16_size %x", size);
860           return false;
861         }
862         break;
863       case 0x01:
864       case 0x02:
865       case 0x03:
866       case 0x04:
867       case 0x05:
868       case 0x06:
869       case 0x07:
870         // No extra checks necessary for bit pattern 0xxx.
871         break;
872       case 0x08:
873       case 0x09:
874       case 0x0a:
875       case 0x0b:
876       case 0x0f:
877         // Illegal bit patterns 10xx or 1111.
878         // Note: 1111 is valid for normal UTF-8, but not here.
879         ErrorStringPrintf("Illegal start byte %x in string data", byte);
880         return false;
881       case 0x0c:
882       case 0x0d: {
883         // Bit pattern 110x has an additional byte.
884         uint8_t byte2 = *(ptr_++);
885         if (UNLIKELY((byte2 & 0xc0) != 0x80)) {
886           ErrorStringPrintf("Illegal continuation byte %x in string data", byte2);
887           return false;
888         }
889         uint16_t value = ((byte & 0x1f) << 6) | (byte2 & 0x3f);
890         if (UNLIKELY((value != 0) && (value < 0x80))) {
891           ErrorStringPrintf("Illegal representation for value %x in string data", value);
892           return false;
893         }
894         break;
895       }
896       case 0x0e: {
897         // Bit pattern 1110 has 2 additional bytes.
898         uint8_t byte2 = *(ptr_++);
899         if (UNLIKELY((byte2 & 0xc0) != 0x80)) {
900           ErrorStringPrintf("Illegal continuation byte %x in string data", byte2);
901           return false;
902         }
903         uint8_t byte3 = *(ptr_++);
904         if (UNLIKELY((byte3 & 0xc0) != 0x80)) {
905           ErrorStringPrintf("Illegal continuation byte %x in string data", byte3);
906           return false;
907         }
908         uint16_t value = ((byte & 0x0f) << 12) | ((byte2 & 0x3f) << 6) | (byte3 & 0x3f);
909         if (UNLIKELY(value < 0x800)) {
910           ErrorStringPrintf("Illegal representation for value %x in string data", value);
911           return false;
912         }
913         break;
914       }
915     }
916   }
917 
918   if (UNLIKELY(*(ptr_++) != '\0')) {
919     ErrorStringPrintf("String longer than indicated size %x", size);
920     return false;
921   }
922 
923   return true;
924 }
925 
CheckIntraDebugInfoItem()926 bool DexFileVerifier::CheckIntraDebugInfoItem() {
927   DecodeUnsignedLeb128(&ptr_);
928   uint32_t parameters_size = DecodeUnsignedLeb128(&ptr_);
929   if (UNLIKELY(parameters_size > 65536)) {
930     ErrorStringPrintf("Invalid parameters_size: %x", parameters_size);
931     return false;
932   }
933 
934   for (uint32_t j = 0; j < parameters_size; j++) {
935     uint32_t parameter_name = DecodeUnsignedLeb128(&ptr_);
936     if (parameter_name != 0) {
937       parameter_name--;
938       if (!CheckIndex(parameter_name, header_->string_ids_size_, "debug_info_item parameter_name")) {
939         return false;
940       }
941     }
942   }
943 
944   while (true) {
945     uint8_t opcode = *(ptr_++);
946     switch (opcode) {
947       case DexFile::DBG_END_SEQUENCE: {
948         return true;
949       }
950       case DexFile::DBG_ADVANCE_PC: {
951         DecodeUnsignedLeb128(&ptr_);
952         break;
953       }
954       case DexFile::DBG_ADVANCE_LINE: {
955         DecodeSignedLeb128(&ptr_);
956         break;
957       }
958       case DexFile::DBG_START_LOCAL: {
959         uint32_t reg_num = DecodeUnsignedLeb128(&ptr_);
960         if (UNLIKELY(reg_num >= 65536)) {
961           ErrorStringPrintf("Bad reg_num for opcode %x", opcode);
962           return false;
963         }
964         uint32_t name_idx = DecodeUnsignedLeb128(&ptr_);
965         if (name_idx != 0) {
966           name_idx--;
967           if (!CheckIndex(name_idx, header_->string_ids_size_, "DBG_START_LOCAL name_idx")) {
968             return false;
969           }
970         }
971         uint32_t type_idx = DecodeUnsignedLeb128(&ptr_);
972         if (type_idx != 0) {
973           type_idx--;
974           if (!CheckIndex(type_idx, header_->type_ids_size_, "DBG_START_LOCAL type_idx")) {
975             return false;
976           }
977         }
978         break;
979       }
980       case DexFile::DBG_END_LOCAL:
981       case DexFile::DBG_RESTART_LOCAL: {
982         uint32_t reg_num = DecodeUnsignedLeb128(&ptr_);
983         if (UNLIKELY(reg_num >= 65536)) {
984           ErrorStringPrintf("Bad reg_num for opcode %x", opcode);
985           return false;
986         }
987         break;
988       }
989       case DexFile::DBG_START_LOCAL_EXTENDED: {
990         uint32_t reg_num = DecodeUnsignedLeb128(&ptr_);
991         if (UNLIKELY(reg_num >= 65536)) {
992           ErrorStringPrintf("Bad reg_num for opcode %x", opcode);
993           return false;
994         }
995         uint32_t name_idx = DecodeUnsignedLeb128(&ptr_);
996         if (name_idx != 0) {
997           name_idx--;
998           if (!CheckIndex(name_idx, header_->string_ids_size_, "DBG_START_LOCAL_EXTENDED name_idx")) {
999             return false;
1000           }
1001         }
1002         uint32_t type_idx = DecodeUnsignedLeb128(&ptr_);
1003         if (type_idx != 0) {
1004           type_idx--;
1005           if (!CheckIndex(type_idx, header_->type_ids_size_, "DBG_START_LOCAL_EXTENDED type_idx")) {
1006             return false;
1007           }
1008         }
1009         uint32_t sig_idx = DecodeUnsignedLeb128(&ptr_);
1010         if (sig_idx != 0) {
1011           sig_idx--;
1012           if (!CheckIndex(sig_idx, header_->string_ids_size_, "DBG_START_LOCAL_EXTENDED sig_idx")) {
1013             return false;
1014           }
1015         }
1016         break;
1017       }
1018       case DexFile::DBG_SET_FILE: {
1019         uint32_t name_idx = DecodeUnsignedLeb128(&ptr_);
1020         if (name_idx != 0) {
1021           name_idx--;
1022           if (!CheckIndex(name_idx, header_->string_ids_size_, "DBG_SET_FILE name_idx")) {
1023             return false;
1024           }
1025         }
1026         break;
1027       }
1028     }
1029   }
1030 }
1031 
CheckIntraAnnotationItem()1032 bool DexFileVerifier::CheckIntraAnnotationItem() {
1033   if (!CheckListSize(ptr_, 1, sizeof(uint8_t), "annotation visibility")) {
1034     return false;
1035   }
1036 
1037   // Check visibility
1038   switch (*(ptr_++)) {
1039     case DexFile::kDexVisibilityBuild:
1040     case DexFile::kDexVisibilityRuntime:
1041     case DexFile::kDexVisibilitySystem:
1042       break;
1043     default:
1044       ErrorStringPrintf("Bad annotation visibility: %x", *ptr_);
1045       return false;
1046   }
1047 
1048   if (!CheckEncodedAnnotation()) {
1049     return false;
1050   }
1051 
1052   return true;
1053 }
1054 
CheckIntraAnnotationsDirectoryItem()1055 bool DexFileVerifier::CheckIntraAnnotationsDirectoryItem() {
1056   const DexFile::AnnotationsDirectoryItem* item =
1057       reinterpret_cast<const DexFile::AnnotationsDirectoryItem*>(ptr_);
1058   if (!CheckListSize(item, 1, sizeof(DexFile::AnnotationsDirectoryItem), "annotations_directory")) {
1059     return false;
1060   }
1061 
1062   // Field annotations follow immediately after the annotations directory.
1063   const DexFile::FieldAnnotationsItem* field_item =
1064       reinterpret_cast<const DexFile::FieldAnnotationsItem*>(item + 1);
1065   uint32_t field_count = item->fields_size_;
1066   if (!CheckListSize(field_item, field_count, sizeof(DexFile::FieldAnnotationsItem), "field_annotations list")) {
1067     return false;
1068   }
1069 
1070   uint32_t last_idx = 0;
1071   for (uint32_t i = 0; i < field_count; i++) {
1072     if (UNLIKELY(last_idx >= field_item->field_idx_ && i != 0)) {
1073       ErrorStringPrintf("Out-of-order field_idx for annotation: %x then %x", last_idx, field_item->field_idx_);
1074       return false;
1075     }
1076     last_idx = field_item->field_idx_;
1077     field_item++;
1078   }
1079 
1080   // Method annotations follow immediately after field annotations.
1081   const DexFile::MethodAnnotationsItem* method_item =
1082       reinterpret_cast<const DexFile::MethodAnnotationsItem*>(field_item);
1083   uint32_t method_count = item->methods_size_;
1084   if (!CheckListSize(method_item, method_count, sizeof(DexFile::MethodAnnotationsItem), "method_annotations list")) {
1085     return false;
1086   }
1087 
1088   last_idx = 0;
1089   for (uint32_t i = 0; i < method_count; i++) {
1090     if (UNLIKELY(last_idx >= method_item->method_idx_ && i != 0)) {
1091       ErrorStringPrintf("Out-of-order method_idx for annotation: %x then %x",
1092                        last_idx, method_item->method_idx_);
1093       return false;
1094     }
1095     last_idx = method_item->method_idx_;
1096     method_item++;
1097   }
1098 
1099   // Parameter annotations follow immediately after method annotations.
1100   const DexFile::ParameterAnnotationsItem* parameter_item =
1101       reinterpret_cast<const DexFile::ParameterAnnotationsItem*>(method_item);
1102   uint32_t parameter_count = item->parameters_size_;
1103   if (!CheckListSize(parameter_item, parameter_count, sizeof(DexFile::ParameterAnnotationsItem),
1104                      "parameter_annotations list")) {
1105     return false;
1106   }
1107 
1108   last_idx = 0;
1109   for (uint32_t i = 0; i < parameter_count; i++) {
1110     if (UNLIKELY(last_idx >= parameter_item->method_idx_ && i != 0)) {
1111       ErrorStringPrintf("Out-of-order method_idx for annotation: %x then %x",
1112                         last_idx, parameter_item->method_idx_);
1113       return false;
1114     }
1115     last_idx = parameter_item->method_idx_;
1116     parameter_item++;
1117   }
1118 
1119   // Return a pointer to the end of the annotations.
1120   ptr_ = reinterpret_cast<const uint8_t*>(parameter_item);
1121   return true;
1122 }
1123 
CheckIntraSectionIterate(size_t offset,uint32_t section_count,uint16_t type)1124 bool DexFileVerifier::CheckIntraSectionIterate(size_t offset, uint32_t section_count,
1125                                                uint16_t type) {
1126   // Get the right alignment mask for the type of section.
1127   size_t alignment_mask;
1128   switch (type) {
1129     case DexFile::kDexTypeClassDataItem:
1130     case DexFile::kDexTypeStringDataItem:
1131     case DexFile::kDexTypeDebugInfoItem:
1132     case DexFile::kDexTypeAnnotationItem:
1133     case DexFile::kDexTypeEncodedArrayItem:
1134       alignment_mask = sizeof(uint8_t) - 1;
1135       break;
1136     default:
1137       alignment_mask = sizeof(uint32_t) - 1;
1138       break;
1139   }
1140 
1141   // Iterate through the items in the section.
1142   for (uint32_t i = 0; i < section_count; i++) {
1143     size_t aligned_offset = (offset + alignment_mask) & ~alignment_mask;
1144 
1145     // Check the padding between items.
1146     if (!CheckPadding(offset, aligned_offset)) {
1147       return false;
1148     }
1149 
1150     // Check depending on the section type.
1151     switch (type) {
1152       case DexFile::kDexTypeStringIdItem: {
1153         if (!CheckListSize(ptr_, 1, sizeof(DexFile::StringId), "string_ids")) {
1154           return false;
1155         }
1156         ptr_ += sizeof(DexFile::StringId);
1157         break;
1158       }
1159       case DexFile::kDexTypeTypeIdItem: {
1160         if (!CheckListSize(ptr_, 1, sizeof(DexFile::TypeId), "type_ids")) {
1161           return false;
1162         }
1163         ptr_ += sizeof(DexFile::TypeId);
1164         break;
1165       }
1166       case DexFile::kDexTypeProtoIdItem: {
1167         if (!CheckListSize(ptr_, 1, sizeof(DexFile::ProtoId), "proto_ids")) {
1168           return false;
1169         }
1170         ptr_ += sizeof(DexFile::ProtoId);
1171         break;
1172       }
1173       case DexFile::kDexTypeFieldIdItem: {
1174         if (!CheckListSize(ptr_, 1, sizeof(DexFile::FieldId), "field_ids")) {
1175           return false;
1176         }
1177         ptr_ += sizeof(DexFile::FieldId);
1178         break;
1179       }
1180       case DexFile::kDexTypeMethodIdItem: {
1181         if (!CheckListSize(ptr_, 1, sizeof(DexFile::MethodId), "method_ids")) {
1182           return false;
1183         }
1184         ptr_ += sizeof(DexFile::MethodId);
1185         break;
1186       }
1187       case DexFile::kDexTypeClassDefItem: {
1188         if (!CheckListSize(ptr_, 1, sizeof(DexFile::ClassDef), "class_defs")) {
1189           return false;
1190         }
1191         ptr_ += sizeof(DexFile::ClassDef);
1192         break;
1193       }
1194       case DexFile::kDexTypeTypeList: {
1195         if (!CheckList(sizeof(DexFile::TypeItem), "type_list", &ptr_)) {
1196           return false;
1197         }
1198         break;
1199       }
1200       case DexFile::kDexTypeAnnotationSetRefList: {
1201         if (!CheckList(sizeof(DexFile::AnnotationSetRefItem), "annotation_set_ref_list", &ptr_)) {
1202           return false;
1203         }
1204         break;
1205       }
1206       case DexFile::kDexTypeAnnotationSetItem: {
1207         if (!CheckList(sizeof(uint32_t), "annotation_set_item", &ptr_)) {
1208           return false;
1209         }
1210         break;
1211       }
1212       case DexFile::kDexTypeClassDataItem: {
1213         if (!CheckIntraClassDataItem()) {
1214           return false;
1215         }
1216         break;
1217       }
1218       case DexFile::kDexTypeCodeItem: {
1219         if (!CheckIntraCodeItem()) {
1220           return false;
1221         }
1222         break;
1223       }
1224       case DexFile::kDexTypeStringDataItem: {
1225         if (!CheckIntraStringDataItem()) {
1226           return false;
1227         }
1228         break;
1229       }
1230       case DexFile::kDexTypeDebugInfoItem: {
1231         if (!CheckIntraDebugInfoItem()) {
1232           return false;
1233         }
1234         break;
1235       }
1236       case DexFile::kDexTypeAnnotationItem: {
1237         if (!CheckIntraAnnotationItem()) {
1238           return false;
1239         }
1240         break;
1241       }
1242       case DexFile::kDexTypeEncodedArrayItem: {
1243         if (!CheckEncodedArray()) {
1244           return false;
1245         }
1246         break;
1247       }
1248       case DexFile::kDexTypeAnnotationsDirectoryItem: {
1249         if (!CheckIntraAnnotationsDirectoryItem()) {
1250           return false;
1251         }
1252         break;
1253       }
1254       default:
1255         ErrorStringPrintf("Unknown map item type %x", type);
1256         return false;
1257     }
1258 
1259     if (IsDataSectionType(type)) {
1260       offset_to_type_map_.Put(aligned_offset, type);
1261     }
1262 
1263     aligned_offset = ptr_ - begin_;
1264     if (UNLIKELY(aligned_offset > size_)) {
1265       ErrorStringPrintf("Item %d at ends out of bounds", i);
1266       return false;
1267     }
1268 
1269     offset = aligned_offset;
1270   }
1271 
1272   return true;
1273 }
1274 
CheckIntraIdSection(size_t offset,uint32_t count,uint16_t type)1275 bool DexFileVerifier::CheckIntraIdSection(size_t offset, uint32_t count, uint16_t type) {
1276   uint32_t expected_offset;
1277   uint32_t expected_size;
1278 
1279   // Get the expected offset and size from the header.
1280   switch (type) {
1281     case DexFile::kDexTypeStringIdItem:
1282       expected_offset = header_->string_ids_off_;
1283       expected_size = header_->string_ids_size_;
1284       break;
1285     case DexFile::kDexTypeTypeIdItem:
1286       expected_offset = header_->type_ids_off_;
1287       expected_size = header_->type_ids_size_;
1288       break;
1289     case DexFile::kDexTypeProtoIdItem:
1290       expected_offset = header_->proto_ids_off_;
1291       expected_size = header_->proto_ids_size_;
1292       break;
1293     case DexFile::kDexTypeFieldIdItem:
1294       expected_offset = header_->field_ids_off_;
1295       expected_size = header_->field_ids_size_;
1296       break;
1297     case DexFile::kDexTypeMethodIdItem:
1298       expected_offset = header_->method_ids_off_;
1299       expected_size = header_->method_ids_size_;
1300       break;
1301     case DexFile::kDexTypeClassDefItem:
1302       expected_offset = header_->class_defs_off_;
1303       expected_size = header_->class_defs_size_;
1304       break;
1305     default:
1306       ErrorStringPrintf("Bad type for id section: %x", type);
1307       return false;
1308   }
1309 
1310   // Check that the offset and size are what were expected from the header.
1311   if (UNLIKELY(offset != expected_offset)) {
1312     ErrorStringPrintf("Bad offset for section: got %zx, expected %x", offset, expected_offset);
1313     return false;
1314   }
1315   if (UNLIKELY(count != expected_size)) {
1316     ErrorStringPrintf("Bad size for section: got %x, expected %x", count, expected_size);
1317     return false;
1318   }
1319 
1320   return CheckIntraSectionIterate(offset, count, type);
1321 }
1322 
CheckIntraDataSection(size_t offset,uint32_t count,uint16_t type)1323 bool DexFileVerifier::CheckIntraDataSection(size_t offset, uint32_t count, uint16_t type) {
1324   size_t data_start = header_->data_off_;
1325   size_t data_end = data_start + header_->data_size_;
1326 
1327   // Sanity check the offset of the section.
1328   if (UNLIKELY((offset < data_start) || (offset > data_end))) {
1329     ErrorStringPrintf("Bad offset for data subsection: %zx", offset);
1330     return false;
1331   }
1332 
1333   if (!CheckIntraSectionIterate(offset, count, type)) {
1334     return false;
1335   }
1336 
1337   size_t next_offset = ptr_ - begin_;
1338   if (next_offset > data_end) {
1339     ErrorStringPrintf("Out-of-bounds end of data subsection: %zx", next_offset);
1340     return false;
1341   }
1342 
1343   return true;
1344 }
1345 
CheckIntraSection()1346 bool DexFileVerifier::CheckIntraSection() {
1347   const DexFile::MapList* map = reinterpret_cast<const DexFile::MapList*>(begin_ + header_->map_off_);
1348   const DexFile::MapItem* item = map->list_;
1349 
1350   uint32_t count = map->size_;
1351   size_t offset = 0;
1352   ptr_ = begin_;
1353 
1354   // Check the items listed in the map.
1355   while (count--) {
1356     uint32_t section_offset = item->offset_;
1357     uint32_t section_count = item->size_;
1358     uint16_t type = item->type_;
1359 
1360     // Check for padding and overlap between items.
1361     if (!CheckPadding(offset, section_offset)) {
1362       return false;
1363     } else if (UNLIKELY(offset > section_offset)) {
1364       ErrorStringPrintf("Section overlap or out-of-order map: %zx, %x", offset, section_offset);
1365       return false;
1366     }
1367 
1368     // Check each item based on its type.
1369     switch (type) {
1370       case DexFile::kDexTypeHeaderItem:
1371         if (UNLIKELY(section_count != 1)) {
1372           ErrorStringPrintf("Multiple header items");
1373           return false;
1374         }
1375         if (UNLIKELY(section_offset != 0)) {
1376           ErrorStringPrintf("Header at %x, not at start of file", section_offset);
1377           return false;
1378         }
1379         ptr_ = begin_ + header_->header_size_;
1380         offset = header_->header_size_;
1381         break;
1382       case DexFile::kDexTypeStringIdItem:
1383       case DexFile::kDexTypeTypeIdItem:
1384       case DexFile::kDexTypeProtoIdItem:
1385       case DexFile::kDexTypeFieldIdItem:
1386       case DexFile::kDexTypeMethodIdItem:
1387       case DexFile::kDexTypeClassDefItem:
1388         if (!CheckIntraIdSection(section_offset, section_count, type)) {
1389           return false;
1390         }
1391         offset = ptr_ - begin_;
1392         break;
1393       case DexFile::kDexTypeMapList:
1394         if (UNLIKELY(section_count != 1)) {
1395           ErrorStringPrintf("Multiple map list items");
1396           return false;
1397         }
1398         if (UNLIKELY(section_offset != header_->map_off_)) {
1399           ErrorStringPrintf("Map not at header-defined offset: %x, expected %x",
1400                             section_offset, header_->map_off_);
1401           return false;
1402         }
1403         ptr_ += sizeof(uint32_t) + (map->size_ * sizeof(DexFile::MapItem));
1404         offset = section_offset + sizeof(uint32_t) + (map->size_ * sizeof(DexFile::MapItem));
1405         break;
1406       case DexFile::kDexTypeTypeList:
1407       case DexFile::kDexTypeAnnotationSetRefList:
1408       case DexFile::kDexTypeAnnotationSetItem:
1409       case DexFile::kDexTypeClassDataItem:
1410       case DexFile::kDexTypeCodeItem:
1411       case DexFile::kDexTypeStringDataItem:
1412       case DexFile::kDexTypeDebugInfoItem:
1413       case DexFile::kDexTypeAnnotationItem:
1414       case DexFile::kDexTypeEncodedArrayItem:
1415       case DexFile::kDexTypeAnnotationsDirectoryItem:
1416         if (!CheckIntraDataSection(section_offset, section_count, type)) {
1417           return false;
1418         }
1419         offset = ptr_ - begin_;
1420         break;
1421       default:
1422         ErrorStringPrintf("Unknown map item type %x", type);
1423         return false;
1424     }
1425 
1426     item++;
1427   }
1428 
1429   return true;
1430 }
1431 
CheckOffsetToTypeMap(size_t offset,uint16_t type)1432 bool DexFileVerifier::CheckOffsetToTypeMap(size_t offset, uint16_t type) {
1433   auto it = offset_to_type_map_.find(offset);
1434   if (UNLIKELY(it == offset_to_type_map_.end())) {
1435     ErrorStringPrintf("No data map entry found @ %zx; expected %x", offset, type);
1436     return false;
1437   }
1438   if (UNLIKELY(it->second != type)) {
1439     ErrorStringPrintf("Unexpected data map entry @ %zx; expected %x, found %x",
1440                       offset, type, it->second);
1441     return false;
1442   }
1443   return true;
1444 }
1445 
FindFirstClassDataDefiner(const uint8_t * ptr,bool * success)1446 uint16_t DexFileVerifier::FindFirstClassDataDefiner(const uint8_t* ptr, bool* success) {
1447   ClassDataItemIterator it(*dex_file_, ptr);
1448   *success = true;
1449 
1450   if (it.HasNextStaticField() || it.HasNextInstanceField()) {
1451     LOAD_FIELD(field, it.GetMemberIndex(), "first_class_data_definer field_id",
1452                *success = false; return DexFile::kDexNoIndex16)
1453     return field->class_idx_;
1454   }
1455 
1456   if (it.HasNextDirectMethod() || it.HasNextVirtualMethod()) {
1457     LOAD_METHOD(method, it.GetMemberIndex(), "first_class_data_definer method_id",
1458                 *success = false; return DexFile::kDexNoIndex16)
1459     return method->class_idx_;
1460   }
1461 
1462   return DexFile::kDexNoIndex16;
1463 }
1464 
FindFirstAnnotationsDirectoryDefiner(const uint8_t * ptr,bool * success)1465 uint16_t DexFileVerifier::FindFirstAnnotationsDirectoryDefiner(const uint8_t* ptr, bool* success) {
1466   const DexFile::AnnotationsDirectoryItem* item =
1467       reinterpret_cast<const DexFile::AnnotationsDirectoryItem*>(ptr);
1468   *success = true;
1469 
1470   if (item->fields_size_ != 0) {
1471     DexFile::FieldAnnotationsItem* field_items = (DexFile::FieldAnnotationsItem*) (item + 1);
1472     LOAD_FIELD(field, field_items[0].field_idx_, "first_annotations_dir_definer field_id",
1473                *success = false; return DexFile::kDexNoIndex16)
1474     return field->class_idx_;
1475   }
1476 
1477   if (item->methods_size_ != 0) {
1478     DexFile::MethodAnnotationsItem* method_items = (DexFile::MethodAnnotationsItem*) (item + 1);
1479     LOAD_METHOD(method, method_items[0].method_idx_, "first_annotations_dir_definer method id",
1480                 *success = false; return DexFile::kDexNoIndex16)
1481     return method->class_idx_;
1482   }
1483 
1484   if (item->parameters_size_ != 0) {
1485     DexFile::ParameterAnnotationsItem* parameter_items = (DexFile::ParameterAnnotationsItem*) (item + 1);
1486     LOAD_METHOD(method, parameter_items[0].method_idx_, "first_annotations_dir_definer method id",
1487                 *success = false; return DexFile::kDexNoIndex16)
1488     return method->class_idx_;
1489   }
1490 
1491   return DexFile::kDexNoIndex16;
1492 }
1493 
CheckInterStringIdItem()1494 bool DexFileVerifier::CheckInterStringIdItem() {
1495   const DexFile::StringId* item = reinterpret_cast<const DexFile::StringId*>(ptr_);
1496 
1497   // Check the map to make sure it has the right offset->type.
1498   if (!CheckOffsetToTypeMap(item->string_data_off_, DexFile::kDexTypeStringDataItem)) {
1499     return false;
1500   }
1501 
1502   // Check ordering between items.
1503   if (previous_item_ != nullptr) {
1504     const DexFile::StringId* prev_item = reinterpret_cast<const DexFile::StringId*>(previous_item_);
1505     const char* prev_str = dex_file_->GetStringData(*prev_item);
1506     const char* str = dex_file_->GetStringData(*item);
1507     if (UNLIKELY(CompareModifiedUtf8ToModifiedUtf8AsUtf16CodePointValues(prev_str, str) >= 0)) {
1508       ErrorStringPrintf("Out-of-order string_ids: '%s' then '%s'", prev_str, str);
1509       return false;
1510     }
1511   }
1512 
1513   ptr_ += sizeof(DexFile::StringId);
1514   return true;
1515 }
1516 
CheckInterTypeIdItem()1517 bool DexFileVerifier::CheckInterTypeIdItem() {
1518   const DexFile::TypeId* item = reinterpret_cast<const DexFile::TypeId*>(ptr_);
1519 
1520   LOAD_STRING(descriptor, item->descriptor_idx_, "inter_type_id_item descriptor_idx")
1521 
1522   // Check that the descriptor is a valid type.
1523   if (UNLIKELY(!IsValidDescriptor(descriptor))) {
1524     ErrorStringPrintf("Invalid type descriptor: '%s'", descriptor);
1525     return false;
1526   }
1527 
1528   // Check ordering between items.
1529   if (previous_item_ != nullptr) {
1530     const DexFile::TypeId* prev_item = reinterpret_cast<const DexFile::TypeId*>(previous_item_);
1531     if (UNLIKELY(prev_item->descriptor_idx_ >= item->descriptor_idx_)) {
1532       ErrorStringPrintf("Out-of-order type_ids: %x then %x",
1533                         prev_item->descriptor_idx_, item->descriptor_idx_);
1534       return false;
1535     }
1536   }
1537 
1538   ptr_ += sizeof(DexFile::TypeId);
1539   return true;
1540 }
1541 
CheckInterProtoIdItem()1542 bool DexFileVerifier::CheckInterProtoIdItem() {
1543   const DexFile::ProtoId* item = reinterpret_cast<const DexFile::ProtoId*>(ptr_);
1544 
1545   LOAD_STRING(shorty, item->shorty_idx_, "inter_proto_id_item shorty_idx")
1546 
1547   if (item->parameters_off_ != 0 &&
1548       !CheckOffsetToTypeMap(item->parameters_off_, DexFile::kDexTypeTypeList)) {
1549     return false;
1550   }
1551 
1552   // Check the return type and advance the shorty.
1553   LOAD_STRING_BY_TYPE(return_type, item->return_type_idx_, "inter_proto_id_item return_type_idx")
1554   if (!CheckShortyDescriptorMatch(*shorty, return_type, true)) {
1555     return false;
1556   }
1557   shorty++;
1558 
1559   DexFileParameterIterator it(*dex_file_, *item);
1560   while (it.HasNext() && *shorty != '\0') {
1561     if (!CheckIndex(it.GetTypeIdx(), dex_file_->NumTypeIds(),
1562                     "inter_proto_id_item shorty type_idx")) {
1563       return false;
1564     }
1565     const char* descriptor = it.GetDescriptor();
1566     if (!CheckShortyDescriptorMatch(*shorty, descriptor, false)) {
1567       return false;
1568     }
1569     it.Next();
1570     shorty++;
1571   }
1572   if (UNLIKELY(it.HasNext() || *shorty != '\0')) {
1573     ErrorStringPrintf("Mismatched length for parameters and shorty");
1574     return false;
1575   }
1576 
1577   // Check ordering between items. This relies on type_ids being in order.
1578   if (previous_item_ != nullptr) {
1579     const DexFile::ProtoId* prev = reinterpret_cast<const DexFile::ProtoId*>(previous_item_);
1580     if (UNLIKELY(prev->return_type_idx_ > item->return_type_idx_)) {
1581       ErrorStringPrintf("Out-of-order proto_id return types");
1582       return false;
1583     } else if (prev->return_type_idx_ == item->return_type_idx_) {
1584       DexFileParameterIterator curr_it(*dex_file_, *item);
1585       DexFileParameterIterator prev_it(*dex_file_, *prev);
1586 
1587       while (curr_it.HasNext() && prev_it.HasNext()) {
1588         uint16_t prev_idx = prev_it.GetTypeIdx();
1589         uint16_t curr_idx = curr_it.GetTypeIdx();
1590         if (prev_idx == DexFile::kDexNoIndex16) {
1591           break;
1592         }
1593         if (UNLIKELY(curr_idx == DexFile::kDexNoIndex16)) {
1594           ErrorStringPrintf("Out-of-order proto_id arguments");
1595           return false;
1596         }
1597 
1598         if (prev_idx < curr_idx) {
1599           break;
1600         } else if (UNLIKELY(prev_idx > curr_idx)) {
1601           ErrorStringPrintf("Out-of-order proto_id arguments");
1602           return false;
1603         }
1604 
1605         prev_it.Next();
1606         curr_it.Next();
1607       }
1608     }
1609   }
1610 
1611   ptr_ += sizeof(DexFile::ProtoId);
1612   return true;
1613 }
1614 
CheckInterFieldIdItem()1615 bool DexFileVerifier::CheckInterFieldIdItem() {
1616   const DexFile::FieldId* item = reinterpret_cast<const DexFile::FieldId*>(ptr_);
1617 
1618   // Check that the class descriptor is valid.
1619   LOAD_STRING_BY_TYPE(class_descriptor, item->class_idx_, "inter_field_id_item class_idx")
1620   if (UNLIKELY(!IsValidDescriptor(class_descriptor) || class_descriptor[0] != 'L')) {
1621     ErrorStringPrintf("Invalid descriptor for class_idx: '%s'", class_descriptor);
1622     return false;
1623   }
1624 
1625   // Check that the type descriptor is a valid field name.
1626   LOAD_STRING_BY_TYPE(type_descriptor, item->type_idx_, "inter_field_id_item type_idx")
1627   if (UNLIKELY(!IsValidDescriptor(type_descriptor) || type_descriptor[0] == 'V')) {
1628     ErrorStringPrintf("Invalid descriptor for type_idx: '%s'", type_descriptor);
1629     return false;
1630   }
1631 
1632   // Check that the name is valid.
1633   LOAD_STRING(descriptor, item->name_idx_, "inter_field_id_item name_idx")
1634   if (UNLIKELY(!IsValidMemberName(descriptor))) {
1635     ErrorStringPrintf("Invalid field name: '%s'", descriptor);
1636     return false;
1637   }
1638 
1639   // Check ordering between items. This relies on the other sections being in order.
1640   if (previous_item_ != nullptr) {
1641     const DexFile::FieldId* prev_item = reinterpret_cast<const DexFile::FieldId*>(previous_item_);
1642     if (UNLIKELY(prev_item->class_idx_ > item->class_idx_)) {
1643       ErrorStringPrintf("Out-of-order field_ids");
1644       return false;
1645     } else if (prev_item->class_idx_ == item->class_idx_) {
1646       if (UNLIKELY(prev_item->name_idx_ > item->name_idx_)) {
1647         ErrorStringPrintf("Out-of-order field_ids");
1648         return false;
1649       } else if (prev_item->name_idx_ == item->name_idx_) {
1650         if (UNLIKELY(prev_item->type_idx_ >= item->type_idx_)) {
1651           ErrorStringPrintf("Out-of-order field_ids");
1652           return false;
1653         }
1654       }
1655     }
1656   }
1657 
1658   ptr_ += sizeof(DexFile::FieldId);
1659   return true;
1660 }
1661 
CheckInterMethodIdItem()1662 bool DexFileVerifier::CheckInterMethodIdItem() {
1663   const DexFile::MethodId* item = reinterpret_cast<const DexFile::MethodId*>(ptr_);
1664 
1665   // Check that the class descriptor is a valid reference name.
1666   LOAD_STRING_BY_TYPE(class_descriptor, item->class_idx_, "inter_method_id_item class_idx")
1667   if (UNLIKELY(!IsValidDescriptor(class_descriptor) || (class_descriptor[0] != 'L' &&
1668                                                         class_descriptor[0] != '['))) {
1669     ErrorStringPrintf("Invalid descriptor for class_idx: '%s'", class_descriptor);
1670     return false;
1671   }
1672 
1673   // Check that the name is valid.
1674   LOAD_STRING(descriptor, item->name_idx_, "inter_method_id_item name_idx")
1675   if (UNLIKELY(!IsValidMemberName(descriptor))) {
1676     ErrorStringPrintf("Invalid method name: '%s'", descriptor);
1677     return false;
1678   }
1679 
1680   // Check that the proto id is valid.
1681   if (UNLIKELY(!CheckIndex(item->proto_idx_, dex_file_->NumProtoIds(),
1682                            "inter_method_id_item proto_idx"))) {
1683     return false;
1684   }
1685 
1686   // Check ordering between items. This relies on the other sections being in order.
1687   if (previous_item_ != nullptr) {
1688     const DexFile::MethodId* prev_item = reinterpret_cast<const DexFile::MethodId*>(previous_item_);
1689     if (UNLIKELY(prev_item->class_idx_ > item->class_idx_)) {
1690       ErrorStringPrintf("Out-of-order method_ids");
1691       return false;
1692     } else if (prev_item->class_idx_ == item->class_idx_) {
1693       if (UNLIKELY(prev_item->name_idx_ > item->name_idx_)) {
1694         ErrorStringPrintf("Out-of-order method_ids");
1695         return false;
1696       } else if (prev_item->name_idx_ == item->name_idx_) {
1697         if (UNLIKELY(prev_item->proto_idx_ >= item->proto_idx_)) {
1698           ErrorStringPrintf("Out-of-order method_ids");
1699           return false;
1700         }
1701       }
1702     }
1703   }
1704 
1705   ptr_ += sizeof(DexFile::MethodId);
1706   return true;
1707 }
1708 
CheckInterClassDefItem()1709 bool DexFileVerifier::CheckInterClassDefItem() {
1710   const DexFile::ClassDef* item = reinterpret_cast<const DexFile::ClassDef*>(ptr_);
1711 
1712   // Check for duplicate class def.
1713   if (defined_classes_.find(item->class_idx_) != defined_classes_.end()) {
1714     ErrorStringPrintf("Redefinition of class with type idx: '%d'", item->class_idx_);
1715     return false;
1716   }
1717   defined_classes_.insert(item->class_idx_);
1718 
1719   LOAD_STRING_BY_TYPE(class_descriptor, item->class_idx_, "inter_class_def_item class_idx")
1720   if (UNLIKELY(!IsValidDescriptor(class_descriptor) || class_descriptor[0] != 'L')) {
1721     ErrorStringPrintf("Invalid class descriptor: '%s'", class_descriptor);
1722     return false;
1723   }
1724 
1725   // Only allow non-runtime modifiers.
1726   if ((item->access_flags_ & ~kAccJavaFlagsMask) != 0) {
1727     ErrorStringPrintf("Invalid class flags: '%d'", item->access_flags_);
1728     return false;
1729   }
1730 
1731   if (item->interfaces_off_ != 0 &&
1732       !CheckOffsetToTypeMap(item->interfaces_off_, DexFile::kDexTypeTypeList)) {
1733     return false;
1734   }
1735   if (item->annotations_off_ != 0 &&
1736       !CheckOffsetToTypeMap(item->annotations_off_, DexFile::kDexTypeAnnotationsDirectoryItem)) {
1737     return false;
1738   }
1739   if (item->class_data_off_ != 0 &&
1740       !CheckOffsetToTypeMap(item->class_data_off_, DexFile::kDexTypeClassDataItem)) {
1741     return false;
1742   }
1743   if (item->static_values_off_ != 0 &&
1744       !CheckOffsetToTypeMap(item->static_values_off_, DexFile::kDexTypeEncodedArrayItem)) {
1745     return false;
1746   }
1747 
1748   if (item->superclass_idx_ != DexFile::kDexNoIndex16) {
1749     LOAD_STRING_BY_TYPE(superclass_descriptor, item->superclass_idx_,
1750                         "inter_class_def_item superclass_idx")
1751     if (UNLIKELY(!IsValidDescriptor(superclass_descriptor) || superclass_descriptor[0] != 'L')) {
1752       ErrorStringPrintf("Invalid superclass: '%s'", superclass_descriptor);
1753       return false;
1754     }
1755   }
1756 
1757   const DexFile::TypeList* interfaces = dex_file_->GetInterfacesList(*item);
1758   if (interfaces != nullptr) {
1759     uint32_t size = interfaces->Size();
1760 
1761     // Ensure that all interfaces refer to classes (not arrays or primitives).
1762     for (uint32_t i = 0; i < size; i++) {
1763       LOAD_STRING_BY_TYPE(inf_descriptor, interfaces->GetTypeItem(i).type_idx_,
1764                           "inter_class_def_item interface type_idx")
1765       if (UNLIKELY(!IsValidDescriptor(inf_descriptor) || inf_descriptor[0] != 'L')) {
1766         ErrorStringPrintf("Invalid interface: '%s'", inf_descriptor);
1767         return false;
1768       }
1769     }
1770 
1771     /*
1772      * Ensure that there are no duplicates. This is an O(N^2) test, but in
1773      * practice the number of interfaces implemented by any given class is low.
1774      */
1775     for (uint32_t i = 1; i < size; i++) {
1776       uint32_t idx1 = interfaces->GetTypeItem(i).type_idx_;
1777       for (uint32_t j =0; j < i; j++) {
1778         uint32_t idx2 = interfaces->GetTypeItem(j).type_idx_;
1779         if (UNLIKELY(idx1 == idx2)) {
1780           ErrorStringPrintf("Duplicate interface: '%s'", dex_file_->StringByTypeIdx(idx1));
1781           return false;
1782         }
1783       }
1784     }
1785   }
1786 
1787   // Check that references in class_data_item are to the right class.
1788   if (item->class_data_off_ != 0) {
1789     const uint8_t* data = begin_ + item->class_data_off_;
1790     bool success;
1791     uint16_t data_definer = FindFirstClassDataDefiner(data, &success);
1792     if (!success) {
1793       return false;
1794     }
1795     if (UNLIKELY((data_definer != item->class_idx_) && (data_definer != DexFile::kDexNoIndex16))) {
1796       ErrorStringPrintf("Invalid class_data_item");
1797       return false;
1798     }
1799   }
1800 
1801   // Check that references in annotations_directory_item are to right class.
1802   if (item->annotations_off_ != 0) {
1803     const uint8_t* data = begin_ + item->annotations_off_;
1804     bool success;
1805     uint16_t annotations_definer = FindFirstAnnotationsDirectoryDefiner(data, &success);
1806     if (!success) {
1807       return false;
1808     }
1809     if (UNLIKELY((annotations_definer != item->class_idx_) &&
1810                  (annotations_definer != DexFile::kDexNoIndex16))) {
1811       ErrorStringPrintf("Invalid annotations_directory_item");
1812       return false;
1813     }
1814   }
1815 
1816   ptr_ += sizeof(DexFile::ClassDef);
1817   return true;
1818 }
1819 
CheckInterAnnotationSetRefList()1820 bool DexFileVerifier::CheckInterAnnotationSetRefList() {
1821   const DexFile::AnnotationSetRefList* list =
1822       reinterpret_cast<const DexFile::AnnotationSetRefList*>(ptr_);
1823   const DexFile::AnnotationSetRefItem* item = list->list_;
1824   uint32_t count = list->size_;
1825 
1826   while (count--) {
1827     if (item->annotations_off_ != 0 &&
1828         !CheckOffsetToTypeMap(item->annotations_off_, DexFile::kDexTypeAnnotationSetItem)) {
1829       return false;
1830     }
1831     item++;
1832   }
1833 
1834   ptr_ = reinterpret_cast<const uint8_t*>(item);
1835   return true;
1836 }
1837 
CheckInterAnnotationSetItem()1838 bool DexFileVerifier::CheckInterAnnotationSetItem() {
1839   const DexFile::AnnotationSetItem* set = reinterpret_cast<const DexFile::AnnotationSetItem*>(ptr_);
1840   const uint32_t* offsets = set->entries_;
1841   uint32_t count = set->size_;
1842   uint32_t last_idx = 0;
1843 
1844   for (uint32_t i = 0; i < count; i++) {
1845     if (*offsets != 0 && !CheckOffsetToTypeMap(*offsets, DexFile::kDexTypeAnnotationItem)) {
1846       return false;
1847     }
1848 
1849     // Get the annotation from the offset and the type index for the annotation.
1850     const DexFile::AnnotationItem* annotation =
1851         reinterpret_cast<const DexFile::AnnotationItem*>(begin_ + *offsets);
1852     const uint8_t* data = annotation->annotation_;
1853     uint32_t idx = DecodeUnsignedLeb128(&data);
1854 
1855     if (UNLIKELY(last_idx >= idx && i != 0)) {
1856       ErrorStringPrintf("Out-of-order entry types: %x then %x", last_idx, idx);
1857       return false;
1858     }
1859 
1860     last_idx = idx;
1861     offsets++;
1862   }
1863 
1864   ptr_ = reinterpret_cast<const uint8_t*>(offsets);
1865   return true;
1866 }
1867 
CheckInterClassDataItem()1868 bool DexFileVerifier::CheckInterClassDataItem() {
1869   ClassDataItemIterator it(*dex_file_, ptr_);
1870   bool success;
1871   uint16_t defining_class = FindFirstClassDataDefiner(ptr_, &success);
1872   if (!success) {
1873     return false;
1874   }
1875 
1876   for (; it.HasNextStaticField() || it.HasNextInstanceField(); it.Next()) {
1877     LOAD_FIELD(field, it.GetMemberIndex(), "inter_class_data_item field_id", return false)
1878     if (UNLIKELY(field->class_idx_ != defining_class)) {
1879       ErrorStringPrintf("Mismatched defining class for class_data_item field");
1880       return false;
1881     }
1882   }
1883   for (; it.HasNextDirectMethod() || it.HasNextVirtualMethod(); it.Next()) {
1884     uint32_t code_off = it.GetMethodCodeItemOffset();
1885     if (code_off != 0 && !CheckOffsetToTypeMap(code_off, DexFile::kDexTypeCodeItem)) {
1886       return false;
1887     }
1888     LOAD_METHOD(method, it.GetMemberIndex(), "inter_class_data_item method_id", return false)
1889     if (UNLIKELY(method->class_idx_ != defining_class)) {
1890       ErrorStringPrintf("Mismatched defining class for class_data_item method");
1891       return false;
1892     }
1893   }
1894 
1895   ptr_ = it.EndDataPointer();
1896   return true;
1897 }
1898 
CheckInterAnnotationsDirectoryItem()1899 bool DexFileVerifier::CheckInterAnnotationsDirectoryItem() {
1900   const DexFile::AnnotationsDirectoryItem* item =
1901       reinterpret_cast<const DexFile::AnnotationsDirectoryItem*>(ptr_);
1902   bool success;
1903   uint16_t defining_class = FindFirstAnnotationsDirectoryDefiner(ptr_, &success);
1904   if (!success) {
1905     return false;
1906   }
1907 
1908   if (item->class_annotations_off_ != 0 &&
1909       !CheckOffsetToTypeMap(item->class_annotations_off_, DexFile::kDexTypeAnnotationSetItem)) {
1910     return false;
1911   }
1912 
1913   // Field annotations follow immediately after the annotations directory.
1914   const DexFile::FieldAnnotationsItem* field_item =
1915       reinterpret_cast<const DexFile::FieldAnnotationsItem*>(item + 1);
1916   uint32_t field_count = item->fields_size_;
1917   for (uint32_t i = 0; i < field_count; i++) {
1918     LOAD_FIELD(field, field_item->field_idx_, "inter_annotations_directory_item field_id",
1919                return false)
1920     if (UNLIKELY(field->class_idx_ != defining_class)) {
1921       ErrorStringPrintf("Mismatched defining class for field_annotation");
1922       return false;
1923     }
1924     if (!CheckOffsetToTypeMap(field_item->annotations_off_, DexFile::kDexTypeAnnotationSetItem)) {
1925       return false;
1926     }
1927     field_item++;
1928   }
1929 
1930   // Method annotations follow immediately after field annotations.
1931   const DexFile::MethodAnnotationsItem* method_item =
1932       reinterpret_cast<const DexFile::MethodAnnotationsItem*>(field_item);
1933   uint32_t method_count = item->methods_size_;
1934   for (uint32_t i = 0; i < method_count; i++) {
1935     LOAD_METHOD(method, method_item->method_idx_, "inter_annotations_directory_item method_id",
1936                 return false)
1937     if (UNLIKELY(method->class_idx_ != defining_class)) {
1938       ErrorStringPrintf("Mismatched defining class for method_annotation");
1939       return false;
1940     }
1941     if (!CheckOffsetToTypeMap(method_item->annotations_off_, DexFile::kDexTypeAnnotationSetItem)) {
1942       return false;
1943     }
1944     method_item++;
1945   }
1946 
1947   // Parameter annotations follow immediately after method annotations.
1948   const DexFile::ParameterAnnotationsItem* parameter_item =
1949       reinterpret_cast<const DexFile::ParameterAnnotationsItem*>(method_item);
1950   uint32_t parameter_count = item->parameters_size_;
1951   for (uint32_t i = 0; i < parameter_count; i++) {
1952     LOAD_METHOD(parameter_method, parameter_item->method_idx_,
1953                 "inter_annotations_directory_item parameter method_id", return false)
1954     if (UNLIKELY(parameter_method->class_idx_ != defining_class)) {
1955       ErrorStringPrintf("Mismatched defining class for parameter_annotation");
1956       return false;
1957     }
1958     if (!CheckOffsetToTypeMap(parameter_item->annotations_off_,
1959         DexFile::kDexTypeAnnotationSetRefList)) {
1960       return false;
1961     }
1962     parameter_item++;
1963   }
1964 
1965   ptr_ = reinterpret_cast<const uint8_t*>(parameter_item);
1966   return true;
1967 }
1968 
CheckInterSectionIterate(size_t offset,uint32_t count,uint16_t type)1969 bool DexFileVerifier::CheckInterSectionIterate(size_t offset, uint32_t count, uint16_t type) {
1970   // Get the right alignment mask for the type of section.
1971   size_t alignment_mask;
1972   switch (type) {
1973     case DexFile::kDexTypeClassDataItem:
1974       alignment_mask = sizeof(uint8_t) - 1;
1975       break;
1976     default:
1977       alignment_mask = sizeof(uint32_t) - 1;
1978       break;
1979   }
1980 
1981   // Iterate through the items in the section.
1982   previous_item_ = nullptr;
1983   for (uint32_t i = 0; i < count; i++) {
1984     uint32_t new_offset = (offset + alignment_mask) & ~alignment_mask;
1985     ptr_ = begin_ + new_offset;
1986     const uint8_t* prev_ptr = ptr_;
1987 
1988     // Check depending on the section type.
1989     switch (type) {
1990       case DexFile::kDexTypeStringIdItem: {
1991         if (!CheckInterStringIdItem()) {
1992           return false;
1993         }
1994         break;
1995       }
1996       case DexFile::kDexTypeTypeIdItem: {
1997         if (!CheckInterTypeIdItem()) {
1998           return false;
1999         }
2000         break;
2001       }
2002       case DexFile::kDexTypeProtoIdItem: {
2003         if (!CheckInterProtoIdItem()) {
2004           return false;
2005         }
2006         break;
2007       }
2008       case DexFile::kDexTypeFieldIdItem: {
2009         if (!CheckInterFieldIdItem()) {
2010           return false;
2011         }
2012         break;
2013       }
2014       case DexFile::kDexTypeMethodIdItem: {
2015         if (!CheckInterMethodIdItem()) {
2016           return false;
2017         }
2018         break;
2019       }
2020       case DexFile::kDexTypeClassDefItem: {
2021         if (!CheckInterClassDefItem()) {
2022           return false;
2023         }
2024         break;
2025       }
2026       case DexFile::kDexTypeAnnotationSetRefList: {
2027         if (!CheckInterAnnotationSetRefList()) {
2028           return false;
2029         }
2030         break;
2031       }
2032       case DexFile::kDexTypeAnnotationSetItem: {
2033         if (!CheckInterAnnotationSetItem()) {
2034           return false;
2035         }
2036         break;
2037       }
2038       case DexFile::kDexTypeClassDataItem: {
2039         if (!CheckInterClassDataItem()) {
2040           return false;
2041         }
2042         break;
2043       }
2044       case DexFile::kDexTypeAnnotationsDirectoryItem: {
2045         if (!CheckInterAnnotationsDirectoryItem()) {
2046           return false;
2047         }
2048         break;
2049       }
2050       default:
2051         ErrorStringPrintf("Unknown map item type %x", type);
2052         return false;
2053     }
2054 
2055     previous_item_ = prev_ptr;
2056     offset = ptr_ - begin_;
2057   }
2058 
2059   return true;
2060 }
2061 
CheckInterSection()2062 bool DexFileVerifier::CheckInterSection() {
2063   const DexFile::MapList* map = reinterpret_cast<const DexFile::MapList*>(begin_ + header_->map_off_);
2064   const DexFile::MapItem* item = map->list_;
2065   uint32_t count = map->size_;
2066 
2067   // Cross check the items listed in the map.
2068   while (count--) {
2069     uint32_t section_offset = item->offset_;
2070     uint32_t section_count = item->size_;
2071     uint16_t type = item->type_;
2072 
2073     switch (type) {
2074       case DexFile::kDexTypeHeaderItem:
2075       case DexFile::kDexTypeMapList:
2076       case DexFile::kDexTypeTypeList:
2077       case DexFile::kDexTypeCodeItem:
2078       case DexFile::kDexTypeStringDataItem:
2079       case DexFile::kDexTypeDebugInfoItem:
2080       case DexFile::kDexTypeAnnotationItem:
2081       case DexFile::kDexTypeEncodedArrayItem:
2082         break;
2083       case DexFile::kDexTypeStringIdItem:
2084       case DexFile::kDexTypeTypeIdItem:
2085       case DexFile::kDexTypeProtoIdItem:
2086       case DexFile::kDexTypeFieldIdItem:
2087       case DexFile::kDexTypeMethodIdItem:
2088       case DexFile::kDexTypeClassDefItem:
2089       case DexFile::kDexTypeAnnotationSetRefList:
2090       case DexFile::kDexTypeAnnotationSetItem:
2091       case DexFile::kDexTypeClassDataItem:
2092       case DexFile::kDexTypeAnnotationsDirectoryItem: {
2093         if (!CheckInterSectionIterate(section_offset, section_count, type)) {
2094           return false;
2095         }
2096         break;
2097       }
2098       default:
2099         ErrorStringPrintf("Unknown map item type %x", type);
2100         return false;
2101     }
2102 
2103     item++;
2104   }
2105 
2106   return true;
2107 }
2108 
Verify()2109 bool DexFileVerifier::Verify() {
2110   // Check the header.
2111   if (!CheckHeader()) {
2112     return false;
2113   }
2114 
2115   // Check the map section.
2116   if (!CheckMap()) {
2117     return false;
2118   }
2119 
2120   // Check structure within remaining sections.
2121   if (!CheckIntraSection()) {
2122     return false;
2123   }
2124 
2125   // Check references from one section to another.
2126   if (!CheckInterSection()) {
2127     return false;
2128   }
2129 
2130   return true;
2131 }
2132 
ErrorStringPrintf(const char * fmt,...)2133 void DexFileVerifier::ErrorStringPrintf(const char* fmt, ...) {
2134   va_list ap;
2135   va_start(ap, fmt);
2136   DCHECK(failure_reason_.empty()) << failure_reason_;
2137   failure_reason_ = StringPrintf("Failure to verify dex file '%s': ", location_);
2138   StringAppendV(&failure_reason_, fmt, ap);
2139   va_end(ap);
2140 }
2141 
2142 }  // namespace art
2143