func acctBegin(w io.Writer, req *radius.Packet) {
if e := radius.ValidateAcctRequest(req); e != "" {
config.Log.Printf("WARN: acct.begin err=" + e)
return
}
if !req.HasAttr(radius.FramedIPAddress) {
config.Log.Printf("WARN: acct.begin missing FramedIPAddress")
return
}
user := string(req.Attr(radius.UserName))
sess := string(req.Attr(radius.AcctSessionId))
nasIp := radius.DecodeIP(req.Attr(radius.NASIPAddress)).String()
clientIp := string(req.Attr(radius.CallingStationId))
assignedIp := radius.DecodeIP(req.Attr(radius.FramedIPAddress)).String()
if config.Verbose {
config.Log.Printf("acct.begin sess=%s for user=%s on nasIP=%s", sess, user, nasIp)
}
reply := []radius.AttrEncoder{}
_, e := model.Limits(user)
if e != nil {
if e == model.ErrNoRows {
config.Log.Printf("acct.begin received invalid user="******"acct.begin e=" + e.Error())
return
}
if e := model.SessionAdd(sess, user, nasIp, assignedIp, clientIp); e != nil {
config.Log.Printf("acct.begin e=%s", e.Error())
return
}
w.Write(req.Response(radius.AccountingResponse, reply))
}
func acctStop(w io.Writer, req *radius.Packet) {
if e := radius.ValidateAcctRequest(req); e != "" {
config.Log.Printf("acct.stop e=" + e)
return
}
user := string(req.Attr(radius.UserName))
sess := string(req.Attr(radius.AcctSessionId))
nasIp := radius.DecodeIP(req.Attr(radius.NASIPAddress)).String()
sessTime := radius.DecodeFour(req.Attr(radius.AcctSessionTime))
octIn := radius.DecodeFour(req.Attr(radius.AcctInputOctets))
octOut := radius.DecodeFour(req.Attr(radius.AcctOutputOctets))
packIn := radius.DecodeFour(req.Attr(radius.AcctInputPackets))
packOut := radius.DecodeFour(req.Attr(radius.AcctOutputPackets))
if config.Verbose {
config.Log.Printf(
"acct.stop sess=%s for user=%s sessTime=%d octetsIn=%d octetsOut=%d",
sess, user, sessTime, octIn, octOut,
)
}
txn, e := model.Begin()
if e != nil {
config.Log.Printf("acct.update e=" + e.Error())
return
}
sessModel := createSess(req)
if e := model.SessionUpdate(txn, sessModel); e != nil {
config.Log.Printf("acct.update e=" + e.Error())
return
}
if e := model.SessionLog(txn, sess, user, nasIp); e != nil {
config.Log.Printf("acct.update e=" + e.Error())
return
}
if e := model.SessionRemove(txn, sess, user, nasIp); e != nil {
config.Log.Printf("acct.update e=" + e.Error())
return
}
queue.Queue(user, octIn, octOut, packIn, packOut)
if e := txn.Commit(); e != nil {
config.Log.Printf("acct.update e=" + e.Error())
return
}
w.Write(radius.DefaultPacket(req, radius.AccountingResponse, "Finished accounting."))
}
func auth(w io.Writer, req *radius.Packet) {
if e := radius.ValidateAuthRequest(req); e != "" {
config.Log.Printf("auth.begin e=%s", e)
return
}
reply := []radius.AttrEncoder{}
user := string(req.Attr(radius.UserName))
limits, e := model.Auth(user)
if e != nil {
config.Log.Printf("auth.begin e=" + e.Error())
return
}
if limits.Pass == "" {
w.Write(radius.DefaultPacket(req, radius.AccessReject, "No such user"))
return
}
if req.HasAttr(radius.UserPassword) {
pass := radius.DecryptPassword(req.Attr(radius.UserPassword), req)
if pass != limits.Pass {
w.Write(radius.DefaultPacket(req, radius.AccessReject, "Invalid password"))
return
}
if config.Verbose {
config.Log.Printf("PAP login user=%s", user)
}
} else if req.HasAttr(radius.CHAPPassword) {
challenge := req.Attr(radius.CHAPChallenge)
hash := req.Attr(radius.CHAPPassword)
// TODO: No challenge then use Request Authenticator
if !radius.CHAPMatch(limits.Pass, hash, challenge) {
w.Write(radius.DefaultPacket(req, radius.AccessReject, "Invalid password"))
return
}
if config.Verbose {
config.Log.Printf("CHAP login user=%s", user)
}
} else {
// Search for MSCHAP attrs
attrs := make(map[vendor.AttributeType]radius.AttrEncoder)
for _, attr := range req.Attrs {
if radius.AttributeType(attr.Type()) == radius.VendorSpecific {
hdr := radius.VendorSpecificHeader(attr.Bytes())
if hdr.VendorId == vendor.Microsoft {
attrs[vendor.AttributeType(hdr.VendorType)] = attr
}
}
}
if len(attrs) > 0 && len(attrs) != 2 {
w.Write(radius.DefaultPacket(req, radius.AccessReject, "MSCHAP: Missing attrs? MS-CHAP-Challenge/MS-CHAP-Response"))
return
} else if len(attrs) == 2 {
// Collect our data
challenge := mschap.DecodeChallenge(attrs[vendor.MSCHAPChallenge].Bytes()).Value
if _, isV1 := attrs[vendor.MSCHAPResponse]; isV1 {
// MSCHAPv1
res := mschap.DecodeResponse(attrs[vendor.MSCHAPResponse].Bytes())
if res.Flags == 0 {
// If it is zero, the NT-Response field MUST be ignored and
// the LM-Response field used.
w.Write(radius.DefaultPacket(req, radius.AccessReject, "MSCHAPv1: LM-Response not supported."))
return
}
if bytes.Compare(res.LMResponse, []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}) != 0 {
w.Write(radius.DefaultPacket(req, radius.AccessReject, "MSCHAPv1: LM-Response set."))
return
}
// Check for correctness
calc, e := mschap.Encryptv1(challenge, limits.Pass)
if e != nil {
config.Log.Printf("MSCHAPv1: " + e.Error())
w.Write(radius.DefaultPacket(req, radius.AccessReject, "MSCHAPv1: Server-side processing error"))
return
}
mppe, e := mschap.Mppev1(limits.Pass)
if e != nil {
config.Log.Printf("MPPEv1: " + e.Error())
w.Write(radius.DefaultPacket(req, radius.AccessReject, "MPPEv1: Server-side processing error"))
return
}
if bytes.Compare(res.NTResponse, calc) != 0 {
if config.Verbose {
config.Log.Printf(
"MSCHAPv1 user=%s mismatch expect=%x, received=%x",
user, calc, res.NTResponse,
)
}
w.Write(radius.DefaultPacket(req, radius.AccessReject, "Invalid password"))
return
}
if config.Verbose {
config.Log.Printf("MSCHAPv1 login user=%s", user)
}
reply = append(reply, radius.VendorAttr{
Type: radius.VendorSpecific,
VendorId: vendor.Microsoft,
/* 1 Encryption-Allowed, 2 Encryption-Required */
Values: []radius.VendorAttrString{
radius.VendorAttrString{
Type: vendor.MSMPPEEncryptionPolicy,
Value: []byte{0x0, 0x0, 0x0, 0x01},
},
/* encryption types, allow RC4[40/128bit] */
radius.VendorAttrString{
Type: vendor.MSMPPEEncryptionTypes,
Value: []byte{0x0, 0x0, 0x0, 0x06},
},
/* mppe - encryption negotation key */
radius.VendorAttrString{
Type: vendor.MSCHAPMPPEKeys,
Value: mppe,
},
},
}.Encode())
} else if _, isV2 := attrs[vendor.MSCHAP2Response]; isV2 {
// MSCHAPv2
res := mschap.DecodeResponse2(attrs[vendor.MSCHAP2Response].Bytes())
if res.Flags != 0 {
w.Write(radius.DefaultPacket(req, radius.AccessReject, "MSCHAPv2: Flags should be set to 0"))
return
}
enc, e := mschap.Encryptv2(challenge, res.PeerChallenge, user, limits.Pass)
if e != nil {
config.Log.Printf("MSCHAPv2: " + e.Error())
w.Write(radius.DefaultPacket(req, radius.AccessReject, "MSCHAPv2: Server-side processing error"))
return
}
send, recv := mschap.Mmpev2(req.Secret(), limits.Pass, req.Auth, res.Response)
if bytes.Compare(res.Response, enc.ChallengeResponse) != 0 {
if config.Verbose {
config.Log.Printf(
"MSCHAPv2 user=%s mismatch expect=%x, received=%x",
user, enc.ChallengeResponse, res.Response,
)
}
w.Write(radius.DefaultPacket(req, radius.AccessReject, "Invalid password"))
return
}
if config.Verbose {
config.Log.Printf("MSCHAPv2 login user=%s", user)
}
// TODO: Framed-Protocol = PPP, Framed-Compression = Van-Jacobson-TCP-IP
reply = append(reply, radius.VendorAttr{
Type: radius.VendorSpecific,
VendorId: vendor.Microsoft,
Values: []radius.VendorAttrString{
/* 1 Encryption-Allowed, 2 Encryption-Required */
radius.VendorAttrString{
Type: vendor.MSMPPEEncryptionPolicy,
Value: []byte{0x0, 0x0, 0x0, 0x01},
},
/* encryption types, allow RC4[40/128bit] */
radius.VendorAttrString{
Type: vendor.MSMPPEEncryptionTypes,
Value: []byte{0x0, 0x0, 0x0, 0x06},
},
/* success challenge */
radius.VendorAttrString{
Type: vendor.MSCHAP2Success,
Value: append([]byte{byte(res.Ident)}, []byte(enc.AuthenticatorResponse)...),
},
/* Send-Key */
radius.VendorAttrString{
Type: vendor.MSMPPESendKey,
Value: send,
},
/* Recv-Key */
radius.VendorAttrString{
Type: vendor.MSMPPERecvKey,
Value: recv,
},
},
}.Encode())
} else {
w.Write(radius.DefaultPacket(req, radius.AccessReject, "MSCHAP: Response1/2 not found"))
return
}
}
}
conns, e := model.Conns(user)
if e != nil {
config.Log.Printf("auth.begin e=" + e.Error())
return
}
if conns >= limits.SimultaneousUse {
w.Write(radius.DefaultPacket(req, radius.AccessReject, "Max conns reached"))
return
}
if limits.Ok {
if limits.DedicatedIP != nil {
reply = append(reply, radius.NewAttr(
radius.FramedIPAddress,
net.ParseIP(*limits.DedicatedIP).To4(),
0,
))
}
if limits.Ratelimit != nil {
// MT-Rate-Limit = MikrotikRateLimit
reply = append(reply, radius.VendorAttr{
Type: radius.VendorSpecific,
VendorId: vendor.Mikrotik,
Values: []radius.VendorAttrString{radius.VendorAttrString{
Type: vendor.MikrotikRateLimit,
Value: []byte(*limits.Ratelimit),
}},
}.Encode())
}
if limits.DnsOne != nil {
// MS-Primary-DNS-Server
// MS-Secondary-DNS-Server
reply = append(reply, radius.VendorAttr{
Type: radius.VendorSpecific,
VendorId: vendor.Microsoft,
Values: []radius.VendorAttrString{radius.VendorAttrString{
Type: vendor.MSPrimaryDNSServer,
Value: net.ParseIP(*limits.DnsOne).To4(),
}, radius.VendorAttrString{
Type: vendor.MSSecondaryDNSServer,
Value: net.ParseIP(*limits.DnsTwo).To4(),
}},
}.Encode())
}
//reply = append(reply, radius.PubAttr{Type: radius.PortLimit, Value: radius.EncodeFour(limits.SimultaneousUse-conns)})
w.Write(req.Response(radius.AccessAccept, reply))
return
}
w.Write(radius.DefaultPacket(req, radius.AccessReject, "Invalid user/pass"))
}
func createSess(req *radius.Packet) model.Session {
return model.Session{
BytesIn: radius.DecodeFour(req.Attr(radius.AcctInputOctets)),
BytesOut: radius.DecodeFour(req.Attr(radius.AcctOutputOctets)),
PacketsIn: radius.DecodeFour(req.Attr(radius.AcctInputPackets)),
PacketsOut: radius.DecodeFour(req.Attr(radius.AcctOutputPackets)),
SessionID: string(req.Attr(radius.AcctSessionId)),
SessionTime: radius.DecodeFour(req.Attr(radius.AcctSessionTime)),
User: string(req.Attr(radius.UserName)),
NasIP: radius.DecodeIP(req.Attr(radius.NASIPAddress)).String(),
}
}