1 /* 2 * Copyright (c) 2015, 2017, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. Oracle designates this 8 * particular file as subject to the "Classpath" exception as provided 9 * by Oracle in the LICENSE file that accompanied this code. 10 * 11 * This code is distributed in the hope that it will be useful, but WITHOUT 12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 14 * version 2 for more details (a copy is included in the LICENSE file that 15 * accompanied this code). 16 * 17 * You should have received a copy of the GNU General Public License version 18 * 2 along with this work; if not, write to the Free Software Foundation, 19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 20 * 21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 22 * or visit www.oracle.com if you need additional information or have any 23 * questions. 24 */ 25 26 #include "ByteBuffer.h" 27 28 #include <stdio.h> 29 30 ByteBuffer::ByteBuffer() 31 { 32 buffer.reserve(1024); 33 } 34 35 ByteBuffer::~ByteBuffer() 36 { 37 } 38 39 LPBYTE ByteBuffer::getPtr() { 40 return &buffer[0]; 41 } 42 43 size_t ByteBuffer::getPos() { 44 return buffer.size(); 45 } 46 47 void ByteBuffer::AppendString(wstring str) { 48 size_t len = (str.size() + 1) * sizeof WCHAR; 49 AppendBytes((BYTE*)str.c_str(), len); 50 } 51 52 void ByteBuffer::AppendWORD(WORD word) { 53 AppendBytes((BYTE*)&word, sizeof WORD); 54 } 55 56 void ByteBuffer::Align(size_t bytesNumber) { 57 size_t pos = getPos(); 58 if (pos % bytesNumber) { 59 DWORD dwNull = 0; 60 size_t len = bytesNumber - pos % bytesNumber; 61 AppendBytes((BYTE*)&dwNull, len); 62 } 63 } 64 65 void ByteBuffer::AppendBytes(BYTE* ptr, size_t len) { 66 buffer.insert(buffer.end(), ptr, ptr + len); 67 } 68 69 void ByteBuffer::ReplaceWORD(size_t offset, WORD word) { 70 ReplaceBytes(offset, (BYTE*)&word, sizeof WORD); 71 } 72 73 void ByteBuffer::ReplaceBytes(size_t offset, BYTE* ptr, size_t len) { 74 for (size_t i = 0; i < len; i++) { 75 buffer[offset + i] = *(ptr + i); 76 } 77 }