mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-17 06:09:53 +05:00
101 lines
3.6 KiB
Go
101 lines
3.6 KiB
Go
/*
|
|
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
|
|
Copyright (C) ITsysCOM GmbH
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
*/
|
|
package agents
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/utils"
|
|
"github.com/cgrates/radigo"
|
|
"github.com/cgrates/rpcclient"
|
|
)
|
|
|
|
func NewRadiusAgent(cgrCfg *config.CGRConfig, smg rpcclient.RpcClientConnection) (ra *RadiusAgent, err error) {
|
|
dicts := make(map[string]*radigo.Dictionary, len(cgrCfg.RadiusAgentCfg().ClientDictionaries))
|
|
for clntID, dictPath := range cgrCfg.RadiusAgentCfg().ClientDictionaries {
|
|
if dicts[clntID], err = radigo.NewDictionaryFromFolderWithRFC2865(dictPath); err != nil {
|
|
return
|
|
}
|
|
}
|
|
ra = &RadiusAgent{cgrCfg: cgrCfg, smg: smg}
|
|
ra.rsAuth = radigo.NewServer(cgrCfg.RadiusAgentCfg().ListenNet, cgrCfg.RadiusAgentCfg().ListenAuth, cgrCfg.RadiusAgentCfg().ClientSecrets, dicts,
|
|
map[radigo.PacketCode]func(*radigo.Packet) (*radigo.Packet, error){radigo.AccessRequest: ra.handleAuth}, nil)
|
|
ra.rsAcct = radigo.NewServer(cgrCfg.RadiusAgentCfg().ListenNet, cgrCfg.RadiusAgentCfg().ListenAcct, cgrCfg.RadiusAgentCfg().ClientSecrets, dicts,
|
|
map[radigo.PacketCode]func(*radigo.Packet) (*radigo.Packet, error){radigo.AccountingRequest: ra.handleAcct}, nil)
|
|
return
|
|
|
|
}
|
|
|
|
type RadiusAgent struct {
|
|
cgrCfg *config.CGRConfig // reference for future config reloads
|
|
smg rpcclient.RpcClientConnection // Connection towards CGR-SMG component
|
|
rsAuth *radigo.Server
|
|
rsAcct *radigo.Server
|
|
}
|
|
|
|
func (ra *RadiusAgent) handleAuth(req *radigo.Packet) (rpl *radigo.Packet, err error) {
|
|
utils.Logger.Debug(fmt.Sprintf("RadiusAgent handleAuth, received request: %+v", req))
|
|
rpl = req.Reply()
|
|
rpl.Code = radigo.AccessAccept
|
|
for _, avp := range req.AVPs {
|
|
rpl.AVPs = append(rpl.AVPs, avp)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (ra *RadiusAgent) handleAcct(req *radigo.Packet) (rpl *radigo.Packet, err error) {
|
|
req.SetAVPValues()
|
|
utils.Logger.Debug(fmt.Sprintf("Received request: %s", utils.ToJSON(req)))
|
|
rpl = req.Reply()
|
|
rpl.Code = radigo.AccountingResponse
|
|
rpl.SetAVPValues()
|
|
return
|
|
}
|
|
|
|
func (ra *RadiusAgent) processRequest(req *radigo.Packet, reqProcessor *config.RARequestProcessor) (processed bool, err error) {
|
|
passesAllFilters := true
|
|
for _, fldFilter := range reqProcessor.RequestFilter {
|
|
if passes := radPassesFieldFilter(req, fldFilter, nil); !passes {
|
|
passesAllFilters = false
|
|
}
|
|
}
|
|
if !passesAllFilters { // Not going with this processor further
|
|
return false, nil
|
|
}
|
|
return
|
|
}
|
|
|
|
func (ra *RadiusAgent) ListenAndServe() (err error) {
|
|
var errListen chan error
|
|
go func() {
|
|
utils.Logger.Info(fmt.Sprintf("<RadiusAgent> Start listening for auth requests on <%s>", ra.cgrCfg.RadiusAgentCfg().ListenAuth))
|
|
if err := ra.rsAuth.ListenAndServe(); err != nil {
|
|
errListen <- err
|
|
}
|
|
}()
|
|
go func() {
|
|
utils.Logger.Info(fmt.Sprintf("<RadiusAgent> Start listening for acct req on <%s>", ra.cgrCfg.RadiusAgentCfg().ListenAcct))
|
|
if err := ra.rsAcct.ListenAndServe(); err != nil {
|
|
errListen <- err
|
|
}
|
|
}()
|
|
err = <-errListen
|
|
return
|
|
}
|