1 // Protocol Buffers for Go with Gadgets 2 // 3 // Copyright (c) 2019, The GoGo Authors. All rights reserved. 4 // http://github.com/gogo/protobuf 5 // 6 // Redistribution and use in source and binary forms, with or without 7 // modification, are permitted provided that the following conditions are 8 // met: 9 // 10 // * Redistributions of source code must retain the above copyright 11 // notice, this list of conditions and the following disclaimer. 12 // * Redistributions in binary form must reproduce the above 13 // copyright notice, this list of conditions and the following disclaimer 14 // in the documentation and/or other materials provided with the 15 // distribution. 16 // 17 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 20 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 21 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 22 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 23 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 27 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28 29 package issue503 30 31 import ( 32 "fmt" 33 "github.com/gogo/protobuf/proto" 34 "math/rand" 35 "testing" 36 ) 37 38 // Original test case submitted by issue creator 39 func TestOOMIssue(t *testing.T) { 40 // all repeated filed total has 10+10 elements 41 a := &Foo{ 42 Num1: []int64{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}, 43 Num2: []int32{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}, 44 } 45 dat, err := a.Marshal() 46 if err != nil { 47 t.Fatal(err) 48 } 49 dst := &Foo{} 50 if err := dst.Unmarshal(dat); err != nil { 51 t.Fatal(err) 52 } 53 if !a.Equal(dst) { 54 t.Fatal("message is not equals") 55 } 56 fmt.Printf("%v\n", dst.String()) 57 } 58 59 func BenchmarkOOMIssue(b *testing.B) { 60 popr := rand.New(rand.NewSource(616)) 61 total := 0 62 pops := make([]*Foo, 10000) 63 for i := 0; i < 10000; i++ { 64 pops[i] = NewPopulatedFoo(popr, false) 65 } 66 b.ResetTimer() 67 for i := 0; i < b.N; i++ { 68 dAtA, err := proto.Marshal(pops[i%10000]) 69 if err != nil { 70 panic(err) 71 } 72 total += len(dAtA) 73 } 74 b.SetBytes(int64(total / b.N)) 75 } 76