Go語言RPC Authorization進行簡單ip安全驗證的方法

2020-04-01 19:17:57

字體：大中小

來源：轉載

供稿：網友

這篇文章主要介紹了Go語言RPC Authorization進行簡單ip安全驗證的方法,實例分析了Go語言進行ip驗證的技巧,需要的朋友可以參考下

本文實例講述了Go語言RPC Authorization進行簡單ip安全驗證的方法。分享給大家供大家參考。具體分析如下：

前言：寫網絡服務，總要考慮安全機制，對ip和網段進行判斷是最簡單的一個驗證機制。之后想做一個類似注冊式的安全驗證機制，既可以減少配置文件的麻煩，又可以很好的進行安全管理。

直接上代碼：

復制代碼代碼如下:

		package main
		import(
		    "net"
		    "fmt"
		    "time"
		    "strings"
		)
		func main(){
		    IP_ARRAY := "192.168.1.234,192.168.1.47,192.168.2.0/28"
		    servPort:=":7272"
		    l,err := net.Listen( "tcp",servPort )
		    if err != nil {
		        fmt.Printf( "Listen is error" )
		        return
		    }
		    allowList :=strings.Split( IP_ARRAY,"," )
		    for{ 
		        conn,err:=l.Accept()
		        if err != nil {
		            fmt.Printf( "start connect  is error" )
		            return
		        }
		        ipAddr:=conn.RemoteAddr()
		        Addr := strings.Split( ipAddr.String(), ":")
		        rAddr := net.ParseIP( Addr[0] )
		        var authorized bool = false
		        for v := range allowList{
		            _,ipNet,err := net.ParseCIDR( allowList[v] )
		            if err != nil{
		                fmt.Printf( "parse ip net error" )
		                ipHost := net.ParseIP( allowList[v])
		                if ipHost != nil{
		                   if ipHost.Equal( rAddr ) {
		                      authorized =true
		                   }
		                }else{
		                    fmt.Printf( "ip list error" )
		                }
		            }else{
		                fmt.Printf( "Contains ip " )
		                if ipNet.Contains( rAddr ) {
		                    authorized =true
		                }
		            }
		        }
		        if authorized == true{ 
		            curTime:=time.Now()
		            fmt.Printf( curTime.Format( "2006-01-02 15:04:05" )  )
		            conn.Write( []byte(curTime.Format( "2006-01-02 15:04:05" ) ) )
		            time.Sleep( 10)
		        }else{
		            conn.Close()
		        }
		    }
		}