DNSEntity GetDNS(String key, DNSEntity request) { // 批量请求父级代理 var dic = DNSClient.QueryAll(_Clients, request); if (dic.Count == 0) { return(null); } DNSEntity rs = null; foreach (var item in dic) { rs = item.Value; var nc = item.Key.Client; WriteLog("{0} GetDNS {1}", nc.Remote, rs); if (OnNew != null) { var e = new DNSEventArgs { Request = request, Response = item.Value, Session = nc }; OnNew(this, e); } } return(rs); }
/// <summary>处理响应</summary> /// <param name="session"></param> /// <param name="request"></param> /// <param name="response"></param> protected virtual void Response(INetSession session, DNSEntity request, DNSEntity response) { var ss = session?.Session; if (ss == null) { return; } var isTcp = ss.Local.IsTcp; if (OnResponse != null) { var e = new DNSEventArgs { Request = request, Response = response, Session = ss }; OnResponse(this, e); } session?.Send(response.GetStream(isTcp)); }
/// <summary>处理请求</summary> /// <param name="session"></param> /// <param name="request"></param> /// <returns></returns> protected virtual DNSEntity Request(INetSession session, DNSEntity request) { var local = session.Session.Local; var isTcp = local.IsTcp; // 处理,修改 WriteLog("{0} 请求 {1}", session.Session.Remote, request); // 请求事件,如果第二参数有值,则直接返回 // 结合数据库缓存,可以在这里进行返回 if (OnRequest != null) { var e = new DNSEventArgs { Request = request }; OnRequest(this, e); if (e.Response != null) { return(e.Response); } } // 如果是PTR请求 var rq = request.Questions[0]; if (rq.Type == DNSQueryType.PTR) { var ptr = RequestPTR(request); if (ptr != null) { return(ptr); } } // 读取缓存 //var rs = cache.GetItem(request.ToString(), k => GetDNS(k, request)); var key = request.ToString(); var rs = cache[key]; if (rs == null) { cache[key] = rs = GetDNS(key, request); } // 返回给客户端 if (rs != null) { // 如果是PTR请求 if (rq.Type == DNSQueryType.PTR && rs.Questions[0].Type == DNSQueryType.PTR) { var ptr = rq as DNS_PTR; if (rs.GetAnswer() is DNS_PTR ptr2) { ptr2.Name = ptr.Name; ptr2.DomainName = DomainName; } if (rs.Answers != null && rs.Answers.Length > 0) { foreach (var item in rs.Answers) { if (item.Type == DNSQueryType.PTR) { item.Name = ptr.Name; } } } } } return(rs); }