github.com/icodeface/tls@v0.0.0-20230910023335-34df9250cd12/internal/trace/writer.go (about)

     1  package trace
     2  
     3  import "bytes"
     4  
     5  // Writer is a test trace writer.
     6  type Writer struct {
     7  	bytes.Buffer
     8  }
     9  
    10  func NewWriter() *Writer {
    11  	w := new(Writer)
    12  	w.Write([]byte("go 1.9 trace\x00\x00\x00\x00"))
    13  	return w
    14  }
    15  
    16  // Emit writes an event record to the trace.
    17  // See Event types for valid types and required arguments.
    18  func (w *Writer) Emit(typ byte, args ...uint64) {
    19  	nargs := byte(len(args)) - 1
    20  	if nargs > 3 {
    21  		nargs = 3
    22  	}
    23  	buf := []byte{typ | nargs<<6}
    24  	if nargs == 3 {
    25  		buf = append(buf, 0)
    26  	}
    27  	for _, a := range args {
    28  		buf = appendVarint(buf, a)
    29  	}
    30  	if nargs == 3 {
    31  		buf[1] = byte(len(buf) - 2)
    32  	}
    33  	n, err := w.Write(buf)
    34  	if n != len(buf) || err != nil {
    35  		panic("failed to write")
    36  	}
    37  }
    38  
    39  func appendVarint(buf []byte, v uint64) []byte {
    40  	for ; v >= 0x80; v >>= 7 {
    41  		buf = append(buf, 0x80|byte(v))
    42  	}
    43  	buf = append(buf, byte(v))
    44  	return buf
    45  }