github.com/gocuntian/go@v0.0.0-20160610041250-fee02d270bf8/src/internal/trace/parser_test.go (about)

     1  // Copyright 2015 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package trace
     6  
     7  import (
     8  	"bytes"
     9  	"io/ioutil"
    10  	"path/filepath"
    11  	"strings"
    12  	"testing"
    13  )
    14  
    15  func TestCorruptedInputs(t *testing.T) {
    16  	// These inputs crashed parser previously.
    17  	tests := []string{
    18  		"gotrace\x00\x020",
    19  		"gotrace\x00Q00\x020",
    20  		"gotrace\x00T00\x020",
    21  		"gotrace\x00\xc3\x0200",
    22  		"go 1.5 trace\x00\x00\x00\x00\x020",
    23  		"go 1.5 trace\x00\x00\x00\x00Q00\x020",
    24  		"go 1.5 trace\x00\x00\x00\x00T00\x020",
    25  		"go 1.5 trace\x00\x00\x00\x00\xc3\x0200",
    26  	}
    27  	for _, data := range tests {
    28  		events, err := Parse(strings.NewReader(data), "")
    29  		if err == nil || events != nil {
    30  			t.Fatalf("no error on input: %q", data)
    31  		}
    32  	}
    33  }
    34  
    35  func TestParseCanned(t *testing.T) {
    36  	files, err := ioutil.ReadDir("./testdata")
    37  	if err != nil {
    38  		t.Fatalf("failed to read ./testdata: %v", err)
    39  	}
    40  	for _, f := range files {
    41  		data, err := ioutil.ReadFile(filepath.Join("./testdata", f.Name()))
    42  		if err != nil {
    43  			t.Fatalf("failed to read input file: %v", err)
    44  		}
    45  		_, err = Parse(bytes.NewReader(data), "")
    46  		switch {
    47  		case strings.HasSuffix(f.Name(), "_good"):
    48  			if err != nil {
    49  				t.Errorf("failed to parse good trace %v: %v", f.Name(), err)
    50  			}
    51  		case strings.HasSuffix(f.Name(), "_unordered"):
    52  			if err != ErrTimeOrder {
    53  				t.Errorf("unordered trace is not detected %v: %v", f.Name(), err)
    54  			}
    55  		default:
    56  			t.Errorf("unknown input file suffix: %v", f.Name())
    57  		}
    58  	}
    59  }
    60  
    61  func TestParseVersion(t *testing.T) {
    62  	tests := map[string]int{
    63  		"go 1.5 trace\x00\x00\x00\x00": 1005,
    64  		"go 1.7 trace\x00\x00\x00\x00": 1007,
    65  		"go 1.10 trace\x00\x00\x00":    1010,
    66  		"go 1.25 trace\x00\x00\x00":    1025,
    67  		"go 1.234 trace\x00\x00":       1234,
    68  		"go 1.2345 trace\x00":          -1,
    69  		"go 0.0 trace\x00\x00\x00\x00": -1,
    70  		"go a.b trace\x00\x00\x00\x00": -1,
    71  	}
    72  	for header, ver := range tests {
    73  		ver1, err := parseHeader([]byte(header))
    74  		if ver == -1 {
    75  			if err == nil {
    76  				t.Fatalf("no error on input: %q, version %v", header, ver1)
    77  			}
    78  		} else {
    79  			if err != nil {
    80  				t.Fatalf("failed to parse: %q (%v)", header, err)
    81  			}
    82  			if ver != ver1 {
    83  				t.Fatalf("wrong version: %v, want %v, input: %q", ver1, ver, header)
    84  			}
    85  		}
    86  	}
    87  }
    88  
    89  func TestTimestampOverflow(t *testing.T) {
    90  	// Test that parser correctly handles large timestamps (long tracing).
    91  	w := newWriter()
    92  	w.emit(EvBatch, 0, 0)
    93  	w.emit(EvFrequency, 1e9)
    94  	for ts := uint64(1); ts < 1e16; ts *= 2 {
    95  		w.emit(EvGoCreate, ts, ts, 0, 0)
    96  	}
    97  	if _, err := Parse(w, ""); err != nil {
    98  		t.Fatalf("failed to parse: %v", err)
    99  	}
   100  }
   101  
   102  type writer struct {
   103  	bytes.Buffer
   104  }
   105  
   106  func newWriter() *writer {
   107  	w := new(writer)
   108  	w.Write([]byte("go 1.7 trace\x00\x00\x00\x00"))
   109  	return w
   110  }
   111  
   112  func (w *writer) emit(typ byte, args ...uint64) {
   113  	nargs := byte(len(args)) - 1
   114  	if nargs > 3 {
   115  		nargs = 3
   116  	}
   117  	buf := []byte{typ | nargs<<6}
   118  	if nargs == 3 {
   119  		buf = append(buf, 0)
   120  	}
   121  	for _, a := range args {
   122  		buf = appendVarint(buf, a)
   123  	}
   124  	if nargs == 3 {
   125  		buf[1] = byte(len(buf) - 2)
   126  	}
   127  	n, err := w.Write(buf)
   128  	if n != len(buf) || err != nil {
   129  		panic("failed to write")
   130  	}
   131  }
   132  
   133  func appendVarint(buf []byte, v uint64) []byte {
   134  	for ; v >= 0x80; v >>= 7 {
   135  		buf = append(buf, 0x80|byte(v))
   136  	}
   137  	buf = append(buf, byte(v))
   138  	return buf
   139  }