登录
首页 >  Golang >  Go问答

解析如何执行动态解组

来源:stackoverflow

时间:2024-03-14 14:45:27 174浏览 收藏

从现在开始,努力学习吧!本文《解析如何执行动态解组》主要讲解了等等相关知识点,我会在golang学习网中持续更新相关的系列文章,欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧,希望能帮到你!

问题内容

任何人都可以帮助我根据从直径客户端收到的消息类型执行动态解组。在下面的代码中,我有两个结构,它们代表直径服务器接收到的两条不同的消息。我想修改当前代码,该代码将请求解组到结构 var req handlederrequest,以便根据接收到的与特定结构匹配的消息,对 var req handlederrequestvar challreq handlechallrequest 动态完成解组。我尝试使用下面的代码来实现,但它没有按预期工作。所有答案都同时返回,这不是我所期望的。

func handleder(settings sm.settings) diam.handlerfunc {

    // if received avp messages are of this struct format, unmarshal message to this structure

    type handlederrequest struct {
        sessionid         datatype.utf8string       `avp:"session-id"`
        originhost        datatype.diameteridentity `avp:"origin-host"`
        originrealm       datatype.diameteridentity `avp:"origin-realm"`
        destinationhost   datatype.diameteridentity `avp:"destination-host"`
        destinationrealm  datatype.diameteridentity `avp:"destination-realm"`
        username          datatype.utf8string       `avp:"user-name"`
        authsessionstate  datatype.enumerated       `avp:"auth-session-state"`
        authapplicationid datatype.unsigned32       `avp:"auth-application-id"`
        authrequesttype   datatype.enumerated       `avp:"auth-request-type"`
        eappayload        datatype.octetstring      `avp:"eap-payload"`
        rattype           datatype.enumerated       `avp:"rat-type"`
        anid              datatype.utf8string       `avp:"anid"`
    }

    // if received avp messages are of this struct format, unmarshal message to this structure

    type handlechallrequest struct {
        sessionid        datatype.utf8string       `avp:"session-id"`
        originhost       datatype.diameteridentity `avp:"origin-host"`
        originrealm      datatype.diameteridentity `avp:"origin-realm"`
        destinationhost  datatype.diameteridentity `avp:"destination-host"`
        destinationrealm datatype.diameteridentity `avp:"destination-realm"`
        eappayload       datatype.octetstring      `avp:"eap-payload"`
    }

    return func(c diam.conn, m *diam.message) {

        var err error = nil
        var req handlederrequest

        var code uint32 = diam.success
        err = m.unmarshal(&req)
        if err != nil {
            err = fmt.errorf("unmarshal failed: %s", err)
            code = diam.unabletocomply
            log.printf("invalid der(%d): %s\n", code, err.error())
        }
        a := m.answer(code)
        a.newavp(avp.sessionid, avp.mbit, 0, req.sessionid)
        a.newavp(avp.originhost, avp.mbit, 0, req.destinationhost)
        a.newavp(avp.originrealm, avp.mbit, 0, req.destinationrealm)
        a.newavp(avp.originstateid, avp.mbit, 0, settings.originstateid)
        _, err = aka_challenge_request(settings, c, a)
        if err != nil {
            log.printf("failed to send aaa challenge request: %s", err.error())
        }

        var challreq handlechallagerequest
        err = m.unmarshal(&challreq)
        if err != nil {
            err = fmt.errorf("unmarshal failed: %s", err)
            code = diam.unabletocomply
            log.printf("invalid der(%d): %s\n", code, err.error())
        }
        a = m.answer(code)
        a.newavp(avp.sessionid, avp.mbit, 0, req.sessionid)
        a.newavp(avp.originhost, avp.mbit, 0, req.destinationhost)
        a.newavp(avp.originrealm, avp.mbit, 0, req.destinationrealm)
        a.newavp(avp.originstateid, avp.mbit, 0, settings.originstateid)
        _, err = aka_success_notification(settings, c, a)
        if err != nil {
           log.printf("failed to send success notification: %s", err.error())
       }
    }
}
func AKA_Challenge_Request(settings sm.Settings, w io.Writer, m *diam.Message) (n int64, err error) {
    PayloadSlice := []byte(`RAND, AUTHN, MAC, RESULT_ID`)
    m.NewAVP(avp.EAPPayload, avp.Mbit, 0, datatype.OctetString(PayloadSlice))
    return m.WriteTo(w)
}

func AKA_Success_Notification(settings sm.Settings, w io.Writer, m *diam.Message) (n int64, err error) {
    EAPSlice := []byte(`EAP_Success`)
    MSKSlice := []byte(`EAP-Master-Session-Key`)
    m.NewAVP(avp.EAPPayload, avp.Mbit, 0, datatype.OctetString(EAPSlice))
    m.NewAVP(avp.EAPMasterSessionKey, avp.Mbit, 0, datatype.OctetString(MSKSlice))
    return m.WriteTo(w)
}

我知道返回函数应该有一个if条件,但我不知道如何开始。请知道如何进行。


解决方案


我想说,此时不需要 handlechallrequest 类型。如果我正确读取了字段,两种类型都具有相同的基本属性,但 handlederrequest 类型具有更多基本属性,这些属性在未编组的类型中是否只能是 nil 值?

然后,您可以执行某种 nil 检查来确定一个数据集与另一个数据集,或者更优选的是创建一个包含所有字段和一组通用方法的接口,以便您可以创建两种类型实现这个接口,然后选择创建哪个接口。

编辑 1。

如果我没有抓住要点,再次抱歉,但我仍然没有看到不同类型的需要,在我看来,它们只是数据容器,具有不同的数据。

这是我对单一数据类型的思考。

func HandleDER(settings sm.Settings) diam.HandlerFunc {

// If received AVP messages are of this struct format, Unmarshal message to this structure

type HandleRequest struct {
    SessionID         datatype.UTF8String       `avp:"Session-Id"`
    OriginHost        datatype.DiameterIdentity `avp:"Origin-Host"`
    OriginRealm       datatype.DiameterIdentity `avp:"Origin-Realm"`
    DestinationHost   datatype.DiameterIdentity `avp:"Destination-Host"`
    DestinationRealm  datatype.DiameterIdentity `avp:"Destination-Realm"`
    UserName          datatype.UTF8String       `avp:"User-Name"`
    AuthSessionState  datatype.Enumerated       `avp:"Auth-Session-State"`
    AuthApplicationID datatype.Unsigned32       `avp:"Auth-Application-Id"`
    AuthRequestType   datatype.Enumerated       `avp:"Auth-Request-Type"`
    EAPPayload        datatype.OctetString      `avp:"EAP-Payload"`
    RATType           datatype.Enumerated       `avp:"RAT-Type"`
    ANID              datatype.UTF8String       `avp:"ANID"`
}

// If received AVP messages are of this struct format, Unmarshal message to this structure

return func(c diam.Conn, m *diam.Message) {

    var err error = nil
    var req HandleRequest

    var code uint32 = diam.Success

    err = m.Unmarshal(&req)
    if err != nil {
        err = fmt.Errorf("Unmarshal failed: %s", err)
        code = diam.UnableToComply
        log.Printf("Invalid DER(%d): %s\n", code, err.Error())
    }
    a := m.Answer(code)
    a.NewAVP(avp.SessionID, avp.Mbit, 0, req.SessionID)
    a.NewAVP(avp.OriginHost, avp.Mbit, 0, req.DestinationHost)
    a.NewAVP(avp.OriginRealm, avp.Mbit, 0, req.DestinationRealm)
    a.NewAVP(avp.OriginStateID, avp.Mbit, 0, settings.OriginStateID)

    if ("messagetype" == "first message type") {
        _, err = AKA_Challenge_Request(settings, c, a)
        if err != nil {
            log.Printf("Failed to send AAA challenge request: %s", err.Error())
        }
    } else {
        _, err = AKA_Success_Notification(settings, c, a)
        if err != nil {
            log.Printf("Failed to send Success Notification: %s", err.Error())
        }
    }
}

}

以上就是《解析如何执行动态解组》的详细内容,更多关于的资料请关注golang学习网公众号!

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