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()
		        }
		    }
		}