Merge pull request #38 from cloudstruct/feature/server-handshake

Signal that the handshake is complete when acting as server

Author: agaffney

cmd/go-ouroboros-network/server.go

@@ -79,5 +79,7 @@ func testServer(f *globalFlags) {
 		if err != nil {
 			fmt.Printf("ERROR: %s\n", err)
 		}
+		fmt.Printf("handshake completed...disconnecting\n")
+		conn.Close()
 	}
 }

protocol/handshake/handshake.go

@@ -121,11 +121,16 @@ func (h *Handshake) handleProposeVersions(msgGeneric protocol.Message) error {
 	}
 	if highestVersion > 0 {
 		resp := NewMsgAcceptVersion(highestVersion, versionData)
-		return h.proto.SendMessage(resp, true)
+		if err := h.proto.SendMessage(resp, true); err != nil {
+			return err
+		}
+		h.Version = highestVersion
+		h.Finished <- true
+		return nil
 	} else {
 		// TODO: handle failures
 		// https://github.com/cloudstruct/go-ouroboros-network/issues/32
-		return nil
+		return fmt.Errorf("handshake failed, but we don't yet support this")
 	}
 }