1959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle/* 2959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle * Copyright (C) 2014 The Android Open Source Project 3959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle * 4959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle * Licensed under the Apache License, Version 2.0 (the "License"); 5959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle * you may not use this file except in compliance with the License. 6959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle * You may obtain a copy of the License at 7959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle * 8959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle * http://www.apache.org/licenses/LICENSE-2.0 9959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle * 10959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle * Unless required by applicable law or agreed to in writing, software 11959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle * distributed under the License is distributed on an "AS IS" BASIS, 12959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle * See the License for the specific language governing permissions and 14959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle * limitations under the License. 15959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle */ 16959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle 17959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kylepackage dexfuzz.rawdex; 18959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle 19959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyleimport dexfuzz.Log; 20959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle 21959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyleimport java.io.IOException; 22959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyleimport java.nio.charset.StandardCharsets; 23959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle 24959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kylepublic class StringDataItem implements RawDexObject { 25959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle private int size; 26959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle private String data; 27959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle private byte[] dataAsBytes; 28959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle private boolean writeRawBytes; 29959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle 30959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle @Override 31959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle public void read(DexRandomAccessFile file) throws IOException { 32959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle file.getOffsetTracker().getNewOffsettable(file, this); 33959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle size = file.readUleb128(); 34959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle if (size != 0) { 35959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle dataAsBytes = file.readDexUtf(size); 36959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle data = new String(dataAsBytes, StandardCharsets.US_ASCII); 37959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle if (size != data.length()) { 38959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle Log.warn("Don't have full support for decoding MUTF-8 yet, DEX file " 39959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle + "may be incorrectly mutated. Avoid using this test case for now."); 40959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle writeRawBytes = true; 41959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } 42959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } else { 43959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle // Read past the null byte. 44959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle file.readByte(); 45959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } 46959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } 47959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle 48959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle @Override 49959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle public void write(DexRandomAccessFile file) throws IOException { 50959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle file.getOffsetTracker().updatePositionOfNextOffsettable(file); 51959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle file.writeUleb128(size); 52959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle if (size > 0) { 53959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle if (writeRawBytes) { 54959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle file.writeDexUtf(dataAsBytes); 55959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } else { 56959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle file.writeDexUtf(data.getBytes(StandardCharsets.US_ASCII)); 57959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } 58959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } else { 59959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle // Write out the null byte. 60959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle file.writeByte(0); 61959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } 62959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } 63959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle 64959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle @Override 65959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle public void incrementIndex(IndexUpdateKind kind, int insertedIdx) { 66959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle // Do nothing. 67959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } 68959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle 69959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle public void setSize(int size) { 70959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle this.size = size; 71959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } 72959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle 73959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle public int getSize() { 74959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle return size; 75959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } 76959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle 77959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle public void setString(String data) { 78959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle this.data = data; 79959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } 80959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle 81959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle public String getString() { 82959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle if (writeRawBytes) { 83959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle Log.warn("Reading a string that hasn't been properly decoded! Returning empty string."); 84959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle return ""; 85959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } 86959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle return data; 87959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle } 88959ffdf65f280ee90b7944a8dd610564e7f99e69Stephen Kyle} 89