From 53eeb4f1a18cc5ad9a6250982d168f1c0405aacd Mon Sep 17 00:00:00 2001
From: Masataka Kuwabara <p.ck.t22@gmail.com>
Date: Sat, 4 Jul 2015 16:39:55 +0900
Subject: [PATCH] IP filtering. Close #2

---
 server.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/server.go b/server.go
index 30512f9..7f98605 100644
--- a/server.go
+++ b/server.go
@@ -7,10 +7,17 @@ import (
 	"net/rpc"
 
 	"github.com/atotto/clipboard"
+	"github.com/pocke/go-iprange"
 	"github.com/skratchdot/open-golang/open"
 )
 
 func (c *CLI) Server() int {
+	ra, err := iprange.New(c.Allow)
+	if err != nil {
+		c.writeError(err)
+		return RPCError
+	}
+
 	uri := &URI{}
 	rpc.Register(uri)
 	clipboard := &Clipboard{}
@@ -33,6 +40,9 @@ func (c *CLI) Server() int {
 			log.Println(err)
 		}
 		log.Printf("Request from %s", conn.RemoteAddr())
+		if !ra.InlucdeConn(conn) {
+			continue
+		}
 		rpc.ServeConn(conn)
 	}
 	return Success