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 }