HBASE-21401 Sanity check when constructing the KeyValue
This commit is contained in:
parent
ad819380c7
commit
12786f80c1
|
@ -251,6 +251,15 @@ public class KeyValue implements ExtendedCell, Cloneable {
|
|||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* True to indicate that the byte b is a valid type.
|
||||
* @param b byte to check
|
||||
* @return true or false
|
||||
*/
|
||||
static boolean isValidType(byte b) {
|
||||
return codeArray[b & 0xff] != null;
|
||||
}
|
||||
|
||||
/**
|
||||
* Cannot rely on enum ordinals . They change if item is removed or moved.
|
||||
* Do our own codes.
|
||||
|
@ -332,6 +341,7 @@ public class KeyValue implements ExtendedCell, Cloneable {
|
|||
* @param length length of the KeyValue
|
||||
*/
|
||||
public KeyValue(final byte[] bytes, final int offset, final int length) {
|
||||
KeyValueUtil.checkKeyValueBytes(bytes, offset, length, true);
|
||||
this.bytes = bytes;
|
||||
this.offset = offset;
|
||||
this.length = length;
|
||||
|
|
|
@ -518,17 +518,145 @@ public class KeyValueUtil {
|
|||
return (long) length + Bytes.SIZEOF_INT;
|
||||
}
|
||||
|
||||
static String bytesToHex(byte[] buf, int offset, int length) {
|
||||
return ", KeyValueBytesHex=" + Bytes.toStringBinary(buf, offset, length) + ", offset=" + offset
|
||||
+ ", length=" + length;
|
||||
}
|
||||
|
||||
static void checkKeyValueBytes(byte[] buf, int offset, int length, boolean withTags) {
|
||||
int pos = offset, endOffset = offset + length;
|
||||
// check the key
|
||||
if (pos + Bytes.SIZEOF_INT > endOffset) {
|
||||
throw new IllegalArgumentException(
|
||||
"Overflow when reading key length at position=" + pos + bytesToHex(buf, offset, length));
|
||||
}
|
||||
int keyLen = Bytes.toInt(buf, pos, Bytes.SIZEOF_INT);
|
||||
pos += Bytes.SIZEOF_INT;
|
||||
if (keyLen <= 0 || pos + keyLen > endOffset) {
|
||||
throw new IllegalArgumentException(
|
||||
"Invalid key length in KeyValue. keyLength=" + keyLen + bytesToHex(buf, offset, length));
|
||||
}
|
||||
// check the value
|
||||
if (pos + Bytes.SIZEOF_INT > endOffset) {
|
||||
throw new IllegalArgumentException("Overflow when reading value length at position=" + pos
|
||||
+ bytesToHex(buf, offset, length));
|
||||
}
|
||||
int valLen = Bytes.toInt(buf, pos, Bytes.SIZEOF_INT);
|
||||
pos += Bytes.SIZEOF_INT;
|
||||
if (valLen < 0 || pos + valLen > endOffset) {
|
||||
throw new IllegalArgumentException("Invalid value length in KeyValue, valueLength=" + valLen
|
||||
+ bytesToHex(buf, offset, length));
|
||||
}
|
||||
// check the row
|
||||
if (pos + Bytes.SIZEOF_SHORT > endOffset) {
|
||||
throw new IllegalArgumentException(
|
||||
"Overflow when reading row length at position=" + pos + bytesToHex(buf, offset, length));
|
||||
}
|
||||
short rowLen = Bytes.toShort(buf, pos, Bytes.SIZEOF_SHORT);
|
||||
pos += Bytes.SIZEOF_SHORT;
|
||||
if (rowLen < 0 || pos + rowLen > endOffset) {
|
||||
throw new IllegalArgumentException(
|
||||
"Invalid row length in KeyValue, rowLength=" + rowLen + bytesToHex(buf, offset, length));
|
||||
}
|
||||
pos += rowLen;
|
||||
// check the family
|
||||
if (pos + Bytes.SIZEOF_BYTE > endOffset) {
|
||||
throw new IllegalArgumentException("Overflow when reading family length at position=" + pos
|
||||
+ bytesToHex(buf, offset, length));
|
||||
}
|
||||
int familyLen = buf[pos];
|
||||
pos += Bytes.SIZEOF_BYTE;
|
||||
if (familyLen < 0 || pos + familyLen > endOffset) {
|
||||
throw new IllegalArgumentException("Invalid family length in KeyValue, familyLength="
|
||||
+ familyLen + bytesToHex(buf, offset, length));
|
||||
}
|
||||
pos += familyLen;
|
||||
// check the qualifier
|
||||
int qualifierLen = keyLen - Bytes.SIZEOF_SHORT - rowLen - Bytes.SIZEOF_BYTE - familyLen
|
||||
- Bytes.SIZEOF_LONG - Bytes.SIZEOF_BYTE;
|
||||
if (qualifierLen < 0 || pos + qualifierLen > endOffset) {
|
||||
throw new IllegalArgumentException("Invalid qualifier length in KeyValue, qualifierLen="
|
||||
+ qualifierLen + bytesToHex(buf, offset, length));
|
||||
}
|
||||
pos += qualifierLen;
|
||||
// check the timestamp
|
||||
if (pos + Bytes.SIZEOF_LONG > endOffset) {
|
||||
throw new IllegalArgumentException(
|
||||
"Overflow when reading timestamp at position=" + pos + bytesToHex(buf, offset, length));
|
||||
}
|
||||
long timestamp = Bytes.toLong(buf, pos, Bytes.SIZEOF_LONG);
|
||||
if (timestamp < 0) {
|
||||
throw new IllegalArgumentException(
|
||||
"Timestamp cannot be negative, ts=" + timestamp + bytesToHex(buf, offset, length));
|
||||
}
|
||||
pos += Bytes.SIZEOF_LONG;
|
||||
// check the type
|
||||
if (pos + Bytes.SIZEOF_BYTE > endOffset) {
|
||||
throw new IllegalArgumentException(
|
||||
"Overflow when reading type at position=" + pos + bytesToHex(buf, offset, length));
|
||||
}
|
||||
byte type = buf[pos];
|
||||
if (!Type.isValidType(type)) {
|
||||
throw new IllegalArgumentException(
|
||||
"Invalid type in KeyValue, type=" + type + bytesToHex(buf, offset, length));
|
||||
}
|
||||
pos += Bytes.SIZEOF_BYTE;
|
||||
// check the value
|
||||
if (pos + valLen > endOffset) {
|
||||
throw new IllegalArgumentException(
|
||||
"Overflow when reading value part at position=" + pos + bytesToHex(buf, offset, length));
|
||||
}
|
||||
pos += valLen;
|
||||
// check the tags
|
||||
if (withTags) {
|
||||
if (pos == endOffset) {
|
||||
// withTags is true but no tag in the cell.
|
||||
return;
|
||||
}
|
||||
if (pos + Bytes.SIZEOF_SHORT > endOffset) {
|
||||
throw new IllegalArgumentException("Overflow when reading tags length at position=" + pos
|
||||
+ bytesToHex(buf, offset, length));
|
||||
}
|
||||
short tagsLen = Bytes.toShort(buf, pos);
|
||||
pos += Bytes.SIZEOF_SHORT;
|
||||
if (tagsLen < 0 || pos + tagsLen > endOffset) {
|
||||
throw new IllegalArgumentException("Invalid tags length in KeyValue at position="
|
||||
+ (pos - Bytes.SIZEOF_SHORT) + bytesToHex(buf, offset, length));
|
||||
}
|
||||
int tagsEndOffset = pos + tagsLen;
|
||||
for (; pos < tagsEndOffset;) {
|
||||
if (pos + Tag.TAG_LENGTH_SIZE > endOffset) {
|
||||
throw new IllegalArgumentException("Overflow when reading tag length at position=" + pos
|
||||
+ bytesToHex(buf, offset, length));
|
||||
}
|
||||
short tagLen = Bytes.toShort(buf, pos);
|
||||
pos += Tag.TAG_LENGTH_SIZE;
|
||||
// tagLen contains one byte tag type, so must be not less than 1.
|
||||
if (tagLen < 1 || pos + tagLen > endOffset) {
|
||||
throw new IllegalArgumentException(
|
||||
"Invalid tag length at position=" + (pos - Tag.TAG_LENGTH_SIZE) + ", tagLength="
|
||||
+ tagLen + bytesToHex(buf, offset, length));
|
||||
}
|
||||
pos += tagLen;
|
||||
}
|
||||
}
|
||||
if (pos != endOffset) {
|
||||
throw new IllegalArgumentException("Some redundant bytes in KeyValue's buffer, startOffset="
|
||||
+ pos + ", endOffset=" + endOffset + bytesToHex(buf, offset, length));
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a KeyValue reading from the raw InputStream. Named
|
||||
* <code>iscreate</code> so doesn't clash with {@link #create(DataInput)}
|
||||
*
|
||||
* @param in
|
||||
* <code>createKeyValueFromInputStream</code> so doesn't clash with {@link #create(DataInput)}
|
||||
* @param in inputStream to read.
|
||||
* @param withTags whether the keyvalue should include tags are not
|
||||
* @return Created KeyValue OR if we find a length of zero, we will return
|
||||
* null which can be useful marking a stream as done.
|
||||
* @return Created KeyValue OR if we find a length of zero, we will return null which can be
|
||||
* useful marking a stream as done.
|
||||
* @throws IOException
|
||||
*/
|
||||
public static KeyValue iscreate(final InputStream in, boolean withTags) throws IOException {
|
||||
public static KeyValue createKeyValueFromInputStream(InputStream in, boolean withTags)
|
||||
throws IOException {
|
||||
byte[] intBytes = new byte[Bytes.SIZEOF_INT];
|
||||
int bytesRead = 0;
|
||||
while (bytesRead < intBytes.length) {
|
||||
|
@ -541,14 +669,10 @@ public class KeyValueUtil {
|
|||
}
|
||||
bytesRead += n;
|
||||
}
|
||||
// TODO: perhaps some sanity check is needed here.
|
||||
byte[] bytes = new byte[Bytes.toInt(intBytes)];
|
||||
IOUtils.readFully(in, bytes, 0, bytes.length);
|
||||
if (withTags) {
|
||||
return new KeyValue(bytes, 0, bytes.length);
|
||||
} else {
|
||||
return new NoTagsKeyValue(bytes, 0, bytes.length);
|
||||
}
|
||||
return withTags ? new KeyValue(bytes, 0, bytes.length)
|
||||
: new NoTagsKeyValue(bytes, 0, bytes.length);
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
|
@ -72,7 +72,7 @@ public class KeyValueCodec implements Codec {
|
|||
@Override
|
||||
protected Cell parseCell() throws IOException {
|
||||
// No tags here
|
||||
return KeyValueUtil.iscreate(in, false);
|
||||
return KeyValueUtil.createKeyValueFromInputStream(in, false);
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -114,7 +114,6 @@ public class KeyValueCodec implements Codec {
|
|||
// We know there is not going to be any tags.
|
||||
return new NoTagsByteBufferKeyValue(bb, pos, len);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
|
@ -78,7 +78,7 @@ public class KeyValueCodecWithTags implements Codec {
|
|||
@Override
|
||||
protected Cell parseCell() throws IOException {
|
||||
// create KeyValue with tags
|
||||
return KeyValueUtil.iscreate(in, true);
|
||||
return KeyValueUtil.createKeyValueFromInputStream(in, true);
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
@ -18,11 +18,13 @@
|
|||
*/
|
||||
package org.apache.hadoop.hbase;
|
||||
|
||||
import static org.junit.Assert.assertArrayEquals;
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertFalse;
|
||||
import static org.junit.Assert.assertNotEquals;
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
import static org.junit.Assert.fail;
|
||||
|
||||
import java.io.ByteArrayInputStream;
|
||||
import java.io.ByteArrayOutputStream;
|
||||
|
@ -590,181 +592,152 @@ public class TestKeyValue {
|
|||
|
||||
@Test
|
||||
public void testKeyValueSerialization() throws Exception {
|
||||
KeyValue kvA1 = new KeyValue(Bytes.toBytes("key"), Bytes.toBytes("cf"), Bytes.toBytes("qualA"),
|
||||
Bytes.toBytes("1"));
|
||||
KeyValue kvA2 = new KeyValue(Bytes.toBytes("key"), Bytes.toBytes("cf"), Bytes.toBytes("qualA"),
|
||||
Bytes.toBytes("2"));
|
||||
MockKeyValue mkvA1 = new MockKeyValue(kvA1);
|
||||
MockKeyValue mkvA2 = new MockKeyValue(kvA2);
|
||||
KeyValue[] keyValues = new KeyValue[] {
|
||||
new KeyValue(Bytes.toBytes("key"), Bytes.toBytes("cf"), Bytes.toBytes("qualA"),
|
||||
Bytes.toBytes("1")),
|
||||
new KeyValue(Bytes.toBytes("key"), Bytes.toBytes("cf"), Bytes.toBytes("qualA"),
|
||||
Bytes.toBytes("2")),
|
||||
new KeyValue(Bytes.toBytes("key"), Bytes.toBytes("cf"), Bytes.toBytes("qualA"),
|
||||
System.currentTimeMillis(), Bytes.toBytes("2"),
|
||||
new Tag[] { new ArrayBackedTag((byte) 120, "tagA"),
|
||||
new ArrayBackedTag((byte) 121, Bytes.toBytes("tagB")) }),
|
||||
new KeyValue(Bytes.toBytes("key"), Bytes.toBytes("cf"), Bytes.toBytes("qualA"),
|
||||
System.currentTimeMillis(), Bytes.toBytes("2"),
|
||||
new Tag[] { new ArrayBackedTag((byte) 0, "tagA") }),
|
||||
new KeyValue(Bytes.toBytes("key"), Bytes.toBytes("cf"), Bytes.toBytes(""),
|
||||
Bytes.toBytes("1")) };
|
||||
ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
|
||||
for (KeyValue kv : keyValues) {
|
||||
DataOutputStream os = new DataOutputStream(byteArrayOutputStream);
|
||||
ByteBufferUtils.putInt(os, KeyValueUtil.getSerializedSize(mkvA1, true));
|
||||
KeyValueUtil.oswrite(mkvA1, os, true);
|
||||
ByteBufferUtils.putInt(os, KeyValueUtil.getSerializedSize(mkvA2, true));
|
||||
KeyValueUtil.oswrite(mkvA2, os, true);
|
||||
DataInputStream is = new DataInputStream(new ByteArrayInputStream(
|
||||
byteArrayOutputStream.toByteArray()));
|
||||
KeyValue deSerKV1 = KeyValueUtil.iscreate(is, true);
|
||||
assertTrue(kvA1.equals(deSerKV1));
|
||||
KeyValue deSerKV2 = KeyValueUtil.iscreate(is, true);
|
||||
assertTrue(kvA2.equals(deSerKV2));
|
||||
ByteBufferUtils.putInt(os, KeyValueUtil.getSerializedSize(kv, true));
|
||||
KeyValueUtil.oswrite(kv, os, true);
|
||||
}
|
||||
DataInputStream is =
|
||||
new DataInputStream(new ByteArrayInputStream(byteArrayOutputStream.toByteArray()));
|
||||
for (int i = 0; i < keyValues.length; i++) {
|
||||
LOG.info("Case#" + i + ": deserialize the kv: " + keyValues[i]);
|
||||
KeyValue destKv = KeyValueUtil.createKeyValueFromInputStream(is, true);
|
||||
assertEquals(keyValues[i], destKv);
|
||||
assertArrayEquals(CellUtil.cloneValue(keyValues[i]), CellUtil.cloneValue(destKv));
|
||||
assertArrayEquals(PrivateCellUtil.cloneTags(keyValues[i]), PrivateCellUtil.cloneTags(destKv));
|
||||
}
|
||||
}
|
||||
|
||||
private static class MockKeyValue implements Cell {
|
||||
private final KeyValue kv;
|
||||
private static class FailureCase {
|
||||
byte[] buf;
|
||||
int offset;
|
||||
int length;
|
||||
boolean withTags;
|
||||
String expectedMessage;
|
||||
|
||||
public MockKeyValue(KeyValue kv) {
|
||||
this.kv = kv;
|
||||
public FailureCase(byte[] buf, int offset, int length, boolean withTags,
|
||||
String expectedMessage) {
|
||||
this.buf = buf;
|
||||
this.offset = offset;
|
||||
this.length = length;
|
||||
this.withTags = withTags;
|
||||
this.expectedMessage = expectedMessage;
|
||||
}
|
||||
|
||||
/**
|
||||
* This returns the offset where the tag actually starts.
|
||||
*/
|
||||
@Override
|
||||
public int getTagsOffset() {
|
||||
return this.kv.getTagsOffset();
|
||||
public String toString() {
|
||||
return "FailureCaseDetails: [buf=" + Bytes.toStringBinary(buf, offset, length) + ", offset="
|
||||
+ offset + ", " + "length=" + length + ", expectedMessage=" + expectedMessage
|
||||
+ ", withtags=" + withTags + "]";
|
||||
}
|
||||
|
||||
/**
|
||||
* used to achieve atomic operations in the memstore.
|
||||
*/
|
||||
@Override
|
||||
public long getSequenceId() {
|
||||
return this.kv.getSequenceId();
|
||||
public String getExpectedMessage() {
|
||||
return this.expectedMessage + KeyValueUtil.bytesToHex(buf, offset, length);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* This returns the total length of the tag bytes
|
||||
*/
|
||||
@Override
|
||||
public int getTagsLength() {
|
||||
return this.kv.getTagsLength();
|
||||
@Test
|
||||
public void testCheckKeyValueBytesFailureCase() throws Exception {
|
||||
byte[][] inputs = new byte[][] { HConstants.EMPTY_BYTE_ARRAY, // case.0
|
||||
Bytes.toBytesBinary("a"), // case.1
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x01"), // case.2
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x01\\x00"), // case.3
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x01\\x00\\x00\\x00\\x01"), // case.4
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x01\\x00\\x00\\x00\\x01\\x00"), // case.5
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x01\\x00\\x00\\x00\\x01\\x00\\x01"), // case.6
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x01\\x00\\x00\\x00\\x01\\x00\\x03ROW"), // case.7
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x01\\x00\\x00\\x00\\x01\\x00\\x03ROW\\x01"), // case.8
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x11\\x00\\x00\\x00\\x01\\x00\\x03ROW\\x01FQ\\xFF"
|
||||
+ "\\xFF\\xFF\\xFF\\xFF\\xFF\\xFF\\xFF\\x03"), // case.9
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x11\\x00\\x00\\x00\\x01\\x00\\x03ROW\\x01FQ\\x00"
|
||||
+ "\\x00\\x00\\x00\\x00\\x00\\x00\\x01\\x03"), // case.10
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x11\\x00\\x00\\x00\\x01\\x00\\x03ROW\\x01FQ\\x00"
|
||||
+ "\\x00\\x00\\x00\\x00\\x00\\x00\\x01\\x04"), // case.11
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x11\\x00\\x00\\x00\\x01\\x00\\x03ROW\\x01FQ\\x00"
|
||||
+ "\\x00\\x00\\x00\\x00\\x00\\x00\\x01\\x04VALUE"), // case.12
|
||||
};
|
||||
String[] outputs = new String[] { "Overflow when reading key length at position=0",
|
||||
"Overflow when reading key length at position=0",
|
||||
"Invalid key length in KeyValue. keyLength=1",
|
||||
"Overflow when reading value length at position=4",
|
||||
"Invalid value length in KeyValue, valueLength=1",
|
||||
"Overflow when reading row length at position=8",
|
||||
"Invalid row length in KeyValue, rowLength=1",
|
||||
"Overflow when reading family length at position=13",
|
||||
"Invalid family length in KeyValue, familyLength=1", "Timestamp cannot be negative, ts=-1",
|
||||
"Invalid type in KeyValue, type=3", "Overflow when reading value part at position=25",
|
||||
"Invalid tags length in KeyValue at position=26", };
|
||||
byte[][] withTagsInputs = new byte[][] {
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x11\\x00\\x00\\x00\\x01\\x00\\x03ROW\\x01FQ\\x00"
|
||||
+ "\\x00\\x00\\x00\\x00\\x00\\x00\\x01\\x04V\\x01"), // case.13
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x11\\x00\\x00\\x00\\x01\\x00\\x03ROW\\x01FQ\\x00"
|
||||
+ "\\x00\\x00\\x00\\x00\\x00\\x00\\x01\\x04V\\x00\\x01"), // case.14
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x11\\x00\\x00\\x00\\x01\\x00\\x03ROW\\x01FQ\\x00"
|
||||
+ "\\x00\\x00\\x00\\x00\\x00\\x00\\x01\\x04V\\x00\\x04\\x00\\x03\\x00A"), // case.15
|
||||
// case.16
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x11\\x00\\x00\\x00\\x01\\x00\\x03ROW\\x01FQ\\x00"
|
||||
+ "\\x00\\x00\\x00\\x00\\x00\\x00\\x01\\x04V\\x00\\x0A\\x00\\x04\\x00TAG\\x00\\x04"
|
||||
+ "\\xFFT"),
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x11\\x00\\x00\\x00\\x01\\x00\\x03ROW\\x01FQ\\x00"
|
||||
+ "\\x00\\x00\\x00\\x00\\x00\\x00\\x01\\x04V\\x00\\x0C\\x00\\x04\\x00TAG\\x00\\x05"
|
||||
+ "\\xF0COME\\x00"), // case.17
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x11\\x00\\x00\\x00\\x01\\x00\\x03ROW\\x01FQ\\x00"
|
||||
+ "\\x00\\x00\\x00\\x00\\x00\\x00\\x01\\x04V\\x00\\x0C\\x00\\x04\\x00TAG\\x00\\x05"
|
||||
+ "\\xF0COME"), // case.18
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x11\\x00\\x00\\x00\\x01\\x00\\x03ROW\\x01FQ\\x00"
|
||||
+ "\\x00\\x00\\x00\\x00\\x00\\x00\\x01\\x04V\\x00\\x00"), // case.19
|
||||
Bytes.toBytesBinary("\\x00\\x00\\x00\\x11\\x00\\x00\\x00\\x01\\x00\\x03ROW\\x01FQ\\x00"
|
||||
+ "\\x00\\x00\\x00\\x00\\x00\\x00\\x01\\x04V\\x00\\x1B\\x00\\x05\\x01TAG1\\x00\\x05"
|
||||
+ "\\x02TAG2\\x00\\x05\\x03TAG3\\x00\\x05\\x04TAG4"), // case.20
|
||||
};
|
||||
String[] withTagsOutputs = new String[] { "Overflow when reading tags length at position=26",
|
||||
"Invalid tags length in KeyValue at position=26",
|
||||
"Invalid tag length at position=28, tagLength=3",
|
||||
"Invalid tag length at position=34, tagLength=4",
|
||||
"Some redundant bytes in KeyValue's buffer, startOffset=41, endOffset=42", null, null,
|
||||
null, };
|
||||
assertEquals(inputs.length, outputs.length);
|
||||
assertEquals(withTagsInputs.length, withTagsOutputs.length);
|
||||
|
||||
FailureCase[] cases = new FailureCase[inputs.length + withTagsInputs.length];
|
||||
for (int i = 0; i < inputs.length; i++) {
|
||||
cases[i] = new FailureCase(inputs[i], 0, inputs[i].length, false, outputs[i]);
|
||||
}
|
||||
for (int i = 0; i < withTagsInputs.length; i++) {
|
||||
cases[inputs.length + i] =
|
||||
new FailureCase(withTagsInputs[i], 0, withTagsInputs[i].length, true, withTagsOutputs[i]);
|
||||
}
|
||||
|
||||
/**
|
||||
*
|
||||
* @return Timestamp
|
||||
*/
|
||||
@Override
|
||||
public long getTimestamp() {
|
||||
return this.kv.getTimestamp();
|
||||
for (int i = 0; i < cases.length; i++) {
|
||||
FailureCase c = cases[i];
|
||||
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
||||
DataOutputStream os = new DataOutputStream(baos);
|
||||
ByteBufferUtils.putInt(os, c.length);
|
||||
os.write(c.buf, c.offset, c.length);
|
||||
try {
|
||||
KeyValueUtil.createKeyValueFromInputStream(
|
||||
new DataInputStream(new ByteArrayInputStream(baos.toByteArray())), c.withTags);
|
||||
if (c.expectedMessage != null) {
|
||||
fail("Should fail when parse key value from an invalid bytes for case#" + i + ". " + c);
|
||||
}
|
||||
} catch (IllegalArgumentException e) {
|
||||
assertEquals("Case#" + i + " failed," + c, c.getExpectedMessage(), e.getMessage());
|
||||
}
|
||||
|
||||
/**
|
||||
* @return KeyValue.TYPE byte representation
|
||||
*/
|
||||
@Override
|
||||
public byte getTypeByte() {
|
||||
return this.kv.getTypeByte();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the backing array of the entire KeyValue (all KeyValue fields are
|
||||
* in a single array)
|
||||
*/
|
||||
@Override
|
||||
public byte[] getValueArray() {
|
||||
return this.kv.getValueArray();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the value offset
|
||||
*/
|
||||
@Override
|
||||
public int getValueOffset() {
|
||||
return this.kv.getValueOffset();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return Value length
|
||||
*/
|
||||
@Override
|
||||
public int getValueLength() {
|
||||
return this.kv.getValueLength();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the backing array of the entire KeyValue (all KeyValue fields are
|
||||
* in a single array)
|
||||
*/
|
||||
@Override
|
||||
public byte[] getRowArray() {
|
||||
return this.kv.getRowArray();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return Row offset
|
||||
*/
|
||||
@Override
|
||||
public int getRowOffset() {
|
||||
return this.kv.getRowOffset();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return Row length
|
||||
*/
|
||||
@Override
|
||||
public short getRowLength() {
|
||||
return this.kv.getRowLength();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the backing array of the entire KeyValue (all KeyValue fields are
|
||||
* in a single array)
|
||||
*/
|
||||
@Override
|
||||
public byte[] getFamilyArray() {
|
||||
return this.kv.getFamilyArray();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return Family offset
|
||||
*/
|
||||
@Override
|
||||
public int getFamilyOffset() {
|
||||
return this.kv.getFamilyOffset();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return Family length
|
||||
*/
|
||||
@Override
|
||||
public byte getFamilyLength() {
|
||||
return this.kv.getFamilyLength();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the backing array of the entire KeyValue (all KeyValue fields are
|
||||
* in a single array)
|
||||
*/
|
||||
@Override
|
||||
public byte[] getQualifierArray() {
|
||||
return this.kv.getQualifierArray();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return Qualifier offset
|
||||
*/
|
||||
@Override
|
||||
public int getQualifierOffset() {
|
||||
return this.kv.getQualifierOffset();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return Qualifier length
|
||||
*/
|
||||
@Override
|
||||
public int getQualifierLength() {
|
||||
return this.kv.getQualifierLength();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return the backing array of the entire KeyValue (all KeyValue fields are
|
||||
* in a single array)
|
||||
*/
|
||||
@Override
|
||||
public byte[] getTagsArray() {
|
||||
return this.kv.getTagsArray();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1710,7 +1710,6 @@ public class HStore implements Store, HeapSize, StoreConfigInformation, Propagat
|
|||
|
||||
@Override
|
||||
public boolean hasReferences() {
|
||||
List<HStoreFile> reloadedStoreFiles = null;
|
||||
// Grab the read lock here, because we need to ensure that: only when the atomic
|
||||
// replaceStoreFiles(..) finished, we can get all the complete store file list.
|
||||
this.lock.readLock().lock();
|
||||
|
|
|
@ -332,7 +332,7 @@ public class TestDataBlockEncoders {
|
|||
byte[] family = new byte[0];
|
||||
byte[] qualifier = new byte[0];
|
||||
byte[] value = new byte[0];
|
||||
KeyValue expectedKV = new KeyValue(row, family, qualifier, -1L, Type.Put, value);
|
||||
KeyValue expectedKV = new KeyValue(row, family, qualifier, 1L, Type.Put, value);
|
||||
kvList.add(expectedKV);
|
||||
DataBlockEncoding encoding = DataBlockEncoding.ROW_INDEX_V1;
|
||||
DataBlockEncoder encoder = encoding.getEncoder();
|
||||
|
|
|
@ -392,11 +392,11 @@ public class TestCacheOnWrite {
|
|||
tags[0] = t;
|
||||
kv =
|
||||
new KeyValue(row, 0, row.length, cf, 0, cf.length, qualifier, 0, qualifier.length,
|
||||
rand.nextLong(), generateKeyType(rand), value, 0, value.length, tagList);
|
||||
Math.abs(rand.nextLong()), generateKeyType(rand), value, 0, value.length, tagList);
|
||||
} else {
|
||||
kv =
|
||||
new KeyValue(row, 0, row.length, cf, 0, cf.length, qualifier, 0, qualifier.length,
|
||||
rand.nextLong(), generateKeyType(rand), value, 0, value.length);
|
||||
Math.abs(rand.nextLong()), generateKeyType(rand), value, 0, value.length);
|
||||
}
|
||||
sfw.append(kv);
|
||||
}
|
||||
|
|
|
@ -32,7 +32,6 @@ import org.apache.hadoop.hbase.CellComparatorImpl;
|
|||
import org.apache.hadoop.hbase.HBaseClassTestRule;
|
||||
import org.apache.hadoop.hbase.HBaseTestingUtility;
|
||||
import org.apache.hadoop.hbase.KeyValue;
|
||||
import org.apache.hadoop.hbase.KeyValueUtil;
|
||||
import org.apache.hadoop.hbase.io.hfile.HFile.Reader;
|
||||
import org.apache.hadoop.hbase.io.hfile.HFile.Writer;
|
||||
import org.apache.hadoop.hbase.testclassification.IOTests;
|
||||
|
@ -202,13 +201,13 @@ public class TestHFileSeek extends TestCase {
|
|||
kSampler.next(key);
|
||||
byte[] k = new byte[key.getLength()];
|
||||
System.arraycopy(key.getBytes(), 0, k, 0, key.getLength());
|
||||
if (scanner.seekTo(KeyValueUtil.createKeyValueFromKey(k)) >= 0) {
|
||||
KeyValue kv = new KeyValue(k, CF, QUAL);
|
||||
if (scanner.seekTo(kv) >= 0) {
|
||||
ByteBuffer bbkey = ByteBuffer.wrap(((KeyValue) scanner.getKey()).getKey());
|
||||
ByteBuffer bbval = scanner.getValue();
|
||||
totalBytes += bbkey.limit();
|
||||
totalBytes += bbval.limit();
|
||||
}
|
||||
else {
|
||||
} else {
|
||||
++miss;
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue