Added a registry to Barak.

cmd/barak/main.go

@@ -19,6 +19,7 @@ import (
 	"github.com/tendermint/tendermint/binary"
 	. "github.com/tendermint/tendermint/cmd/barak/types"
 	. "github.com/tendermint/tendermint/common"
+	"github.com/tendermint/tendermint/p2p"
 	pcm "github.com/tendermint/tendermint/process"
 	"github.com/tendermint/tendermint/rpc"
 )
@@ -33,6 +34,7 @@ type Options struct {
 	Validators    []Validator
 	ListenAddress string
 	StartNonce    uint64
+	Registries    []string
 }
 
 // Global instance
@@ -43,6 +45,7 @@ var barak = struct {
 	processes  map[string]*pcm.Process
 	validators []Validator
 	rootDir    string
+	registries []string
 }{
 	mtx:        sync.Mutex{},
 	pid:        os.Getpid(),
@@ -50,6 +53,7 @@ var barak = struct {
 	processes:  make(map[string]*pcm.Process),
 	validators: nil,
 	rootDir:    "",
+	registries: nil,
 }
 
 func main() {
@@ -95,11 +99,27 @@ func main() {
 	fmt.Printf("Barak: %v\n", barak)
 
 	// Start rpc server.
+	listener := p2p.NewDefaultListener("tcp", options.ListenAddress, false)
 	mux := http.NewServeMux()
 	mux.HandleFunc("/download", ServeFile)
+	mux.HandleFunc("/register", Register)
 	// TODO: mux.HandleFunc("/upload", UploadFile)
 	rpc.RegisterRPCFuncs(mux, Routes)
-	rpc.StartHTTPServer(options.ListenAddress, mux)
+	rpc.StartHTTPServer(listener, mux)
+
+	// Register this barak with central listener
+	extAddress := listener.ExternalAddress().String()
+	for _, registry := range barak.registries {
+		go func(registry string) {
+			var response ResponseRegister
+			_, err = rpc.Call(registry, "register", Arr(extAddress), &response)
+			if err != nil {
+				fmt.Printf("Error registering to registry %v:\n  %v\n", registry, err)
+			} else {
+				fmt.Printf("Successfully registered with registry %v\n", registry)
+			}
+		}(registry)
+	}
 
 	TrapSignal(func() {
 		fmt.Println("Barak shutting down")
@@ -255,8 +275,23 @@ func ListProcesses() (*ResponseListProcesses, error) {
 	}, nil
 }
 
-func ServeFile(w http.ResponseWriter, req *http.Request) {
+// Another barak instance registering its external
+// address to a remote barak.
+func Register(w http.ResponseWriter, req *http.Request) {
+	registry, err := os.OpenFile(barak.rootDir+"/registry.log", os.O_RDWR|os.O_APPEND, 0x600)
+	if err != nil {
+		http.Error(w, "Could not open registry file. Please contact the administrator", 500)
+		return
+	}
+	// TODO: Also check the X-FORWARDED-FOR or whatever it's called.
+	registry.Write([]byte(Fmt("++ %v\n", req.RemoteAddr)))
+	registry.Close()
+	w.Header().Set("Content-Type", "application/json")
+	w.WriteHeader(200)
+	w.Write([]byte("Noted!"))
+}
 
+func ServeFile(w http.ResponseWriter, req *http.Request) {
 	authCommandStr := req.FormValue("auth_command")
 	command, err := parseValidateCommandStr(authCommandStr)
 	if err != nil {