2014-04-08 17:03:21 +08:00
|
|
|
/*
|
|
|
|
3APA3A simpliest proxy server
|
|
|
|
(c) 2002-2008 by ZARAZA <3APA3A@security.nnov.ru>
|
|
|
|
|
|
|
|
please read License Agreement
|
|
|
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "proxy.h"
|
|
|
|
|
|
|
|
|
2014-12-14 03:38:03 +08:00
|
|
|
int clientnegotiate(struct chain * redir, struct clientparam * param, struct sockaddr * addr){
|
2016-01-04 06:30:49 +08:00
|
|
|
unsigned char *buf;
|
|
|
|
unsigned char *username;
|
2014-04-08 17:03:21 +08:00
|
|
|
int res;
|
|
|
|
int len=0;
|
|
|
|
unsigned char * user, *pass;
|
|
|
|
|
|
|
|
|
|
|
|
user = redir->extuser;
|
|
|
|
pass = redir->extpass;
|
2016-01-04 06:30:49 +08:00
|
|
|
if (param->srvinbuf < 4096){
|
|
|
|
if(param->srvbuf)myfree(param->srvbuf);
|
|
|
|
param->srvbuf = myalloc(4096);
|
|
|
|
param->srvbufsize = 4096;
|
|
|
|
}
|
|
|
|
buf = param->srvbuf;
|
|
|
|
username = buf + 2048;
|
2014-04-08 17:03:21 +08:00
|
|
|
if(user) {
|
|
|
|
if (*user == '*') {
|
|
|
|
if(!param->username) return 4;
|
|
|
|
user = param->username;
|
|
|
|
pass = param->password;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
switch(redir->type){
|
|
|
|
case R_TCP:
|
|
|
|
case R_HTTP:
|
|
|
|
return 0;
|
|
|
|
case R_CONNECT:
|
|
|
|
case R_CONNECTP:
|
|
|
|
{
|
2014-12-14 03:38:03 +08:00
|
|
|
len = sprintf((char *)buf, "CONNECT ");
|
2014-04-08 17:03:21 +08:00
|
|
|
if(redir->type == R_CONNECTP && param->hostname) {
|
2016-01-04 06:30:49 +08:00
|
|
|
char * needreplace;
|
2016-02-16 20:29:51 +08:00
|
|
|
needreplace = strchr((char *)param->hostname, ':');
|
2016-01-04 06:30:49 +08:00
|
|
|
if(needreplace) buf[len++] = '[';
|
2016-02-16 20:29:51 +08:00
|
|
|
len += sprintf((char *)buf + len, "%.256s", (char *)param->hostname);
|
2016-01-04 06:30:49 +08:00
|
|
|
if(needreplace) buf[len++] = ']';
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
else {
|
2014-12-14 03:38:03 +08:00
|
|
|
if(*SAFAMILY(addr) == AF_INET6) buf[len++] = '[';
|
2016-01-07 02:09:17 +08:00
|
|
|
len += myinet_ntop(*SAFAMILY(addr), SAADDR(addr), (char *)buf+len, 256);
|
2014-12-14 03:38:03 +08:00
|
|
|
if(*SAFAMILY(addr) == AF_INET6) buf[len++] = ']';
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
len += sprintf((char *)buf + len,
|
2016-01-04 00:44:48 +08:00
|
|
|
":%hu HTTP/1.0\r\nProxy-Connection: keep-alive\r\n", ntohs(*SAPORT(addr)));
|
2014-04-08 17:03:21 +08:00
|
|
|
if(user){
|
|
|
|
len += sprintf((char *)buf + len, "Proxy-authorization: basic ");
|
2016-03-10 22:05:56 +08:00
|
|
|
sprintf((char *)username, "%.128s:%.128s", user, pass?pass:(unsigned char *)"");
|
2014-04-08 17:03:21 +08:00
|
|
|
en64(username, buf+len, (int)strlen((char *)username));
|
|
|
|
len = (int)strlen((char *)buf);
|
|
|
|
len += sprintf((char *)buf + len, "\r\n");
|
|
|
|
}
|
|
|
|
len += sprintf((char *)buf + len, "\r\n");
|
|
|
|
if(socksend(param->remsock, buf, len, conf.timeouts[CHAIN_TO]) != (int)strlen((char *)buf))
|
|
|
|
return 31;
|
2014-04-10 07:34:59 +08:00
|
|
|
param->statssrv64+=len;
|
2014-04-08 17:03:21 +08:00
|
|
|
param->nwrites++;
|
|
|
|
if((res = sockgetlinebuf(param, SERVER,buf,13,'\n',conf.timeouts[CHAIN_TO])) < 13)
|
|
|
|
return 32;
|
|
|
|
if(buf[9] != '2') return 33;
|
|
|
|
while((res = sockgetlinebuf(param, SERVER,buf,1023,'\n', conf.timeouts[CHAIN_TO])) > 2);
|
|
|
|
if(res <= 0) return 34;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case R_SOCKS4:
|
|
|
|
case R_SOCKS4P:
|
|
|
|
case R_SOCKS4B:
|
|
|
|
{
|
|
|
|
|
2014-12-14 03:38:03 +08:00
|
|
|
if(*SAFAMILY(addr) != AF_INET) return 44;
|
2014-04-08 17:03:21 +08:00
|
|
|
buf[0] = 4;
|
|
|
|
buf[1] = 1;
|
2014-12-14 03:38:03 +08:00
|
|
|
memcpy(buf+2, SAPORT(addr), 2);
|
2014-04-08 17:03:21 +08:00
|
|
|
if(redir->type == R_SOCKS4P && param->hostname) {
|
|
|
|
buf[4] = buf[5] = buf[6] = 0;
|
|
|
|
buf[7] = 3;
|
|
|
|
}
|
2014-12-14 03:38:03 +08:00
|
|
|
else memcpy(buf+4, SAADDR(addr), 4);
|
2014-04-08 17:03:21 +08:00
|
|
|
if(!user)user = (unsigned char *)"anonymous";
|
|
|
|
len = (int)strlen((char *)user) + 1;
|
|
|
|
memcpy(buf+8, user, len);
|
|
|
|
len += 8;
|
|
|
|
if(redir->type == R_SOCKS4P && param->hostname) {
|
|
|
|
int hostnamelen;
|
|
|
|
|
|
|
|
hostnamelen = (int)strlen((char *)param->hostname) + 1;
|
|
|
|
if(hostnamelen > 255) hostnamelen = 255;
|
|
|
|
memcpy(buf+len, param->hostname, hostnamelen);
|
|
|
|
len += hostnamelen;
|
|
|
|
}
|
|
|
|
if(socksend(param->remsock, buf, len, conf.timeouts[CHAIN_TO]) < len){
|
|
|
|
return 41;
|
|
|
|
}
|
2014-04-10 07:34:59 +08:00
|
|
|
param->statssrv64+=len;
|
2014-04-08 17:03:21 +08:00
|
|
|
param->nwrites++;
|
|
|
|
if((len = sockgetlinebuf(param, SERVER, buf, (redir->type == R_SOCKS4B)? 3:8, EOF, conf.timeouts[CHAIN_TO])) != ((redir->type == R_SOCKS4B)? 3:8)){
|
|
|
|
return 42;
|
|
|
|
}
|
|
|
|
if(buf[1] != 90) {
|
|
|
|
return 43;
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
case R_SOCKS5:
|
|
|
|
case R_SOCKS5P:
|
|
|
|
case R_SOCKS5B:
|
|
|
|
{
|
|
|
|
int inbuf = 0;
|
|
|
|
buf[0] = 5;
|
|
|
|
buf[1] = 1;
|
|
|
|
buf[2] = user? 2 : 0;
|
|
|
|
if(socksend(param->remsock, buf, 3, conf.timeouts[CHAIN_TO]) != 3){
|
|
|
|
return 51;
|
|
|
|
}
|
2014-04-10 07:34:59 +08:00
|
|
|
param->statssrv64+=len;
|
2014-04-08 17:03:21 +08:00
|
|
|
param->nwrites++;
|
|
|
|
if(sockgetlinebuf(param, SERVER, buf, 2, EOF, conf.timeouts[CHAIN_TO]) != 2){
|
|
|
|
return 52;
|
|
|
|
}
|
|
|
|
if(buf[0] != 5) {
|
|
|
|
return 53;
|
|
|
|
}
|
|
|
|
if(buf[1] != 0 && !(buf[1] == 2 && user)){
|
|
|
|
return 54;
|
|
|
|
}
|
|
|
|
if(buf[1] == 2){
|
|
|
|
buf[inbuf++] = 1;
|
|
|
|
buf[inbuf] = (unsigned char)strlen((char *)user);
|
|
|
|
memcpy(buf+inbuf+1, user, buf[inbuf]);
|
|
|
|
inbuf += buf[inbuf] + 1;
|
|
|
|
buf[inbuf] = pass?(unsigned char)strlen((char *)pass):0;
|
|
|
|
if(pass)memcpy(buf+inbuf+1, pass, buf[inbuf]);
|
|
|
|
inbuf += buf[inbuf] + 1;
|
|
|
|
if(socksend(param->remsock, buf, inbuf, conf.timeouts[CHAIN_TO]) != inbuf){
|
|
|
|
return 51;
|
|
|
|
}
|
2014-04-10 07:34:59 +08:00
|
|
|
param->statssrv64+=inbuf;
|
2014-04-08 17:03:21 +08:00
|
|
|
param->nwrites++;
|
|
|
|
if(sockgetlinebuf(param, SERVER, buf, 2, EOF, 60) != 2){
|
|
|
|
return 55;
|
|
|
|
}
|
|
|
|
if(buf[0] != 1 || buf[1] != 0) {
|
|
|
|
return 56;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
buf[0] = 5;
|
|
|
|
buf[1] = 1;
|
|
|
|
buf[2] = 0;
|
|
|
|
if(redir->type == R_SOCKS5P && param->hostname) {
|
|
|
|
buf[3] = 3;
|
|
|
|
len = (int)strlen((char *)param->hostname);
|
|
|
|
if(len > 255) len = 255;
|
|
|
|
buf[4] = len;
|
|
|
|
memcpy(buf + 5, param->hostname, len);
|
|
|
|
len += 5;
|
|
|
|
}
|
|
|
|
else {
|
2014-12-14 03:38:03 +08:00
|
|
|
len = 3;
|
|
|
|
buf[len++] = (*SAFAMILY(addr) == AF_INET)? 1 : 4;
|
|
|
|
memcpy(buf+len, SAADDR(addr), SAADDRLEN(addr));
|
|
|
|
len += SAADDRLEN(addr);
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
2014-12-14 03:38:03 +08:00
|
|
|
memcpy(buf+len, SAPORT(addr), 2);
|
2014-04-08 17:03:21 +08:00
|
|
|
len += 2;
|
|
|
|
if(socksend(param->remsock, buf, len, conf.timeouts[CHAIN_TO]) != len){
|
|
|
|
return 51;
|
|
|
|
}
|
2014-04-10 07:34:59 +08:00
|
|
|
param->statssrv64+=len;
|
2014-04-08 17:03:21 +08:00
|
|
|
param->nwrites++;
|
|
|
|
if(sockgetlinebuf(param, SERVER, buf, 4, EOF, conf.timeouts[CHAIN_TO]) != 4){
|
|
|
|
return 57;
|
|
|
|
}
|
|
|
|
if(buf[0] != 5) {
|
|
|
|
return 53;
|
|
|
|
}
|
|
|
|
if(buf[1] != 0) {
|
|
|
|
return 60 + (buf[1] % 10);
|
|
|
|
}
|
2016-11-21 03:24:04 +08:00
|
|
|
switch (buf[3]) {
|
|
|
|
case 1:
|
|
|
|
if (redir->type == R_SOCKS5B || sockgetlinebuf(param, SERVER, buf, 6, EOF, conf.timeouts[CHAIN_TO]) == 6)
|
|
|
|
break;
|
|
|
|
return 59;
|
|
|
|
case 3:
|
|
|
|
if (sockgetlinebuf(param, SERVER, buf, 256, 0, conf.timeouts[CHAIN_TO]) > 1)
|
|
|
|
break;
|
|
|
|
return 59;
|
|
|
|
case 4:
|
|
|
|
if (sockgetlinebuf(param, SERVER, buf, 18, EOF, conf.timeouts[CHAIN_TO]) == 18)
|
|
|
|
break;
|
|
|
|
return 59;
|
|
|
|
default:
|
|
|
|
return 58;
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
default:
|
|
|
|
|
|
|
|
return 30;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int handleredirect(struct clientparam * param, struct ace * acentry){
|
|
|
|
int connected = 0;
|
|
|
|
int weight = 1000;
|
|
|
|
int res;
|
|
|
|
int done = 0;
|
|
|
|
struct chain * cur;
|
|
|
|
struct chain * redir = NULL;
|
|
|
|
int r2;
|
|
|
|
|
|
|
|
if(param->remsock != INVALID_SOCKET) {
|
|
|
|
return 0;
|
|
|
|
}
|
2014-12-14 03:38:03 +08:00
|
|
|
if(SAISNULL(¶m->req) || !*SAPORT(¶m->req)) {
|
|
|
|
return 100;
|
|
|
|
}
|
2014-04-08 17:03:21 +08:00
|
|
|
|
|
|
|
r2 = (myrand(param, sizeof(struct clientparam))%1000);
|
|
|
|
|
|
|
|
for(cur = acentry->chains; cur; cur=cur->next){
|
|
|
|
if(((weight = weight - cur->weight) > r2)|| done) {
|
|
|
|
if(weight <= 0) {
|
|
|
|
weight += 1000;
|
|
|
|
done = 0;
|
|
|
|
r2 = (myrand(param, sizeof(struct clientparam))%1000);
|
|
|
|
}
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
param->redirected++;
|
|
|
|
done = 1;
|
|
|
|
if(weight <= 0) {
|
|
|
|
weight += 1000;
|
|
|
|
done = 0;
|
|
|
|
r2 = (myrand(param, sizeof(struct clientparam))%1000);
|
|
|
|
}
|
|
|
|
if(!connected){
|
2016-02-05 06:52:45 +08:00
|
|
|
if(cur->type == R_EXTIP){
|
2016-02-05 23:31:17 +08:00
|
|
|
param->sinsl = cur->addr;
|
2016-02-08 19:39:29 +08:00
|
|
|
if(cur->next)continue;
|
|
|
|
return 0;
|
2016-02-05 06:52:45 +08:00
|
|
|
}
|
|
|
|
else if(SAISNULL(&cur->addr) && !*SAPORT(&cur->addr)){
|
2014-04-08 17:03:21 +08:00
|
|
|
if(cur->extuser){
|
|
|
|
if(param->extusername)
|
|
|
|
myfree(param->extusername);
|
|
|
|
param->extusername = (unsigned char *)mystrdup((char *)((*cur->extuser == '*' && param->username)? param->username : cur->extuser));
|
|
|
|
if(cur->extpass){
|
|
|
|
if(param->extpassword)
|
|
|
|
myfree(param->extpassword);
|
|
|
|
param->extpassword = (unsigned char *)mystrdup((char *)((*cur->extuser == '*' && param->password)?param->password : cur->extpass));
|
|
|
|
}
|
|
|
|
if(*cur->extuser == '*' && !param->username) return 4;
|
|
|
|
}
|
|
|
|
switch(cur->type){
|
|
|
|
case R_POP3:
|
|
|
|
param->redirectfunc = pop3pchild;
|
|
|
|
break;
|
|
|
|
case R_FTP:
|
|
|
|
param->redirectfunc = ftpprchild;
|
|
|
|
break;
|
|
|
|
case R_ADMIN:
|
|
|
|
param->redirectfunc = adminchild;
|
|
|
|
break;
|
|
|
|
case R_ICQ:
|
|
|
|
param->redirectfunc = icqprchild;
|
|
|
|
break;
|
2016-02-05 06:52:45 +08:00
|
|
|
case R_SMTP:
|
|
|
|
param->redirectfunc = smtppchild;
|
2014-04-08 17:03:21 +08:00
|
|
|
break;
|
|
|
|
default:
|
|
|
|
param->redirectfunc = proxychild;
|
|
|
|
}
|
2016-01-04 00:44:48 +08:00
|
|
|
if(cur->next)continue;
|
2014-04-08 17:03:21 +08:00
|
|
|
return 0;
|
|
|
|
}
|
2014-12-14 03:38:03 +08:00
|
|
|
else if(!*SAPORT(&cur->addr) && !SAISNULL(&cur->addr)) {
|
2014-11-26 06:50:14 +08:00
|
|
|
unsigned short port = *SAPORT(¶m->sinsr);
|
2016-02-05 23:31:17 +08:00
|
|
|
param->sinsr = cur->addr;
|
2014-11-26 06:50:14 +08:00
|
|
|
*SAPORT(¶m->sinsr) = port;
|
2014-10-21 10:00:56 +08:00
|
|
|
}
|
2014-12-14 03:38:03 +08:00
|
|
|
else if(SAISNULL(&cur->addr) && *SAPORT(&cur->addr)) *SAPORT(¶m->sinsr) = *SAPORT(&cur->addr);
|
|
|
|
else {
|
2016-02-05 23:31:17 +08:00
|
|
|
param->sinsr = cur->addr;
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
if((res = alwaysauth(param))){
|
|
|
|
return (res == 10)? res : 60+res;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else {
|
2014-12-14 03:38:03 +08:00
|
|
|
res = (redir)?clientnegotiate(redir, param, (struct sockaddr *)&cur->addr):0;
|
2014-04-08 17:03:21 +08:00
|
|
|
if(res) return res;
|
|
|
|
}
|
|
|
|
redir = cur;
|
|
|
|
param->redirtype = redir->type;
|
|
|
|
if(redir->type == R_TCP || redir->type ==R_HTTP) {
|
|
|
|
if(cur->extuser){
|
|
|
|
if(*cur -> extuser == '*' && !param->username) return 4;
|
|
|
|
if(param->extusername)
|
|
|
|
myfree(param->extusername);
|
|
|
|
param->extusername = (unsigned char *)mystrdup((char *)((*cur->extuser == '*' && param->username)? param->username : cur->extuser));
|
|
|
|
if(cur->extpass){
|
|
|
|
if(param->extpassword)
|
|
|
|
myfree(param->extpassword);
|
|
|
|
param->extpassword = (unsigned char *)mystrdup((char *)((*cur->extuser == '*' && param->password)?param->password : cur->extpass));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
connected = 1;
|
|
|
|
}
|
|
|
|
|
2016-08-16 20:21:02 +08:00
|
|
|
if(!connected || !redir) return 0;
|
|
|
|
return clientnegotiate(redir, param, (struct sockaddr *)¶m->req);
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
|
2014-05-12 04:21:22 +08:00
|
|
|
int IPInentry(struct sockaddr *sa, struct iplist *ipentry){
|
2014-12-14 00:48:41 +08:00
|
|
|
int addrlen;
|
2014-05-12 04:21:22 +08:00
|
|
|
unsigned char *ip, *ipf, *ipt;
|
2014-12-14 00:48:41 +08:00
|
|
|
|
|
|
|
|
2016-01-21 20:45:39 +08:00
|
|
|
if(!sa || ! ipentry || *SAFAMILY(sa) != ipentry->family) return 0;
|
|
|
|
|
2014-05-12 04:21:22 +08:00
|
|
|
ip = (unsigned char *)SAADDR(sa);
|
|
|
|
ipf = (unsigned char *)&ipentry->ip_from;
|
|
|
|
ipt = (unsigned char *)&ipentry->ip_to;
|
2014-12-14 00:48:41 +08:00
|
|
|
|
2016-01-21 20:45:39 +08:00
|
|
|
|
2014-05-12 04:21:22 +08:00
|
|
|
addrlen = SAADDRLEN(sa);
|
2014-12-14 00:48:41 +08:00
|
|
|
|
|
|
|
if(memcmp(ip,ipf,addrlen) < 0 || memcmp(ip,ipt,addrlen) > 0) return 0;
|
2014-05-12 04:21:22 +08:00
|
|
|
return 1;
|
|
|
|
|
|
|
|
}
|
2014-04-08 17:03:21 +08:00
|
|
|
|
|
|
|
int ACLmatches(struct ace* acentry, struct clientparam * param){
|
|
|
|
struct userlist * userentry;
|
|
|
|
struct iplist *ipentry;
|
|
|
|
struct portlist *portentry;
|
|
|
|
struct period *periodentry;
|
|
|
|
unsigned char * username;
|
|
|
|
struct hostname * hstentry=NULL;
|
|
|
|
int i;
|
|
|
|
int match = 0;
|
|
|
|
|
|
|
|
username = param->username?param->username:(unsigned char *)"-";
|
|
|
|
if(acentry->src) {
|
|
|
|
for(ipentry = acentry->src; ipentry; ipentry = ipentry->next)
|
2014-05-12 04:21:22 +08:00
|
|
|
if(IPInentry((struct sockaddr *)¶m->sincr, ipentry)) {
|
2014-04-08 17:03:21 +08:00
|
|
|
break;
|
|
|
|
}
|
2014-05-12 04:21:22 +08:00
|
|
|
if(!ipentry) return 0;
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
2016-01-21 20:45:39 +08:00
|
|
|
if((acentry->dst && !SAISNULL(¶m->req)) || (acentry->dstnames && param->hostname)) {
|
2014-04-08 17:03:21 +08:00
|
|
|
for(ipentry = acentry->dst; ipentry; ipentry = ipentry->next)
|
2014-05-12 04:21:22 +08:00
|
|
|
if(IPInentry((struct sockaddr *)¶m->req, ipentry)) {
|
2014-04-08 17:03:21 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
if(!ipentry) {
|
|
|
|
if(acentry->dstnames && param->hostname){
|
|
|
|
for(i=0; param->hostname[i]; i++){
|
|
|
|
param->hostname[i] = tolower(param->hostname[i]);
|
|
|
|
}
|
|
|
|
while(i > 5 && param->hostname[i-1] == '.') param->hostname[i-1] = 0;
|
|
|
|
for(hstentry = acentry->dstnames; hstentry; hstentry = hstentry->next){
|
|
|
|
switch(hstentry->matchtype){
|
|
|
|
case 0:
|
2016-02-16 20:29:51 +08:00
|
|
|
if(strstr((char *)param->hostname, (char *)hstentry->name)) match = 1;
|
2014-04-08 17:03:21 +08:00
|
|
|
break;
|
|
|
|
|
|
|
|
case 1:
|
2016-02-16 20:29:51 +08:00
|
|
|
if(strstr((char *)param->hostname, (char *)hstentry->name) == (char *)param->hostname) match = 1;
|
2014-04-08 17:03:21 +08:00
|
|
|
break;
|
|
|
|
|
|
|
|
case 2:
|
2016-02-16 20:29:51 +08:00
|
|
|
if(strstr((char *)param->hostname, (char *)hstentry->name) == (char *)(param->hostname + i - (strlen((char *)hstentry->name)))) match = 1;
|
2014-04-08 17:03:21 +08:00
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
2016-02-16 20:29:51 +08:00
|
|
|
if(!strcmp((char *)param->hostname, (char *)hstentry->name)) match = 1;
|
2014-04-08 17:03:21 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
if(match) break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if(!ipentry && !hstentry) return 0;
|
|
|
|
}
|
2014-10-21 10:00:56 +08:00
|
|
|
if(acentry->ports && *SAPORT(¶m->req)) {
|
2014-04-08 17:03:21 +08:00
|
|
|
for (portentry = acentry->ports; portentry; portentry = portentry->next)
|
2014-10-21 10:00:56 +08:00
|
|
|
if(ntohs(*SAPORT(¶m->req)) >= portentry->startport &&
|
|
|
|
ntohs(*SAPORT(¶m->req)) <= portentry->endport) {
|
2014-04-08 17:03:21 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
if(!portentry) return 0;
|
|
|
|
}
|
|
|
|
if(acentry->wdays){
|
|
|
|
if(!(acentry -> wdays & wday)) return 0;
|
|
|
|
}
|
|
|
|
if(acentry->periods){
|
|
|
|
int start_time = (int)(param->time_start - basetime);
|
|
|
|
for(periodentry = acentry->periods; periodentry; periodentry = periodentry -> next)
|
|
|
|
if(start_time >= periodentry->fromtime && start_time < periodentry->totime){
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if(!periodentry) return 0;
|
|
|
|
}
|
|
|
|
if(acentry->users){
|
|
|
|
for(userentry = acentry->users; userentry; userentry = userentry->next)
|
|
|
|
if(!strcmp((char *)username, (char *)userentry->user)){
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if(!userentry) return 0;
|
|
|
|
}
|
|
|
|
if(acentry->operation) {
|
|
|
|
if((acentry->operation & param->operation) != param->operation){
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if(acentry->weight && (acentry->weight < param->weight)) return 0;
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void initbandlims (struct clientparam *param){
|
|
|
|
struct bandlim * be;
|
|
|
|
int i;
|
|
|
|
for(i=0, be = conf.bandlimiter; be && i<MAXBANDLIMS; be = be->next) {
|
|
|
|
if(ACLmatches(be->ace, param)){
|
|
|
|
if(be->ace->action == NOBANDLIM) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
param->bandlims[i++] = be;
|
|
|
|
param->bandlimfunc = conf.bandlimfunc;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if(i<MAXBANDLIMS)param->bandlims[i] = NULL;
|
|
|
|
for(i=0, be = conf.bandlimiterout; be && i<MAXBANDLIMS; be = be->next) {
|
|
|
|
if(ACLmatches(be->ace, param)){
|
|
|
|
if(be->ace->action == NOBANDLIM) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
param->bandlimsout[i++] = be;
|
|
|
|
param->bandlimfunc = conf.bandlimfunc;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if(i<MAXBANDLIMS)param->bandlimsout[i] = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
unsigned bandlimitfunc(struct clientparam *param, unsigned nbytesin, unsigned nbytesout){
|
|
|
|
unsigned sleeptime = 0, nsleeptime;
|
|
|
|
unsigned long sec;
|
|
|
|
unsigned msec;
|
|
|
|
unsigned now;
|
|
|
|
int i;
|
|
|
|
|
|
|
|
#ifdef _WIN32
|
|
|
|
struct timeb tb;
|
|
|
|
|
|
|
|
ftime(&tb);
|
|
|
|
sec = (unsigned)tb.time;
|
|
|
|
msec = (unsigned)tb.millitm*1000;
|
|
|
|
#else
|
|
|
|
struct timeval tv;
|
|
|
|
gettimeofday(&tv, NULL);
|
|
|
|
|
|
|
|
sec = tv.tv_sec;
|
|
|
|
msec = tv.tv_usec;
|
|
|
|
#endif
|
2014-12-14 05:46:03 +08:00
|
|
|
|
2014-04-08 17:03:21 +08:00
|
|
|
if(!nbytesin && !nbytesout) return 0;
|
|
|
|
pthread_mutex_lock(&bandlim_mutex);
|
2016-03-28 22:49:27 +08:00
|
|
|
if(param->paused != conf.paused){
|
|
|
|
return (1);
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
for(i=0; nbytesin&& i<MAXBANDLIMS && param->bandlims[i]; i++){
|
|
|
|
if( !param->bandlims[i]->basetime ||
|
|
|
|
param->bandlims[i]->basetime > sec ||
|
|
|
|
param->bandlims[i]->basetime < (sec - 120)
|
|
|
|
)
|
|
|
|
{
|
|
|
|
param->bandlims[i]->basetime = sec;
|
|
|
|
param->bandlims[i]->nexttime = 0;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
now = ((sec - param->bandlims[i]->basetime) * 1000000) + msec;
|
|
|
|
nsleeptime = (param->bandlims[i]->nexttime > now)?
|
|
|
|
param->bandlims[i]->nexttime - now : 0;
|
|
|
|
sleeptime = (nsleeptime > sleeptime)? nsleeptime : sleeptime;
|
|
|
|
param->bandlims[i]->basetime = sec;
|
|
|
|
param->bandlims[i]->nexttime = msec + nsleeptime + ((param->bandlims[i]->rate > 1000000)? ((nbytesin/32)*(256000000/param->bandlims[i]->rate)) : (nbytesin * (8000000/param->bandlims[i]->rate)));
|
|
|
|
}
|
|
|
|
for(i=0; nbytesout && i<MAXBANDLIMS && param->bandlimsout[i]; i++){
|
|
|
|
if( !param->bandlimsout[i]->basetime ||
|
|
|
|
param->bandlimsout[i]->basetime > sec ||
|
|
|
|
param->bandlimsout[i]->basetime < (sec - 120)
|
|
|
|
)
|
|
|
|
{
|
|
|
|
param->bandlimsout[i]->basetime = sec;
|
|
|
|
param->bandlimsout[i]->nexttime = 0;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
now = ((sec - param->bandlimsout[i]->basetime) * 1000000) + msec;
|
|
|
|
nsleeptime = (param->bandlimsout[i]->nexttime > now)?
|
|
|
|
param->bandlimsout[i]->nexttime - now : 0;
|
|
|
|
sleeptime = (nsleeptime > sleeptime)? nsleeptime : sleeptime;
|
|
|
|
param->bandlimsout[i]->basetime = sec;
|
|
|
|
param->bandlimsout[i]->nexttime = msec + nsleeptime + ((param->bandlimsout[i]->rate > 1000000)? ((nbytesout/32)*(256000000/param->bandlimsout[i]->rate)) : (nbytesout * (8000000/param->bandlimsout[i]->rate)));
|
|
|
|
}
|
|
|
|
pthread_mutex_unlock(&bandlim_mutex);
|
|
|
|
return sleeptime/1000;
|
|
|
|
}
|
|
|
|
|
|
|
|
void trafcountfunc(struct clientparam *param){
|
|
|
|
struct trafcount * tc;
|
|
|
|
int countout = 0;
|
|
|
|
|
|
|
|
pthread_mutex_lock(&tc_mutex);
|
|
|
|
for(tc = conf.trafcounter; tc; tc = tc->next) {
|
|
|
|
if(ACLmatches(tc->ace, param)){
|
|
|
|
time_t t;
|
|
|
|
if(tc->ace->action == NOCOUNTIN) break;
|
|
|
|
if(tc->ace->action != COUNTIN) {
|
|
|
|
countout = 1;
|
|
|
|
continue;
|
|
|
|
}
|
2014-04-10 07:34:59 +08:00
|
|
|
tc->traf64 += param->statssrv64;
|
2014-04-08 17:03:21 +08:00
|
|
|
time(&t);
|
|
|
|
tc->updated = t;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if(countout) for(tc = conf.trafcounter; tc; tc = tc->next) {
|
|
|
|
if(ACLmatches(tc->ace, param)){
|
|
|
|
time_t t;
|
|
|
|
if(tc->ace->action == NOCOUNTOUT) break;
|
|
|
|
if(tc->ace->action != COUNTOUT) {
|
|
|
|
continue;
|
|
|
|
}
|
2014-04-10 07:34:59 +08:00
|
|
|
tc->traf64 += param->statscli64;
|
2014-04-08 17:03:21 +08:00
|
|
|
time(&t);
|
|
|
|
tc->updated = t;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pthread_mutex_unlock(&tc_mutex);
|
|
|
|
}
|
|
|
|
|
|
|
|
int alwaysauth(struct clientparam * param){
|
|
|
|
int res;
|
|
|
|
struct trafcount * tc;
|
|
|
|
int countout = 0;
|
|
|
|
|
|
|
|
res = doconnect(param);
|
|
|
|
if(!res){
|
|
|
|
initbandlims(param);
|
|
|
|
for(tc = conf.trafcounter; tc; tc = tc->next) {
|
|
|
|
if(tc->disabled) continue;
|
|
|
|
if(ACLmatches(tc->ace, param)){
|
|
|
|
if(tc->ace->action == NOCOUNTIN) break;
|
|
|
|
if(tc->ace->action != COUNTIN) {
|
|
|
|
countout = 1;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2014-04-10 07:34:59 +08:00
|
|
|
if(tc->traflim64 <= tc->traf64) return 10;
|
|
|
|
param->trafcountfunc = conf.trafcountfunc;
|
|
|
|
param->maxtrafin64 = tc->traflim64 - tc->traf64;
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
if(countout)for(tc = conf.trafcounter; tc; tc = tc->next) {
|
|
|
|
if(tc->disabled) continue;
|
|
|
|
if(ACLmatches(tc->ace, param)){
|
|
|
|
if(tc->ace->action == NOCOUNTOUT) break;
|
|
|
|
if(tc->ace->action != COUNTOUT) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2014-04-10 07:34:59 +08:00
|
|
|
if(tc->traflim64 <= tc->traf64) return 10;
|
|
|
|
param->trafcountfunc = conf.trafcountfunc;
|
|
|
|
param->maxtrafout64 = tc->traflim64 - tc->traf64;
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
int checkACL(struct clientparam * param){
|
|
|
|
struct ace* acentry;
|
|
|
|
|
|
|
|
if(!param->srv->acl) {
|
|
|
|
return alwaysauth(param);
|
|
|
|
}
|
|
|
|
for(acentry = param->srv->acl; acentry; acentry = acentry->next) {
|
|
|
|
if(ACLmatches(acentry, param)) {
|
|
|
|
param->nolog = acentry->nolog;
|
|
|
|
param->weight = acentry->weight;
|
|
|
|
if(acentry->action == 2) {
|
|
|
|
struct ace dup;
|
|
|
|
|
|
|
|
if(param->operation < 256 && !(param->operation & CONNECT)){
|
|
|
|
continue;
|
|
|
|
}
|
2014-12-14 03:38:03 +08:00
|
|
|
if(param->redirected && acentry->chains && SAISNULL(&acentry->chains->addr) && !*SAPORT(&acentry->chains->addr)) {
|
2014-04-08 17:03:21 +08:00
|
|
|
continue;
|
|
|
|
}
|
2016-02-05 23:31:17 +08:00
|
|
|
dup = *acentry;
|
2014-04-08 17:03:21 +08:00
|
|
|
return handleredirect(param, &dup);
|
|
|
|
}
|
|
|
|
return acentry->action;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 3;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct authcache {
|
|
|
|
char * username;
|
|
|
|
char * password;
|
|
|
|
time_t expires;
|
2014-05-12 04:21:22 +08:00
|
|
|
#ifndef NOIPV6
|
|
|
|
struct sockaddr_in6 sa;
|
|
|
|
#else
|
|
|
|
struct sockaddr_in sa;
|
|
|
|
#endif
|
2014-04-08 17:03:21 +08:00
|
|
|
struct authcache *next;
|
|
|
|
} *authc = NULL;
|
|
|
|
|
|
|
|
|
|
|
|
int cacheauth(struct clientparam * param){
|
|
|
|
struct authcache *ac, *last=NULL;
|
|
|
|
|
|
|
|
pthread_mutex_lock(&hash_mutex);
|
|
|
|
for(ac = authc; ac; ){
|
|
|
|
if(ac->expires <= conf.time){
|
|
|
|
if(ac->username)myfree(ac->username);
|
|
|
|
if(ac->password)myfree(ac->password);
|
|
|
|
if(!last){
|
|
|
|
authc = ac->next;
|
|
|
|
myfree(ac);
|
|
|
|
ac = authc;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
last->next = ac->next;
|
|
|
|
myfree(ac);
|
|
|
|
ac = last->next;
|
|
|
|
}
|
|
|
|
continue;
|
|
|
|
|
|
|
|
}
|
2016-02-16 20:29:51 +08:00
|
|
|
if(((!(conf.authcachetype&2)) || (param->username && ac->username && !strcmp(ac->username, (char *)param->username))) &&
|
2016-05-18 04:27:10 +08:00
|
|
|
((!(conf.authcachetype&1)) || (*SAFAMILY(&ac->sa) == *SAFAMILY(¶m->sincr) && !memcmp(SAADDR(&ac->sa), SAADDR(¶m->sincr), SAADDRLEN(&ac->sa)))) &&
|
2016-02-16 20:29:51 +08:00
|
|
|
(!(conf.authcachetype&4) || (ac->password && param->password && !strcmp(ac->password, (char *)param->password)))) {
|
2014-04-08 17:03:21 +08:00
|
|
|
if(param->username){
|
|
|
|
myfree(param->username);
|
|
|
|
}
|
2016-02-16 20:29:51 +08:00
|
|
|
param->username = (unsigned char *)mystrdup(ac->username);
|
2014-04-08 17:03:21 +08:00
|
|
|
pthread_mutex_unlock(&hash_mutex);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
last = ac;
|
|
|
|
ac = ac->next;
|
|
|
|
}
|
|
|
|
|
|
|
|
pthread_mutex_unlock(&hash_mutex);
|
|
|
|
return 4;
|
|
|
|
}
|
|
|
|
|
|
|
|
int doauth(struct clientparam * param){
|
|
|
|
int res = 0;
|
|
|
|
struct auth *authfuncs;
|
|
|
|
struct authcache *ac;
|
|
|
|
char * tmp;
|
|
|
|
int ret = 0;
|
|
|
|
|
|
|
|
for(authfuncs=param->srv->authfuncs; authfuncs; authfuncs=authfuncs->next){
|
|
|
|
res = authfuncs->authenticate?(*authfuncs->authenticate)(param):0;
|
|
|
|
if(!res) {
|
|
|
|
if(authfuncs->authorize &&
|
|
|
|
(res = (*authfuncs->authorize)(param)))
|
|
|
|
return res;
|
|
|
|
if(conf.authcachetype && authfuncs->authenticate && authfuncs->authenticate != cacheauth && param->username && (!(conf.authcachetype&4) || (!param->pwtype && param->password))){
|
|
|
|
pthread_mutex_lock(&hash_mutex);
|
|
|
|
for(ac = authc; ac; ac = ac->next){
|
2016-02-16 20:29:51 +08:00
|
|
|
if((!(conf.authcachetype&2) || !strcmp(ac->username, (char *)param->username)) &&
|
2016-05-18 04:27:10 +08:00
|
|
|
(!(conf.authcachetype&1) || (*SAFAMILY(&ac->sa) == *SAFAMILY(¶m->sincr) && !memcmp(SAADDR(&ac->sa), SAADDR(¶m->sincr), SAADDRLEN(&ac->sa)))) &&
|
2016-02-16 20:29:51 +08:00
|
|
|
(!(conf.authcachetype&4) || (ac->password && !strcmp(ac->password, (char *)param->password)))) {
|
2014-04-08 17:03:21 +08:00
|
|
|
ac->expires = conf.time + conf.authcachetime;
|
2016-02-16 20:29:51 +08:00
|
|
|
if(strcmp(ac->username, (char *)param->username)){
|
2014-04-08 17:03:21 +08:00
|
|
|
tmp = ac->username;
|
2016-02-16 20:29:51 +08:00
|
|
|
ac->username = mystrdup((char *)param->username);
|
2014-04-08 17:03:21 +08:00
|
|
|
myfree(tmp);
|
|
|
|
}
|
|
|
|
if((conf.authcachetype&4)){
|
|
|
|
tmp = ac->password;
|
2016-02-16 20:29:51 +08:00
|
|
|
ac->password = mystrdup((char *)param->password);
|
2014-04-08 17:03:21 +08:00
|
|
|
myfree(tmp);
|
|
|
|
}
|
2016-02-05 23:31:17 +08:00
|
|
|
ac->sa = param->sincr;
|
2014-04-08 17:03:21 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if(!ac){
|
|
|
|
ac = myalloc(sizeof(struct authcache));
|
|
|
|
if(ac){
|
|
|
|
ac->expires = conf.time + conf.authcachetime;
|
2016-05-18 04:27:10 +08:00
|
|
|
ac->username = param->username?mystrdup((char *)param->username):NULL;
|
2016-02-05 23:31:17 +08:00
|
|
|
ac->sa = param->sincr;
|
2014-04-08 17:03:21 +08:00
|
|
|
ac->password = NULL;
|
2016-02-16 20:29:51 +08:00
|
|
|
if((conf.authcachetype&4) && param->password) ac->password = mystrdup((char *)param->password);
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
ac->next = authc;
|
|
|
|
authc = ac;
|
|
|
|
}
|
|
|
|
pthread_mutex_unlock(&hash_mutex);
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
if(res > ret) ret = res;
|
|
|
|
}
|
|
|
|
if(!res){
|
|
|
|
return alwaysauth(param);
|
|
|
|
}
|
|
|
|
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
int ipauth(struct clientparam * param){
|
|
|
|
int res;
|
|
|
|
unsigned char *username;
|
|
|
|
username = param->username;
|
|
|
|
param->username = NULL;
|
|
|
|
res = checkACL(param);
|
|
|
|
param->username = username;
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
|
|
|
int userauth(struct clientparam * param){
|
|
|
|
return (param->username)? 0:4;
|
|
|
|
}
|
|
|
|
|
|
|
|
int dnsauth(struct clientparam * param){
|
2014-12-14 10:33:08 +08:00
|
|
|
char buf[128];
|
|
|
|
char addr[16];
|
|
|
|
char dig[]="0123456789abcdef";
|
2014-12-14 01:30:20 +08:00
|
|
|
|
2014-12-14 10:33:08 +08:00
|
|
|
unsigned u;
|
|
|
|
int i;
|
2014-04-08 17:03:21 +08:00
|
|
|
|
2014-12-14 10:33:08 +08:00
|
|
|
if(*SAFAMILY(¶m->sincr)!=AF_INET){
|
|
|
|
char *s = buf;
|
|
|
|
for(i=15; i>=0; i--){
|
|
|
|
unsigned char c=((unsigned char *)SAADDR(¶m->sincr))[i];
|
|
|
|
*s++ = dig[(c&0xf)];
|
|
|
|
*s++ = '.';
|
|
|
|
*s++ = dig[(c>>4)];
|
|
|
|
*s++ = '.';
|
|
|
|
}
|
|
|
|
sprintf(s, "ip6.arpa");
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
u = ntohl(*(unsigned long *)SAADDR(¶m->sincr));
|
2014-04-08 17:03:21 +08:00
|
|
|
|
2014-12-14 10:33:08 +08:00
|
|
|
sprintf(buf, "%u.%u.%u.%u.in-addr.arpa",
|
|
|
|
((u&0x000000FF)),
|
|
|
|
((u&0x0000FF00)>>8),
|
|
|
|
((u&0x00FF0000)>>16),
|
|
|
|
((u&0xFF000000)>>24));
|
2014-04-08 17:03:21 +08:00
|
|
|
|
2014-12-14 10:33:08 +08:00
|
|
|
}
|
2016-02-16 20:29:51 +08:00
|
|
|
if(!udpresolve(*SAFAMILY(¶m->sincr), (unsigned char *)buf, (unsigned char *)addr, NULL, param, 1)) return 6;
|
2014-12-14 10:33:08 +08:00
|
|
|
if(!memcmp(SAADDR(¶m->sincr), addr, SAADDRLEN(¶m->sincr))) return 6;
|
2014-04-08 17:03:21 +08:00
|
|
|
|
|
|
|
return param->username? 0:4;
|
|
|
|
}
|
|
|
|
|
|
|
|
int strongauth(struct clientparam * param){
|
|
|
|
struct passwords * pwl;
|
|
|
|
unsigned char buf[256];
|
|
|
|
|
|
|
|
|
|
|
|
if(!param->username) return 4;
|
|
|
|
pthread_mutex_lock(&pwl_mutex);
|
|
|
|
for(pwl = conf.pwl; pwl; pwl=pwl->next){
|
|
|
|
if(!strcmp((char *)pwl->user, (char *)param->username)) switch(pwl->pwtype) {
|
|
|
|
case CL:
|
|
|
|
if(!pwl->password || !*pwl->password){
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
else if (!param->pwtype && param->password && !strcmp((char *)param->password, (char *)pwl->password)){
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
#ifndef NOCRYPT
|
|
|
|
else if (param->pwtype == 2 && param->password) {
|
|
|
|
ntpwdhash(buf, pwl->password, 0);
|
|
|
|
mschap(buf, param->password, buf + 16);
|
|
|
|
if(!memcmp(buf+16, param->password+8, 24)) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
pthread_mutex_unlock(&pwl_mutex);
|
|
|
|
return 6;
|
|
|
|
#ifndef NOCRYPT
|
|
|
|
case CR:
|
|
|
|
if(param->password && !param->pwtype && !strcmp((char *)pwl->password, (char *)mycrypt(param->password, pwl->password,buf))) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
pthread_mutex_unlock(&pwl_mutex);
|
|
|
|
return 7;
|
|
|
|
case NT:
|
|
|
|
if(param->password && !param->pwtype && !memcmp(pwl->password, ntpwdhash(buf,param->password, 1), 32)) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
else if (param->pwtype == 2){
|
|
|
|
fromhex(pwl->password, buf, 16);
|
|
|
|
mschap(buf, param->password, buf + 16);
|
|
|
|
if(!memcmp(buf + 16, param->password+8, 24)) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
pthread_mutex_unlock(&pwl_mutex);
|
|
|
|
return 8;
|
|
|
|
#endif
|
|
|
|
default:
|
|
|
|
pthread_mutex_unlock(&pwl_mutex);
|
|
|
|
return 999;
|
|
|
|
}
|
|
|
|
else continue;
|
|
|
|
pthread_mutex_unlock(&pwl_mutex);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
pthread_mutex_unlock(&pwl_mutex);
|
|
|
|
return 5;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
struct auth authfuncs[] = {
|
|
|
|
{authfuncs+1, NULL, NULL, ""},
|
|
|
|
{authfuncs+2, ipauth, NULL, "iponly"},
|
|
|
|
{authfuncs+3, userauth, checkACL, "useronly"},
|
2014-05-12 04:21:22 +08:00
|
|
|
{authfuncs+4, dnsauth, checkACL, "dnsname"},
|
|
|
|
{authfuncs+5, strongauth, checkACL, "strong"},
|
|
|
|
{authfuncs+6, cacheauth, checkACL, "cache"},
|
|
|
|
{authfuncs+7, NULL, NULL, "none"},
|
2014-04-08 17:03:21 +08:00
|
|
|
|
|
|
|
{NULL, NULL, NULL, ""}
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
|
2014-12-17 05:20:21 +08:00
|
|
|
struct hashtable dns_table = {0, 4, {0,0,0,0}, NULL, NULL, NULL};
|
|
|
|
struct hashtable dns6_table = {0, 16, {0,0,0,0}, NULL, NULL, NULL};
|
2014-04-08 17:03:21 +08:00
|
|
|
|
|
|
|
|
2014-12-14 05:46:03 +08:00
|
|
|
void nametohash(const unsigned char * name, unsigned char *hash, unsigned char *rnd){
|
|
|
|
unsigned i, j, k;
|
2014-12-14 10:33:08 +08:00
|
|
|
memcpy(hash, rnd, sizeof(unsigned)*4);
|
2014-12-14 05:46:03 +08:00
|
|
|
for(i=0, j=0, k=0; name[j]; j++){
|
|
|
|
hash[i] += (toupper(name[j]) - 32)+rnd[((toupper(name[j]))*29277+rnd[(k+j+i)%16]+k+j+i)%16];
|
|
|
|
if(++i == sizeof(unsigned)*4) {
|
|
|
|
i = 0;
|
|
|
|
k++;
|
|
|
|
}
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
unsigned hashindex(struct hashtable *ht, const unsigned char* hash){
|
|
|
|
unsigned t1, t2, t3, t4;
|
|
|
|
t1 = *(unsigned *)hash;
|
|
|
|
t2 = *(unsigned *)(hash + sizeof(unsigned));
|
|
|
|
t3 = *(unsigned *)(hash + (2*sizeof(unsigned)));
|
|
|
|
t4 = *(unsigned *)(hash + (3*sizeof(unsigned)));
|
|
|
|
return (t1 + (t2 * 7) + (t3 * 17) + (t4 * 29) ) % (ht->hashsize >> 2);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void destroyhashtable(struct hashtable *ht){
|
|
|
|
pthread_mutex_lock(&hash_mutex);
|
|
|
|
if(ht->hashtable){
|
|
|
|
myfree(ht->hashtable);
|
|
|
|
ht->hashtable = NULL;
|
|
|
|
}
|
|
|
|
if(ht->hashvalues){
|
|
|
|
myfree(ht->hashvalues);
|
|
|
|
ht->hashvalues = NULL;
|
|
|
|
}
|
|
|
|
ht->hashsize = 0;
|
|
|
|
pthread_mutex_unlock(&hash_mutex);
|
|
|
|
}
|
|
|
|
|
2014-12-17 05:20:21 +08:00
|
|
|
#define hvalue(I) ((struct hashentry *)((char *)ht->hashvalues + (I)*(sizeof(struct hashentry) + ht->recsize - 4)))
|
2014-04-08 17:03:21 +08:00
|
|
|
int inithashtable(struct hashtable *ht, unsigned nhashsize){
|
|
|
|
unsigned i;
|
2014-12-14 05:46:03 +08:00
|
|
|
clock_t c;
|
|
|
|
|
|
|
|
|
|
|
|
#ifdef _WIN32
|
|
|
|
struct timeb tb;
|
|
|
|
|
|
|
|
ftime(&tb);
|
|
|
|
|
|
|
|
#else
|
|
|
|
struct timeval tb;
|
|
|
|
struct timezone tz;
|
|
|
|
gettimeofday(&tb, &tz);
|
|
|
|
#endif
|
|
|
|
c = clock();
|
2014-04-08 17:03:21 +08:00
|
|
|
|
|
|
|
if(nhashsize<4) return 1;
|
2015-05-04 22:39:24 +08:00
|
|
|
pthread_mutex_lock(&hash_mutex);
|
2014-04-08 17:03:21 +08:00
|
|
|
if(ht->hashtable){
|
|
|
|
myfree(ht->hashtable);
|
|
|
|
ht->hashtable = NULL;
|
|
|
|
}
|
|
|
|
if(ht->hashvalues){
|
|
|
|
myfree(ht->hashvalues);
|
|
|
|
ht->hashvalues = NULL;
|
|
|
|
}
|
|
|
|
ht->hashsize = 0;
|
2014-12-14 10:33:08 +08:00
|
|
|
if(!(ht->hashtable = myalloc((nhashsize>>2) * sizeof(struct hashentry *)))){
|
2015-05-04 22:39:24 +08:00
|
|
|
pthread_mutex_unlock(&hash_mutex);
|
2014-04-08 17:03:21 +08:00
|
|
|
return 2;
|
|
|
|
}
|
2014-12-14 10:33:08 +08:00
|
|
|
if(!(ht->hashvalues = myalloc(nhashsize * (sizeof(struct hashentry) + (ht->recsize-4))))){
|
2014-04-08 17:03:21 +08:00
|
|
|
myfree(ht->hashtable);
|
|
|
|
ht->hashtable = NULL;
|
2015-05-04 22:39:24 +08:00
|
|
|
pthread_mutex_unlock(&hash_mutex);
|
2014-04-08 17:03:21 +08:00
|
|
|
return 3;
|
|
|
|
}
|
|
|
|
ht->hashsize = nhashsize;
|
2014-12-14 05:46:03 +08:00
|
|
|
ht->rnd[0] = myrand(&tb, sizeof(tb));
|
|
|
|
ht->rnd[1] = myrand(ht->hashtable, sizeof(ht->hashtable));
|
|
|
|
ht->rnd[2] = myrand(&c, sizeof(c));
|
|
|
|
ht->rnd[3] = myrand(ht->hashvalues,sizeof(ht->hashvalues));
|
2014-04-08 17:03:21 +08:00
|
|
|
memset(ht->hashtable, 0, (ht->hashsize>>2) * sizeof(struct hashentry *));
|
2014-12-14 10:33:08 +08:00
|
|
|
memset(ht->hashvalues, 0, ht->hashsize * (sizeof(struct hashentry) + ht->recsize -4));
|
2014-12-14 05:46:03 +08:00
|
|
|
|
2014-04-08 17:03:21 +08:00
|
|
|
for(i = 0; i< (ht->hashsize - 1); i++) {
|
2014-12-14 10:33:08 +08:00
|
|
|
hvalue(i)->next = hvalue(i+1);
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
ht->hashempty = ht->hashvalues;
|
2015-05-04 22:39:24 +08:00
|
|
|
pthread_mutex_unlock(&hash_mutex);
|
2014-04-08 17:03:21 +08:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-12-14 10:33:08 +08:00
|
|
|
void hashadd(struct hashtable *ht, const unsigned char* name, unsigned char* value, time_t expires){
|
|
|
|
struct hashentry * hen, *he;
|
|
|
|
struct hashentry ** hep;
|
2014-04-08 17:03:21 +08:00
|
|
|
|
|
|
|
unsigned index;
|
|
|
|
|
|
|
|
pthread_mutex_lock(&hash_mutex);
|
2015-05-04 23:06:07 +08:00
|
|
|
if(!ht||!value||!name||!ht->hashtable||!ht->hashempty) {
|
|
|
|
pthread_mutex_unlock(&hash_mutex);
|
|
|
|
return;
|
|
|
|
}
|
2014-12-14 10:33:08 +08:00
|
|
|
hen = ht->hashempty;
|
2014-04-08 17:03:21 +08:00
|
|
|
ht->hashempty = ht->hashempty->next;
|
2014-12-14 10:33:08 +08:00
|
|
|
nametohash(name, hen->hash, (unsigned char *)ht->rnd);
|
|
|
|
memcpy(hen->value, value, ht->recsize);
|
|
|
|
hen->expires = expires;
|
|
|
|
hen->next = NULL;
|
|
|
|
index = hashindex(ht, hen->hash);
|
|
|
|
|
|
|
|
for(hep = ht->hashtable + index; (he = *hep)!=NULL; ){
|
|
|
|
if(he->expires < conf.time || !memcmp(hen->hash, he->hash, sizeof(he->hash))) {
|
|
|
|
(*hep) = he->next;
|
|
|
|
he->expires = 0;
|
|
|
|
he->next = ht->hashempty;
|
|
|
|
ht->hashempty = he;
|
|
|
|
}
|
|
|
|
else hep=&(he->next);
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
2014-12-14 10:33:08 +08:00
|
|
|
hen->next = ht->hashtable[index];
|
|
|
|
ht->hashtable[index] = hen;
|
2014-04-08 17:03:21 +08:00
|
|
|
pthread_mutex_unlock(&hash_mutex);
|
|
|
|
}
|
|
|
|
|
2014-12-14 10:33:08 +08:00
|
|
|
unsigned long hashresolv(struct hashtable *ht, const unsigned char* name, unsigned char* value, unsigned *ttl){
|
2014-04-08 17:03:21 +08:00
|
|
|
unsigned char hash[sizeof(unsigned)*4];
|
|
|
|
struct hashentry ** hep;
|
|
|
|
struct hashentry *he;
|
|
|
|
unsigned index;
|
|
|
|
|
2015-05-04 22:39:24 +08:00
|
|
|
pthread_mutex_lock(&hash_mutex);
|
|
|
|
if(!ht || !ht->hashtable || !name) {
|
|
|
|
pthread_mutex_unlock(&hash_mutex);
|
|
|
|
return 0;
|
|
|
|
}
|
2014-12-14 05:46:03 +08:00
|
|
|
nametohash(name, hash, (unsigned char *)ht->rnd);
|
2014-04-08 17:03:21 +08:00
|
|
|
index = hashindex(ht, hash);
|
|
|
|
for(hep = ht->hashtable + index; (he = *hep)!=NULL; ){
|
2014-12-14 10:33:08 +08:00
|
|
|
if(he->expires < conf.time) {
|
2014-04-08 17:03:21 +08:00
|
|
|
(*hep) = he->next;
|
|
|
|
he->expires = 0;
|
|
|
|
he->next = ht->hashempty;
|
|
|
|
ht->hashempty = he;
|
|
|
|
}
|
|
|
|
else if(!memcmp(hash, he->hash, sizeof(unsigned)*4)){
|
2014-12-14 10:33:08 +08:00
|
|
|
if(ttl) *ttl = (unsigned)(he->expires - conf.time);
|
|
|
|
memcpy(value, he->value, ht->recsize);
|
2014-12-17 05:20:21 +08:00
|
|
|
pthread_mutex_unlock(&hash_mutex);
|
2014-12-14 10:33:08 +08:00
|
|
|
return 1;
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
else hep=&(he->next);
|
|
|
|
}
|
|
|
|
pthread_mutex_unlock(&hash_mutex);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-12-13 08:56:01 +08:00
|
|
|
struct nserver nservers[MAXNSERVERS] = {{{0},0}, {{0},0}, {{0},0}, {{0},0}, {{0},0}};
|
2014-12-14 03:56:31 +08:00
|
|
|
struct nserver authnserver;
|
2014-04-08 17:03:21 +08:00
|
|
|
|
|
|
|
|
2014-12-14 10:33:08 +08:00
|
|
|
unsigned long udpresolve(int af, unsigned char * name, unsigned char * value, unsigned *retttl, struct clientparam* param, int makeauth){
|
2014-04-08 17:03:21 +08:00
|
|
|
|
2014-12-13 08:56:01 +08:00
|
|
|
int i,n;
|
2014-04-08 17:03:21 +08:00
|
|
|
unsigned long retval;
|
|
|
|
|
2014-12-14 10:33:08 +08:00
|
|
|
if((af == AF_INET) && (retval = hashresolv(&dns_table, name, value, retttl))) {
|
|
|
|
return retval;
|
|
|
|
}
|
|
|
|
if((af == AF_INET6) && (retval = hashresolv(&dns6_table, name, value, retttl))) {
|
2014-04-08 17:03:21 +08:00
|
|
|
return retval;
|
|
|
|
}
|
2014-12-14 03:56:31 +08:00
|
|
|
n = (makeauth && !SAISNULL(&authnserver.addr))? 1 : numservers;
|
2014-12-13 08:56:01 +08:00
|
|
|
for(i=0; i<n; i++){
|
2014-12-11 08:13:16 +08:00
|
|
|
unsigned short nq, na;
|
2014-12-11 08:06:34 +08:00
|
|
|
unsigned char b[4098], *buf, *s1, *s2;
|
2014-04-08 17:03:21 +08:00
|
|
|
int j, k, len, flen;
|
|
|
|
SOCKET sock;
|
|
|
|
unsigned ttl;
|
2014-12-13 08:56:01 +08:00
|
|
|
#ifndef NOIPV6
|
|
|
|
struct sockaddr_in6 addr;
|
2016-02-05 23:31:17 +08:00
|
|
|
struct sockaddr_in6 *sinsr, *sinsl;
|
2014-12-13 08:56:01 +08:00
|
|
|
#else
|
|
|
|
struct sockaddr_in addr;
|
2016-02-05 23:31:17 +08:00
|
|
|
struct sockaddr_in *sinsr, *sinsl;
|
2014-12-13 08:56:01 +08:00
|
|
|
#endif
|
2014-12-11 08:06:34 +08:00
|
|
|
int usetcp = 0;
|
2014-12-11 08:49:26 +08:00
|
|
|
unsigned short serial = 1;
|
2014-04-08 17:03:21 +08:00
|
|
|
|
2014-12-11 08:06:34 +08:00
|
|
|
buf = b+2;
|
|
|
|
|
2016-02-05 23:31:17 +08:00
|
|
|
sinsl = (param && !makeauth)? ¶m->sinsl : &addr;
|
|
|
|
sinsr = (param && !makeauth)? ¶m->sinsr : &addr;
|
2014-12-13 08:56:01 +08:00
|
|
|
memset(sinsl, 0, sizeof(addr));
|
|
|
|
memset(sinsr, 0, sizeof(addr));
|
2014-04-08 17:03:21 +08:00
|
|
|
|
|
|
|
|
2014-12-14 03:56:31 +08:00
|
|
|
if(makeauth && !SAISNULL(&authnserver.addr)){
|
|
|
|
usetcp = authnserver.usetcp;
|
|
|
|
*SAFAMILY(sinsl) = *SAFAMILY(&authnserver.addr);
|
2014-12-11 08:06:34 +08:00
|
|
|
}
|
|
|
|
else {
|
|
|
|
usetcp = nservers[i].usetcp;
|
2014-12-13 08:56:01 +08:00
|
|
|
*SAFAMILY(sinsl) = *SAFAMILY(&nservers[i].addr);
|
2014-12-11 08:06:34 +08:00
|
|
|
}
|
2014-12-14 03:56:31 +08:00
|
|
|
if((sock=so._socket(SASOCK(sinsl), usetcp?SOCK_STREAM:SOCK_DGRAM, usetcp?IPPROTO_TCP:IPPROTO_UDP)) == INVALID_SOCKET) break;
|
2016-02-05 23:31:17 +08:00
|
|
|
if(so._bind(sock,(struct sockaddr *)sinsl,SASIZE(sinsl))){
|
2014-04-08 17:03:21 +08:00
|
|
|
so._shutdown(sock, SHUT_RDWR);
|
|
|
|
so._closesocket(sock);
|
|
|
|
break;
|
|
|
|
}
|
2014-12-14 03:56:31 +08:00
|
|
|
if(makeauth && !SAISNULL(&authnserver.addr)){
|
2016-02-05 23:31:17 +08:00
|
|
|
*sinsr = authnserver.addr;
|
2014-12-11 08:06:34 +08:00
|
|
|
}
|
|
|
|
else {
|
2016-02-05 23:31:17 +08:00
|
|
|
*sinsr = nservers[i].addr;
|
2014-12-11 08:06:34 +08:00
|
|
|
}
|
|
|
|
if(usetcp){
|
2016-02-05 23:31:17 +08:00
|
|
|
if(so._connect(sock,(struct sockaddr *)sinsr,SASIZE(sinsr))) {
|
2014-12-11 08:06:34 +08:00
|
|
|
so._shutdown(sock, SHUT_RDWR);
|
|
|
|
so._closesocket(sock);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
2014-04-08 17:03:21 +08:00
|
|
|
len = (int)strlen((char *)name);
|
2014-12-11 08:49:26 +08:00
|
|
|
|
|
|
|
serial = myrand(name,len);
|
|
|
|
*(unsigned short*)buf = serial; /* query id */
|
2014-04-08 17:03:21 +08:00
|
|
|
buf[2] = 1; /* recursive */
|
|
|
|
buf[3] = 0;
|
|
|
|
buf[4] = 0;
|
|
|
|
buf[5] = 1; /* 1 request */
|
|
|
|
buf[6] = buf[7] = 0; /* no replies */
|
|
|
|
buf[8] = buf[9] = 0; /* no ns count */
|
|
|
|
buf[10] = buf[11] = 0; /* no additional */
|
|
|
|
if(len > 255) {
|
|
|
|
len = 255;
|
|
|
|
}
|
|
|
|
memcpy(buf + 13, name, len);
|
|
|
|
len += 13;
|
|
|
|
buf[len] = 0;
|
|
|
|
for(s2 = buf + 12; (s1 = (unsigned char *)strchr((char *)s2 + 1, '.')); s2 = s1)*s2 = (unsigned char)((s1 - s2) - 1);
|
|
|
|
*s2 = (len - (int)(s2 - buf)) - 1;
|
|
|
|
len++;
|
|
|
|
buf[len++] = 0;
|
2014-12-14 10:33:08 +08:00
|
|
|
buf[len++] = (makeauth == 1)? 0x0c : (af==AF_INET6? 0x1c:0x01); /* PTR:host address */
|
2014-04-08 17:03:21 +08:00
|
|
|
buf[len++] = 0;
|
|
|
|
buf[len++] = 1; /* INET */
|
2014-12-11 08:06:34 +08:00
|
|
|
if(usetcp){
|
|
|
|
buf-=2;
|
|
|
|
*(unsigned short*)buf = htons(len);
|
|
|
|
len+=2;
|
|
|
|
}
|
|
|
|
|
2016-02-05 23:31:17 +08:00
|
|
|
if(socksendto(sock, (struct sockaddr *)sinsr, buf, len, conf.timeouts[SINGLEBYTE_L]*1000) != len){
|
2014-04-08 17:03:21 +08:00
|
|
|
so._shutdown(sock, SHUT_RDWR);
|
|
|
|
so._closesocket(sock);
|
|
|
|
continue;
|
|
|
|
}
|
2014-04-10 07:34:59 +08:00
|
|
|
if(param) param->statscli64 += len;
|
2016-02-05 23:31:17 +08:00
|
|
|
len = sockrecvfrom(sock, (struct sockaddr *)sinsr, buf, 4096, conf.timeouts[DNS_TO]*1000);
|
2014-04-08 17:03:21 +08:00
|
|
|
so._shutdown(sock, SHUT_RDWR);
|
|
|
|
so._closesocket(sock);
|
2014-12-11 08:06:34 +08:00
|
|
|
if(len <= 13) {
|
|
|
|
continue;
|
|
|
|
}
|
2014-04-10 07:34:59 +08:00
|
|
|
if(param) param->statssrv64 += len;
|
2014-12-11 08:06:34 +08:00
|
|
|
if(usetcp){
|
2014-12-13 08:56:01 +08:00
|
|
|
unsigned short us;
|
|
|
|
us = ntohs(*(unsigned short*)buf);
|
2014-12-11 08:06:34 +08:00
|
|
|
len-=2;
|
2014-12-13 08:56:01 +08:00
|
|
|
buf+=2;
|
2016-02-05 23:31:17 +08:00
|
|
|
if(us > 4096 || us < len || (us > len && sockrecvfrom(sock, (struct sockaddr *)sinsr, buf+len, us-len, conf.timeouts[DNS_TO]*1000) != us-len)) {
|
2014-12-13 08:56:01 +08:00
|
|
|
continue;
|
|
|
|
}
|
2014-12-11 08:06:34 +08:00
|
|
|
}
|
2014-12-11 08:49:26 +08:00
|
|
|
if(*(unsigned short *)buf != serial)continue;
|
2014-04-08 17:03:21 +08:00
|
|
|
if((na = buf[7] + (((unsigned short)buf[6])<<8)) < 1) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
nq = buf[5] + (((unsigned short)buf[4])<<8);
|
|
|
|
if (nq != 1) {
|
|
|
|
continue; /* we did only 1 request */
|
|
|
|
}
|
|
|
|
for(k = 13; k<len && buf[k]; k++) {
|
|
|
|
}
|
|
|
|
k++;
|
|
|
|
if( (k+4) >= len) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
k += 4;
|
|
|
|
if(na > 255) na = 255; /* somebody is very evil */
|
|
|
|
for (j = 0; j < na; j++) { /* now there should be answers */
|
2016-12-11 08:25:18 +08:00
|
|
|
while(buf[k] < 192 && buf[k] !=0 && (k+buf[k]) < len) k+= buf[k];
|
|
|
|
if(!buf[k]) k--;
|
2014-12-14 10:33:08 +08:00
|
|
|
if((k+(af == AF_INET6?28:16)) > len) {
|
2014-04-08 17:03:21 +08:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
flen = buf[k+11] + (((unsigned short)buf[k+10])<<8);
|
2014-12-14 10:33:08 +08:00
|
|
|
if((k+12+flen) > len) {
|
|
|
|
break;
|
|
|
|
}
|
2014-04-08 17:03:21 +08:00
|
|
|
if(makeauth != 1){
|
2014-12-14 10:33:08 +08:00
|
|
|
if(buf[k+2] != 0 || buf[k+3] != (af == AF_INET6?0x1c:0x1) || flen != (af == AF_INET6?16:4)) {
|
2014-04-08 17:03:21 +08:00
|
|
|
k+= (12 + flen);
|
|
|
|
continue; /* we need A IPv4 */
|
|
|
|
}
|
|
|
|
ttl = ntohl(*(unsigned long *)(buf + k + 6));
|
2014-12-14 10:33:08 +08:00
|
|
|
memcpy(value, buf + k + 12, af == AF_INET6? 16:4);
|
2014-12-17 05:20:21 +08:00
|
|
|
if(ttl < 60 || ttl > (3600*12)) ttl = 300;
|
2016-09-07 23:35:01 +08:00
|
|
|
hashadd(af == AF_INET6?&dns6_table:&dns_table, name, value, conf.time+ttl);
|
2014-04-08 17:03:21 +08:00
|
|
|
if(retttl) *retttl = ttl;
|
2014-12-14 10:33:08 +08:00
|
|
|
return 1;
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
else {
|
|
|
|
|
|
|
|
if(buf[k+2] != 0 || buf[k+3] != 0x0c) {
|
|
|
|
k+= (12 + flen);
|
|
|
|
continue; /* we need A PTR */
|
|
|
|
}
|
|
|
|
for (s2 = buf + k + 12; s2 < (buf + k + 12 + len) && *s2; ){
|
|
|
|
s1 = s2 + ((unsigned)*s2) + 1;
|
|
|
|
*s2 = '.';
|
|
|
|
s2 = s1;
|
|
|
|
}
|
|
|
|
*s2 = 0;
|
|
|
|
if(param->username)myfree(param->username);
|
2016-02-16 20:29:51 +08:00
|
|
|
param->username = (unsigned char *)mystrdup ((char *)buf + k + 13);
|
2014-04-08 17:03:21 +08:00
|
|
|
|
2014-12-14 10:33:08 +08:00
|
|
|
return udpresolve(af,param->username, value, NULL, NULL, 2);
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-12-14 10:33:08 +08:00
|
|
|
unsigned long myresolver(int af, unsigned char * name, unsigned char * value){
|
|
|
|
return udpresolve(af, name, value, NULL, NULL, 0);
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
|
2014-12-14 10:33:08 +08:00
|
|
|
unsigned long fakeresolver (int af, unsigned char *name, unsigned char * value){
|
|
|
|
memset(value, 0, af == AF_INET6? 16 : 4);
|
|
|
|
if(af == AF_INET6){
|
|
|
|
memset(value, 0, 16);
|
|
|
|
value[15] = 2;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
value[0] = 127;
|
|
|
|
value[1] = 0;
|
|
|
|
value[2] = 0;
|
|
|
|
value[3] = 2;
|
|
|
|
}
|
|
|
|
return 1;
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
#ifndef NOODBC
|
|
|
|
|
|
|
|
SQLHENV henv = NULL;
|
|
|
|
SQLHSTMT hstmt = NULL;
|
|
|
|
SQLHDBC hdbc = NULL;
|
|
|
|
char * sqlstring = NULL;
|
|
|
|
|
|
|
|
|
|
|
|
void close_sql(){
|
|
|
|
if(hstmt) {
|
|
|
|
SQLFreeHandle(SQL_HANDLE_STMT, hstmt);
|
|
|
|
hstmt = NULL;
|
|
|
|
}
|
|
|
|
if(hdbc){
|
|
|
|
SQLDisconnect(hdbc);
|
|
|
|
SQLFreeHandle(SQL_HANDLE_DBC, hdbc);
|
|
|
|
hdbc = NULL;
|
|
|
|
}
|
|
|
|
if(henv) {
|
|
|
|
SQLFreeHandle(SQL_HANDLE_ENV, henv);
|
|
|
|
henv = NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
int attempt = 0;
|
|
|
|
time_t attempt_time = 0;
|
|
|
|
|
|
|
|
int init_sql(char * s){
|
|
|
|
SQLRETURN retcode;
|
|
|
|
char * datasource;
|
|
|
|
char * username;
|
|
|
|
char * password;
|
|
|
|
char * string;
|
|
|
|
|
|
|
|
if(!s) return 0;
|
|
|
|
if(!sqlstring || strcmp(sqlstring, s)){
|
|
|
|
string = sqlstring;
|
|
|
|
sqlstring=mystrdup(s);
|
|
|
|
if(string)myfree(string);
|
|
|
|
}
|
|
|
|
|
|
|
|
if(hstmt || hdbc || henv) close_sql();
|
|
|
|
attempt++;
|
|
|
|
attempt_time = time(0);
|
|
|
|
if(!henv){
|
|
|
|
retcode = SQLAllocHandle(SQL_HANDLE_ENV, SQL_NULL_HANDLE, &henv);
|
|
|
|
if (!henv || (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO)){
|
|
|
|
henv = NULL;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
retcode = SQLSetEnvAttr(henv, SQL_ATTR_ODBC_VERSION, (void*)SQL_OV_ODBC3, 0);
|
|
|
|
|
|
|
|
if (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO) {
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if(!hdbc){
|
|
|
|
retcode = SQLAllocHandle(SQL_HANDLE_DBC, henv, &hdbc);
|
|
|
|
if (!hdbc || (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO)) {
|
|
|
|
hdbc = NULL;
|
|
|
|
SQLFreeHandle(SQL_HANDLE_ENV, henv);
|
|
|
|
henv = NULL;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
SQLSetConnectAttr(hdbc, SQL_LOGIN_TIMEOUT, (void*)15, 0);
|
|
|
|
}
|
|
|
|
string = mystrdup(sqlstring);
|
|
|
|
if(!string) return 0;
|
|
|
|
datasource = strtok(string, ",");
|
|
|
|
username = strtok(NULL, ",");
|
|
|
|
password = strtok(NULL, ",");
|
|
|
|
|
|
|
|
|
|
|
|
/* Connect to data source */
|
|
|
|
retcode = SQLConnect(hdbc, (SQLCHAR*) datasource, (SQLSMALLINT)strlen(datasource),
|
|
|
|
(SQLCHAR*) username, (SQLSMALLINT)((username)?strlen(username):0),
|
|
|
|
(SQLCHAR*) password, (SQLSMALLINT)((password)?strlen(password):0));
|
|
|
|
|
|
|
|
myfree(string);
|
|
|
|
if (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO){
|
|
|
|
SQLFreeHandle(SQL_HANDLE_DBC, hdbc);
|
|
|
|
hdbc = NULL;
|
|
|
|
SQLFreeHandle(SQL_HANDLE_ENV, henv);
|
|
|
|
henv = NULL;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
retcode = SQLAllocHandle(SQL_HANDLE_STMT, hdbc, &hstmt);
|
|
|
|
if (retcode != SQL_SUCCESS && retcode != SQL_SUCCESS_WITH_INFO){
|
|
|
|
close_sql();
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
void sqlerr (char *buf){
|
|
|
|
if(conf.stdlog){
|
|
|
|
fprintf(conf.stdlog, "%s\n", buf);
|
|
|
|
fflush(conf.stdlog);
|
|
|
|
}
|
2015-12-28 00:27:17 +08:00
|
|
|
pthread_mutex_unlock(&log_mutex);
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
void logsql(struct clientparam * param, const unsigned char *s) {
|
|
|
|
SQLRETURN ret;
|
|
|
|
int len;
|
|
|
|
|
|
|
|
|
|
|
|
if(param->nolog) return;
|
2015-12-28 00:27:17 +08:00
|
|
|
pthread_mutex_lock(&log_mutex);
|
2016-02-16 20:29:51 +08:00
|
|
|
len = dobuf(param, tmpbuf, s, (unsigned char *)"\'");
|
2014-04-08 17:03:21 +08:00
|
|
|
|
|
|
|
if(attempt > 5){
|
|
|
|
time_t t;
|
|
|
|
|
|
|
|
t = time(0);
|
|
|
|
if (t - attempt_time < 180){
|
2016-02-16 20:29:51 +08:00
|
|
|
sqlerr((char *)tmpbuf);
|
2014-04-08 17:03:21 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if(!hstmt){
|
|
|
|
if(!init_sql(sqlstring)) {
|
2016-02-16 20:29:51 +08:00
|
|
|
sqlerr((char *)tmpbuf);
|
2014-04-08 17:03:21 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if(hstmt){
|
2015-12-28 00:27:17 +08:00
|
|
|
ret = SQLExecDirect(hstmt, (SQLCHAR *)tmpbuf, (SQLINTEGER)len);
|
2014-04-08 17:03:21 +08:00
|
|
|
if(ret != SQL_SUCCESS && ret != SQL_SUCCESS_WITH_INFO){
|
|
|
|
close_sql();
|
|
|
|
if(!init_sql(sqlstring)){
|
2016-02-16 20:29:51 +08:00
|
|
|
sqlerr((char *)tmpbuf);
|
2014-04-08 17:03:21 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
if(hstmt) {
|
2015-12-28 00:27:17 +08:00
|
|
|
ret = SQLExecDirect(hstmt, (SQLCHAR *)tmpbuf, (SQLINTEGER)len);
|
2014-04-08 17:03:21 +08:00
|
|
|
if(ret != SQL_SUCCESS && ret != SQL_SUCCESS_WITH_INFO){
|
2016-02-16 20:29:51 +08:00
|
|
|
sqlerr((char *)tmpbuf);
|
2014-04-08 17:03:21 +08:00
|
|
|
return;
|
|
|
|
}
|
|
|
|
attempt = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
attempt = 0;
|
|
|
|
}
|
2015-12-28 00:27:17 +08:00
|
|
|
pthread_mutex_unlock(&log_mutex);
|
2014-04-08 17:03:21 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
2014-12-14 03:56:31 +08:00
|
|
|
|