1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
/*
* Content before git sha 34fdeebefcbf183ed7f916f931aa0586fdaa1b40
* Copyright (c) 2016, The Gocql authors,
* provided under the BSD-3-Clause License.
* See the NOTICE file distributed with this work for additional information.
*/
package gocql
import (
"bytes"
"os"
"testing"
)
func TestFuzzBugs(t *testing.T) {
// these inputs are found using go-fuzz (https://github.com/dvyukov/go-fuzz)
// and should cause a panic unless fixed.
tests := [][]byte{
[]byte("00000\xa0000"),
[]byte("\x8000\x0e\x00\x00\x00\x000"),
[]byte("\x8000\x00\x00\x00\x00\t0000000000"),
[]byte("\xa0\xff\x01\xae\xefqE\xf2\x1a"),
[]byte("\x8200\b\x00\x00\x00c\x00\x00\x00\x02000\x01\x00\x00\x00\x03" +
"\x00\n0000000000\x00\x14000000" +
"00000000000000\x00\x020000" +
"\x00\a000000000\x00\x050000000" +
"\xff0000000000000000000" +
"0000000"),
[]byte("\x82\xe600\x00\x00\x00\x000"),
[]byte("\x8200\b\x00\x00\x00\b0\x00\x00\x00\x040000"),
[]byte("\x8200\x00\x00\x00\x00\x100\x00\x00\x12\x00\x00\x0000000" +
"00000"),
[]byte("\x83000\b\x00\x00\x00\x14\x00\x00\x00\x020000000" +
"000000000"),
[]byte("\x83000\b\x00\x00\x000\x00\x00\x00\x04\x00\x1000000" +
"00000000000000e00000" +
"000\x800000000000000000" +
"0000000000000"),
}
for i, test := range tests {
t.Logf("test %d input: %q", i, test)
r := bytes.NewReader(test)
head, err := readHeader(r, make([]byte, 9))
if err != nil {
continue
}
framer := newFramer(nil, byte(head.version))
err = framer.readFrame(r, &head)
if err != nil {
continue
}
frame, err := framer.parseFrame()
if err != nil {
continue
}
t.Errorf("(%d) expected to fail for input % X", i, test)
t.Errorf("(%d) frame=%+#v", i, frame)
}
}
func TestFrameWriteTooLong(t *testing.T) {
if os.Getenv("TRAVIS") == "true" {
t.Skip("skipping test in travis due to memory pressure with the race detecor")
}
framer := newFramer(nil, 2)
framer.writeHeader(0, opStartup, 1)
framer.writeBytes(make([]byte, maxFrameSize+1))
err := framer.finish()
if err != ErrFrameTooBig {
t.Fatalf("expected to get %v got %v", ErrFrameTooBig, err)
}
}
func TestFrameReadTooLong(t *testing.T) {
if os.Getenv("TRAVIS") == "true" {
t.Skip("skipping test in travis due to memory pressure with the race detecor")
}
r := &bytes.Buffer{}
r.Write(make([]byte, maxFrameSize+1))
// write a new header right after this frame to verify that we can read it
r.Write([]byte{0x02, 0x00, 0x00, byte(opReady), 0x00, 0x00, 0x00, 0x00})
framer := newFramer(nil, 2)
head := frameHeader{
version: 2,
op: opReady,
length: r.Len() - 8,
}
err := framer.readFrame(r, &head)
if err != ErrFrameTooBig {
t.Fatalf("expected to get %v got %v", ErrFrameTooBig, err)
}
head, err = readHeader(r, make([]byte, 8))
if err != nil {
t.Fatal(err)
}
if head.op != opReady {
t.Fatalf("expected to get header %v got %v", opReady, head.op)
}
}
|
