Go编写一个简单的http中间件

小鱼儿 2022-02-02 03:27 234阅读 0赞

写一个只允许一个域名下的用户访问的中间件，通过HTTP的HOSTheader实现。这样的中间件可以防止主机欺骗攻击。  
定义一个结构

type SingleHost struct {
     //处理请求
    	handler http.Handler
    	//可用host
    	allowHost string
    }

如果是可用的host，嵌入handler，如果handler，则给出错误信息  
通过 实现ServeHTTP来实现http.Handler

func (c *SingleHost)ServeHTTP(w http.ResponseWriter,r *http.Request){
    	//println("rHost:",r.Host)
    	//println("allowHost:",c.allowHost)
    	if r.Host==c.allowHost{
    		c.handler.ServeHTTP(w,r)
    	}else{
    		w.WriteHeader(403)
    	}
    }

定义自己的handler函数处理自己的业务逻辑

func myHandler(w http.ResponseWriter,r *http.Request){
    
    	w.Write([]byte("\nHello,world"))
    	//fmt.Println(r.Host)
    	//w.WriteHeader(1)
    
    }

使用http.ListenAndServe函数构建http服务

func main(){
    	single:=&SingleHost{
    		http.HandlerFunc(myHandler),
    		"sample.com",
    	}
    	http.ListenAndServe("127.0.0.1:8080",single)
    }

go bulid main.go

运行程序  
打开git命令行，输入`curl --head 127.0.0.1：8080`  
![在这里插入图片描述][20190510155234656.png]  
访问127.0.0.1:8080  
![在这里插入图片描述][20190510155849720.png]  
请求没有响应，[因为可用host为sample.com][host_sample.com]，而http服务器监听127.0.0.1:8080,只有127.0.0.1:8080才能访问到服务器  
把定义的SingleHost的allowHost改成127.0.0.1:8080后，  
![在这里插入图片描述][20190510155552802.png]  
成功返回http头信息，访问127.0.0.1:8080  
![在这里插入图片描述][20190510155700173.png]  
访问成功，这样实现了一个简单的http中间件

[20190510155234656.png]: /images/20220202/18dcbf76e43d465ab06da8fc75b595fd.png
[20190510155849720.png]: /images/20220202/7d5b5e1d4e84493583cda74606fe2a11.png
[host_sample.com]: http://xn--hostsample-xx2pea201qfwjmq8i.com
[20190510155552802.png]: /images/20220202/ba83c9705696487dad3d781e3bc4abac.png
[20190510155700173.png]: /images/20220202/be0859749adb42bb9ddaf7ee9271e7e4.png