在Golang中实现XSS保护

时间:2017-05-26 13:29:21

标签: go xss

我正在使用Golang构建API Rest。我有一个包含很多字段(超过100个)的结构,所以我使用gorilla/schema将来自客户端的值分配给结构,这样做非常好。

现在,我想避免用户在任何字符串字段中插入Javascript代码,在我定义bool,strings,byte []和int值的结构中。所以,现在我想知道验证它的最佳方法是什么。

我正在考虑仅在字符串字段中对结构进行交互,并制作如下内容:

Loop over the struct {
     myProperty := JSEscapeString(myProperty)
}

可以吗?在这种情况下,我如何循环结构但只包含字符串字段?

1 个答案:

答案 0 :(得分:1)

您可以使用反射循环字段并转义字符串字段。 例如:

myStruct := struct {
        IntField int
        StringField string
    } {
        IntField: 42,
        StringField: "<script>alert('foo');</script>",
    }

    value := reflect.ValueOf(&myStruct).Elem()

    // loop over the struct
    for i := 0; i < value.NumField(); i++ {
        field := value.Field(i)

        // check if the field is a string
        if field.Type() != reflect.TypeOf("") {
            continue
        }

        str := field.Interface().(string)
        // set field to escaped version of the string
        field.SetString(html.EscapeString(str))
    }

    fmt.Printf("%#v", myStruct)
    // prints: struct { IntField int; StringField string }{IntField:42, StringField:"&lt;script&gt;alert(&#39;foo&#39;);&lt;/script&gt;"}

请注意,html包中有EscapeString函数。无需实现自己的。