2018-06-01 08:51:51 +00:00
|
|
|
package request
|
2018-05-29 13:42:41 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/json"
|
|
|
|
"fmt"
|
2018-06-15 13:28:28 +00:00
|
|
|
"git.xdrm.io/go/xb-api/multipart"
|
2018-05-29 13:42:41 +00:00
|
|
|
"log"
|
|
|
|
"net/http"
|
|
|
|
"strings"
|
|
|
|
)
|
|
|
|
|
2018-06-01 08:51:51 +00:00
|
|
|
func NewDataset() *DataSet {
|
|
|
|
return &DataSet{
|
|
|
|
Uri: make([]*Parameter, 0),
|
|
|
|
Get: make(map[string]*Parameter),
|
|
|
|
Form: make(map[string]*Parameter),
|
|
|
|
Set: make(map[string]*Parameter),
|
2018-05-29 13:42:41 +00:00
|
|
|
}
|
2018-06-01 08:51:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Build builds a 'DataSet' from an http request
|
|
|
|
func (i *DataSet) Build(req *http.Request) {
|
2018-05-29 13:42:41 +00:00
|
|
|
|
2018-06-01 08:51:51 +00:00
|
|
|
/* (1) GET (query) data */
|
2018-05-29 13:42:41 +00:00
|
|
|
i.fetchGet(req)
|
|
|
|
|
2018-06-01 08:51:51 +00:00
|
|
|
/* (2) We are done if GET method */
|
2018-05-29 13:42:41 +00:00
|
|
|
if req.Method == "GET" {
|
2018-06-01 08:51:51 +00:00
|
|
|
return
|
2018-05-29 13:42:41 +00:00
|
|
|
}
|
|
|
|
|
2018-06-01 08:51:51 +00:00
|
|
|
/* (3) POST (body) data */
|
2018-05-29 13:42:41 +00:00
|
|
|
i.fetchForm(req)
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2018-06-01 08:51:51 +00:00
|
|
|
// setUriData stores URL data and fills 'Set'
|
2018-05-29 13:42:41 +00:00
|
|
|
// with creating pointers inside 'Url'
|
2018-06-01 08:51:51 +00:00
|
|
|
func (i *DataSet) SetUri(data []string) {
|
2018-05-29 13:42:41 +00:00
|
|
|
|
|
|
|
for index, value := range data {
|
|
|
|
|
|
|
|
// create set index
|
|
|
|
setindex := fmt.Sprintf("URL#%d", index)
|
|
|
|
|
|
|
|
// store value in 'Set'
|
2018-06-01 08:51:51 +00:00
|
|
|
i.Set[setindex] = &Parameter{
|
2018-05-29 13:42:41 +00:00
|
|
|
Parsed: false,
|
|
|
|
Value: value,
|
|
|
|
}
|
|
|
|
|
|
|
|
// create link in 'Url'
|
2018-06-01 08:51:51 +00:00
|
|
|
i.Uri = append(i.Uri, i.Set[setindex])
|
2018-05-29 13:42:41 +00:00
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
// fetchGet stores data from the QUERY (in url parameters)
|
2018-06-01 08:51:51 +00:00
|
|
|
func (i *DataSet) fetchGet(req *http.Request) {
|
2018-05-29 13:42:41 +00:00
|
|
|
|
|
|
|
for name, value := range req.URL.Query() {
|
|
|
|
|
|
|
|
// prevent injections
|
2018-06-01 08:51:51 +00:00
|
|
|
if nameInjection(name) {
|
2018-05-29 13:42:41 +00:00
|
|
|
log.Printf("get.injection: '%s'\n", name)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
// create set index
|
|
|
|
setindex := fmt.Sprintf("GET@%s", name)
|
|
|
|
|
|
|
|
// store value in 'Set'
|
2018-06-01 08:51:51 +00:00
|
|
|
i.Set[setindex] = &Parameter{
|
2018-05-29 13:42:41 +00:00
|
|
|
Parsed: false,
|
|
|
|
Value: value,
|
|
|
|
}
|
|
|
|
|
|
|
|
// create link in 'Get'
|
|
|
|
i.Get[name] = i.Set[setindex]
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
// fetchForm stores FORM data
|
|
|
|
//
|
|
|
|
// - parse 'form-data' if not supported (not POST requests)
|
|
|
|
// - parse 'x-www-form-urlencoded'
|
|
|
|
// - parse 'application/json'
|
2018-06-01 08:51:51 +00:00
|
|
|
func (i *DataSet) fetchForm(req *http.Request) {
|
|
|
|
|
2018-05-29 13:42:41 +00:00
|
|
|
contentType := req.Header.Get("Content-Type")
|
|
|
|
|
|
|
|
// parse json
|
|
|
|
if strings.HasPrefix(contentType, "application/json") {
|
2018-06-01 08:51:51 +00:00
|
|
|
i.parseJson(req)
|
2018-05-29 13:42:41 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// parse urlencoded
|
|
|
|
if strings.HasPrefix(contentType, "application/x-www-form-urlencoded") {
|
2018-06-01 08:51:51 +00:00
|
|
|
i.parseUrlencoded(req)
|
2018-05-29 13:42:41 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// parse multipart
|
|
|
|
if strings.HasPrefix(contentType, "multipart/form-data; boundary=") {
|
2018-06-01 08:51:51 +00:00
|
|
|
i.parseMultipart(req)
|
2018-05-29 13:42:41 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// if unknown type store nothing
|
|
|
|
}
|
|
|
|
|
2018-06-01 08:51:51 +00:00
|
|
|
// parseJson parses JSON from the request body inside 'Form'
|
2018-05-29 13:42:41 +00:00
|
|
|
// and 'Set'
|
2018-06-01 08:51:51 +00:00
|
|
|
func (i *DataSet) parseJson(req *http.Request) {
|
2018-05-29 13:42:41 +00:00
|
|
|
|
|
|
|
parsed := make(map[string]interface{}, 0)
|
|
|
|
|
|
|
|
decoder := json.NewDecoder(req.Body)
|
|
|
|
|
|
|
|
// if parse error: do nothing
|
|
|
|
if err := decoder.Decode(&parsed); err != nil {
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// else store values 'parsed' values
|
|
|
|
for name, value := range parsed {
|
|
|
|
|
|
|
|
// prevent injections
|
2018-06-01 08:51:51 +00:00
|
|
|
if nameInjection(name) {
|
2018-05-29 13:42:41 +00:00
|
|
|
log.Printf("post.injection: '%s'\n", name)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
// store value in 'Set'
|
2018-06-01 08:51:51 +00:00
|
|
|
i.Set[name] = &Parameter{
|
2018-05-29 13:42:41 +00:00
|
|
|
Parsed: true,
|
|
|
|
Value: value,
|
|
|
|
}
|
|
|
|
|
|
|
|
// create link in 'Form'
|
|
|
|
i.Form[name] = i.Set[name]
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2018-06-01 08:51:51 +00:00
|
|
|
// parseUrlencoded parses urlencoded from the request body inside 'Form'
|
2018-05-29 13:42:41 +00:00
|
|
|
// and 'Set'
|
2018-06-01 08:51:51 +00:00
|
|
|
func (i *DataSet) parseUrlencoded(req *http.Request) {
|
2018-05-29 13:42:41 +00:00
|
|
|
|
|
|
|
// use http.Request interface
|
|
|
|
req.ParseForm()
|
|
|
|
|
|
|
|
for name, value := range req.PostForm {
|
|
|
|
|
|
|
|
// prevent injections
|
2018-06-01 08:51:51 +00:00
|
|
|
if nameInjection(name) {
|
2018-05-29 13:42:41 +00:00
|
|
|
log.Printf("post.injection: '%s'\n", name)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
// store value in 'Set'
|
2018-06-01 08:51:51 +00:00
|
|
|
i.Set[name] = &Parameter{
|
2018-05-29 13:42:41 +00:00
|
|
|
Parsed: false,
|
|
|
|
Value: value,
|
|
|
|
}
|
|
|
|
|
|
|
|
// create link in 'Form'
|
|
|
|
i.Form[name] = i.Set[name]
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2018-06-01 08:51:51 +00:00
|
|
|
// parseMultipart parses multi-part from the request body inside 'Form'
|
2018-05-29 13:42:41 +00:00
|
|
|
// and 'Set'
|
2018-06-01 08:51:51 +00:00
|
|
|
func (i *DataSet) parseMultipart(req *http.Request) {
|
2018-05-29 13:42:41 +00:00
|
|
|
|
|
|
|
/* (1) Create reader */
|
|
|
|
mpr := multipart.CreateReader(req)
|
|
|
|
|
|
|
|
/* (2) Parse multipart */
|
|
|
|
mpr.Parse()
|
|
|
|
|
|
|
|
/* (3) Store data into 'Form' and 'Set */
|
|
|
|
for name, component := range mpr.Components {
|
|
|
|
|
|
|
|
// prevent injections
|
2018-06-01 08:51:51 +00:00
|
|
|
if nameInjection(name) {
|
2018-05-29 13:42:41 +00:00
|
|
|
log.Printf("post.injection: '%s'\n", name)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
// store value in 'Set'
|
2018-06-01 08:51:51 +00:00
|
|
|
i.Set[name] = &Parameter{
|
2018-05-29 13:42:41 +00:00
|
|
|
Parsed: false,
|
|
|
|
File: component.File,
|
|
|
|
Value: component.Data,
|
|
|
|
}
|
|
|
|
|
|
|
|
// create link in 'Form'
|
|
|
|
i.Form[name] = i.Set[name]
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
}
|