Move fuzzing code to seperate file
This commit is contained in:
parent
f204a14e00
commit
3f51dd25c7
16
irc.go
16
irc.go
@ -86,18 +86,6 @@ func (irc *Connection) readLoop() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// +build gofuzz
|
|
||||||
func Fuzz(data []byte) int {
|
|
||||||
b := bytes.NewBuffer(data)
|
|
||||||
event, err := parseToEvent(b.String())
|
|
||||||
if err == nil {
|
|
||||||
irc := IRC("go-eventirc", "go-eventirc")
|
|
||||||
irc.RunCallbacks(event)
|
|
||||||
return 1
|
|
||||||
}
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
//Parse raw irc messages
|
//Parse raw irc messages
|
||||||
func parseToEvent(msg string) (*Event, error) {
|
func parseToEvent(msg string) (*Event, error) {
|
||||||
msg = strings.TrimSuffix(msg, "\n") //Remove \r\n
|
msg = strings.TrimSuffix(msg, "\n") //Remove \r\n
|
||||||
@ -362,7 +350,9 @@ func (irc *Connection) Disconnect() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
irc.Wait()
|
irc.Wait()
|
||||||
irc.socket.Close()
|
if irc.socket != nil {
|
||||||
|
irc.socket.Close()
|
||||||
|
}
|
||||||
irc.socket = nil
|
irc.socket = nil
|
||||||
irc.ErrorChan() <- ErrDisconnected
|
irc.ErrorChan() <- ErrDisconnected
|
||||||
}
|
}
|
||||||
|
14
irc_test_fuzz.go
Normal file
14
irc_test_fuzz.go
Normal file
@ -0,0 +1,14 @@
|
|||||||
|
// +build gofuzz
|
||||||
|
|
||||||
|
package irc
|
||||||
|
|
||||||
|
func Fuzz(data []byte) int {
|
||||||
|
b := bytes.NewBuffer(data)
|
||||||
|
event, err := parseToEvent(b.String())
|
||||||
|
if err == nil {
|
||||||
|
irc := IRC("go-eventirc", "go-eventirc")
|
||||||
|
irc.RunCallbacks(event)
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
return 0
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user