1 /*

     2  * Copyright (c) 2015, 2019, 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     buffer.reserve(1024);

    32 }

    33 

    34 ByteBuffer::~ByteBuffer() {

    35 }

    36 

    37 LPBYTE ByteBuffer::getPtr() {

    38     return &buffer[0];

    39 }

    40 

    41 size_t ByteBuffer::getPos() {

    42     return buffer.size();

    43 }

    44 

    45 void ByteBuffer::AppendString(wstring str) {

    46     size_t len = (str.size() + 1) * sizeof (WCHAR);

    47     AppendBytes((BYTE*) str.c_str(), len);

    48 }

    49 

    50 void ByteBuffer::AppendWORD(WORD word) {

    51     AppendBytes((BYTE*) & word, sizeof (WORD));

    52 }

    53 

    54 void ByteBuffer::Align(size_t bytesNumber) {

    55     size_t pos = getPos();

    56     if (pos % bytesNumber) {

    57         DWORD dwNull = 0;

    58         size_t len = bytesNumber - pos % bytesNumber;

    59         AppendBytes((BYTE*) & dwNull, len);

    60     }

    61 }

    62 

    63 void ByteBuffer::AppendBytes(BYTE* ptr, size_t len) {

    64     buffer.insert(buffer.end(), ptr, ptr + len);

    65 }

    66 

    67 void ByteBuffer::ReplaceWORD(size_t offset, WORD word) {

    68     ReplaceBytes(offset, (BYTE*) & word, sizeof (WORD));

    69 }

    70 

    71 void ByteBuffer::ReplaceBytes(size_t offset, BYTE* ptr, size_t len) {

    72     for (size_t i = 0; i < len; i++) {

    73         buffer[offset + i] = *(ptr + i);

    74     }

    75 }