func (l *Listener) handshake(conn *blend.Conn) (sourceAddr *Addr, err error) {
if conn == nil {
return nil, errors.NewError("listener off")
}
defer conn.Close()
//
var msg interface{}
msg, err = conn.Read()
if err != nil {
return nil, err
}
defer func() {
if err != nil {
conn.Write(&RejectMsg{err})
} else {
err = conn.Write(&WelcomeMsg{})
}
}()
hello, ok := msg.(*HelloMsg)
if !ok {
log.Println("rejecting", conn.RemoteAddr().String(), "unknown hello message type")
return nil, errors.NewError("rejecting unknown hello type")
}
// Accept user connections
da, ok := hello.SourceAddr.(*Addr)
if !ok {
log.Println("rejecting", conn.RemoteAddr().String(), "unknown source address type")
return nil, errors.NewError("rejecting unknown source address type")
}
la, ok := hello.TargetAddr.(*Addr)
if !ok {
log.Println("rejecting ", conn.RemoteAddr().String(), "unknown target address type")
return nil, errors.NewError("rejecting unknown target address type")
}
if la.WorkerID() != l.addr.WorkerID() {
log.Println("rejecting", conn.RemoteAddr().String(), "due to worker identity mismatch")
return nil, errors.NewError("rejecting worker identity mismatch, looks for %s, got %s", la.WorkerID(), l.addr.WorkerID())
}
if la.PID != os.Getpid() {
log.Println("rejecting", conn.RemoteAddr().String(), "due to worker PID mismatch")
return nil, errors.NewError("rejecting worker PID mismatch, looks for %d, got %d", la.PID, os.Getpid())
}
return da, nil
}
