登录
首页 >  Golang >  Go问答

页面重定向无效

来源:stackoverflow

时间:2024-02-19 09:00:21 351浏览 收藏

目前golang学习网上已经有很多关于Golang的文章了,自己在初次阅读这些文章中,也见识到了很多学习思路;那么本文《页面重定向无效》,也希望能帮助到大家,如果阅读完后真的对你学习Golang有帮助,欢迎动动手指,评论留言并分享~

问题内容

我正在尝试使用扫描功能插入扫描结果,完成后我想重定向以显示结果,这些结果只不过是对曾经完成的插入的选择查询。单独测试时,两个功能均工作正常。这不是从第一页重定向到下一页的正确方法吗?

package main

import (
    "bytes"
    "database/sql"
    "fmt"
    "html/template"
    "log"
    "net/http"
    "os/exec"
    "regexp"
    "strconv"
    "strings"
    "time"

    _ "github.com/go-sql-driver/mysql"
)

func dbconn() (db *sql.db) {
    dbdriver := "mysql"
    dbuser := "root"
    dbpass := "root"
    dbname := "test"
    db, err := sql.open(dbdriver, dbuser+":"+dbpass+"@/"+dbname+"?parsetime=true")
    if err != nil {
        log.fatal(err)
    }
    return db
}

var tmpl = template.must(template.parseglob("form/*"))
var hostnameinput string

func main() {

    http.handlefunc("/login", login)
    http.handlefunc("/show", show)

    lerr := http.listenandserve(":9090", nil) // setting listening port
    if lerr != nil {
        log.fatal("listenandserve: ", lerr)
    }

}

func insert(hostname string, ports []int) {
    db := dbconn()

    // insert into db
    // prepare
    stmt, e := db.prepare("insert into nmap_reports(reported_at,  host_address) values ( ?, ?)")
    if e != nil {
        log.fatal(e)
    }

    //execute
    res, er := stmt.exec(time.now(), hostname)
    if er != nil {
        log.fatal(er)
    }

    for _, v := range ports {
        insform, err := db.prepare("insert into nmap_report_ports(nmap_report_id,port) values (?,?)")
        if err != nil {
            log.fatal(err)
        }
        id, fail := res.lastinsertid()
        if fail != nil {
            log.fatal(fail)
        }
        _, er := insform.exec(id, v)
        if er != nil {
            log.fatal(er)
        }
    }
    defer db.close()
}
func select(hostname string) []result {
    db := dbconn()

    // query all data
    stmt, err := db.prepare("select nm.nmap_report_id, nm.host_address,nm.reported_at,group_concat(port) as `ports` from nmap_reports nm left join  nmap_report_ports nrp on nm.nmap_report_id = nrp.nmap_report_id where nm.host_address = ? group by nm.nmap_report_id order by nmap_report_id desc")
    if err != nil {
        log.fatal(err)
    }
    rows, er := stmt.query(hostname)
    if er != nil {
        log.fatal(er)
    }

    // declare empty result variable
    var resulti = result{}

    // iterate over rows
    resultset := make([]result, 0)
    for rows.next() {
        e := rows.scan(&resulti.reportid, &resulti.hostname, &resulti.date, &resulti.ports)
        if e != nil {
            log.fatal(er)
        }
        resultset = append(resultset, resulti)
    }
    defer db.close()
    return resultset
}

type result struct {
    reportid int       `json:"reportid"`
    hostname string    `json:"hostname"`
    date     time.time `json:"date"`
    ports    string    `json:"ports"`
}

func show(w http.responsewriter, r *http.request) {

    data := select(hostnameinput)
    temp, temperr := template.parsefiles("form/show.tmpl")
    if temperr != nil {
        log.fatal(temperr)
    }

    temp.executetemplate(w, "show", data)

}

func login(w http.responsewriter, r *http.request) {
    fmt.println("method:", r.method) //get request method
    if r.method == "get" {
        t, _ := template.parsefiles("form/input.tmpl")
        t.execute(w, nil)
    } else {
        //if r.method == "post" {
        r.parseform()

        fmt.println("hostname:", r.form["hname"])

        var rxhostip = regexp.mustcompile("^(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\\.){3}(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$|^(([a-za-z0-9]|[a-za-z0-9][a-za-z0-9\\-]*[a-za-z0-9])\\.)+([a-za-z]|[a-za-z][a-za-z0-9\\-]*[a-za-z0-9])$")
        hostnameinput = strings.join(r.form["hname"], "")
        fmt.println("hnmae is:" + hostnameinput)
        if !rxhostip.matchstring(hostnameinput) {
            w.writeheader(400)
            w.write([]byte("invalid ip/hostname"))
        }

        command := []string{"nmap", "-p", "1-1000"}
        command = append(command, r.form["hname"]...)
        finalcommand := strings.join(command, " ")
        fmt.println(finalcommand)
        cmd := exec.command(`c:\windows\system32\cmd.exe`, "/c", finalcommand)
        cmdoutput := &bytes.buffer{}
        cmd.stdout = cmdoutput

        err := cmd.run()
        if err != nil {
            log.fatal(err)
        }
        check := string(cmdoutput.bytes())

        re := regexp.mustcompile(`([0-9]+)./`)

        stringarr := re.findallstring(check, -1)

        intarr := make([]int, 0)

        for _, v := range stringarr {
            n := strings.replace(v, "/", "", -1)

            i, cerr := strconv.atoi(n)
            if cerr != nil {
                log.fatal(cerr)
            }
            intarr = append(intarr, i)
        }

        insert(hostnameinput, intarr)
    }
    http.redirect(w, r, "/show", 301)
}

尝试从登录页面重定向到显示页面

http.Redirect(w, r, "/show", 301)

解决方案


我认为重定向代码很好。 login 处理程序内部可能存在某些阻止重定向发生的内容。

我在下面注释并修改了 login 处理程序的某些部分,如果它可能对您有帮助的话。

func login(w http.ResponseWriter, r *http.Request) {
    fmt.Println("method:", r.Method) //get request method
    if r.Method == "GET" {
        // note#1: don't ignore the error
        t, err := template.ParseFiles("form/input.tmpl")
        if err != nil {
            log.Fatal(err)
        }

        // note#2: check the error
        err = t.Execute(w, nil)
        if err != nil {
            log.Fatal(err)
        }
    } else {
        var err error
        // note#3: check the error
        err = r.ParseForm()
        if err != nil {
            log.Fatal(err)
        }

        // ...
        // other code, omitted for brevity
        // ...
    }

    // note#4: this is fine if you intend to redirect all the request,
    // regardless of the method, to the `/show` endpoint
    http.Redirect(w, r, "/show", 301)
}

以上就是本文的全部内容了,是否有顺利帮助你解决问题?若是能给你带来学习上的帮助,请大家多多支持golang学习网!更多关于Golang的相关知识,也可关注golang学习网公众号。

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>