/* qADSL final release 1.0 (010403)
 * Copyright (c) 2001 Jakob "kUbA" Stasilowicz (qba@linux.nu)
 *
 * 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 2
 * 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, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <unistd.h>
#include <sys/time.h>
#include <signal.h>
#include <ctype.h>
#include <syslog.h>
#include <time.h>

#define VERSION "1.0"
#define PORT 80
#define MAXDATASIZE 4091
#define MAXLEN 256

#if defined(__OpenBSD__)
#define TMP_PID "/var/run/qadsl.pid"
#else
#define TMP_PID "/tmp/qadsl.pid"
#endif

int login_sw=0,sockfd,i;
char send_msg[MAXDATASIZE], get_msg[MAXDATASIZE];
char user[MAXLEN], pass[MAXLEN], serv[MAXLEN], deamon_s[MAXLEN], deamon_t[MAXLEN], deamon_d[MAXLEN];

char *conf="/etc/qadsl.conf";

void preconnect(), telia_logout(), read_conf(), create_pid(), kill_pid(), log_login(), quit_deamon();
void send_func(const char *send_msg_to_server);
void url_encode(const char *, char *);
int telia_login();

struct hostent *he;
struct sockaddr_in address;

int main(int argc, char *argv[])
{
	int con_sw=0;
	int dologin=0;
	int dologout=0;

        struct tm *timep;
        time_t the_time;

        (void) time(&the_time);
        timep=localtime(&the_time);

	printf("qADSL %s av kUbA(qba@linux.nu) - http://qba.mine.nu\n",VERSION);

	if(argc < 2) {
		fprintf(stderr,"\tUsage: %s -h (help) -c <config_file> -l (login) | -o (logout)\n",argv[0]);
		exit(0);
	}

	for(i = 0; i < argc; i++) {
		if((strncmp("-l",argv[i],2))==0) {
			if(!dologout) dologin=1;
		} 

		if((strncmp("-c",argv[i],2))==0) {
			conf=argv[i+1];
		}

		if((strncmp("-o",argv[i],2))==0) {
			if(!dologin) dologout=1;
		}

	        if((strncmp("-h",argv[i],2))==0) {
		        fprintf(stderr,"\tUsage: %s -h (help) -c <config_file> -l (login) | -o (logout)\n ",argv[0]);
        		exit(0);
                }

	}
	
	if((!dologin) && (!dologout)) {
		fprintf(stderr,"\tUsage: %s -h (help) -c <config_file> -l (login) | -o (logout)\n ",argv[0]);
		exit(0);
        }
			
	read_conf();

	if ((he=gethostbyname(serv)) == NULL) {   
		fprintf(stderr,"\tKunde inte få fram hostname !\n\tÄr du säker på att %s är din login server ?\n", serv);
		exit(1);
	}       
      
	if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
		perror("socket");
		exit(1);
	}         

	address.sin_family = PF_INET;      
	address.sin_port = htons(PORT);    
	address.sin_addr = *((struct in_addr *)he->h_addr);
	memset(&(address.sin_zero),0,sizeof(address.sin_zero));   

	if (connect(sockfd, (struct sockaddr *)&address, sizeof(address)) == -1) {
		fprintf(stderr,"\tKunde inte ansluta !\n\tÄr du säker på att %s är din login server ?\n", serv);
		exit(1);
	}
   
	if(dologin) {

		preconnect();

		close(sockfd);

		if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
			perror("socket");
			exit(1);
		}         

		printf("\tAnsluter till %s..\n", serv);

		if (connect(sockfd, (struct sockaddr *)&address, sizeof(address)) == -1) {
			fprintf(stderr,"\tKunde inte ansluta !\n\tÄr du säker på att %s är din login server ?\n", serv);
			exit(1);
		}   

		printf("\tAnsluten !\n");

		log_login(1);

		close(sockfd);

	} else if(dologout) {

		telia_logout();

		close(sockfd);
		exit(0);
	}

	if((strncmp(deamon_s,"YES",3))==0) {
		kill_pid();
		if(fork() == 0) {
			printf("qADSL deamon startad med pid: [%d]\n",getpid());		
			while(1) {
				(void)signal(SIGTERM, quit_deamon);

				create_pid();
				
				sleep(60 * atoi(deamon_d));

				if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1) exit(1);

				if (connect(sockfd, (struct sockaddr *)&address, sizeof(address)) == -1) {
					for(i=0; i < 4; i++) {
						sleep(60 * 30);

						if(connect(sockfd, (struct sockaddr *)&address, sizeof(address))) {
							con_sw=1; 
							break;
						}
					}
					
				} else con_sw=1;

				if(!con_sw) {
					openlog("qADSL ", LOG_PID, LOG_USER);
					syslog(LOG_ERR,"Failed to connect to telia (%02d:%02d:%02d)\n",timep->tm_hour, timep->tm_min, timep->tm_sec);
					exit(1);
				}

				if((strncmp(deamon_t,"LOGIN",5))==0) {
					preconnect();

					if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1) exit(1);

					if (connect(sockfd, (struct sockaddr *)&address, sizeof(address)) == -1) exit(1);
					
					log_login(0);	
				}

				if((strncmp(deamon_t,"PING",4))==0) {
					sprintf(send_msg,"GET /sd/init HTTP/1.0\n" \
					"Referer: http://%s/sd/init\n" \
					"Connection: Keep-Alive\n" \
					"User-Agent: Mozilla/4.73 [en] (X11; U; Linux 2.4.0 i686)\n" \
					"Host: %s\n" \
					"Accept: image/gif, image/x-xbitmap, image/jpeg, " \
					"image/pjpeg, image/png, */*\n" \
					"Accept-Encoding: gzip\n" \
					"Accept-Language: en\n" \
					"Accept-Charset: iso-8859-1,*,utf-8\n", serv, serv);

					send_func(send_msg);

				}

			}
		} 
	}
   usleep(50000);
 return 0;
}

int telia_login(int check) 
{
	sprintf(send_msg,"POST /sd/login HTTP/1.0\n" \
	"Referer: http://%s/sd/init\n" \
	"Connection: Keep-Alive\n" \
	"User-Agent: Mozilla/4.73 [en] (X11; U; Linux 2.4.0 i686)\n" \
	"Host: %s\n" \
	"Accept: image/gif, image/x-xbitmap, image/jpeg, " \
	"image/pjpeg, image/png, */*\n" \
	"Accept-Encoding: gzip\n" \
	"Accept-Language: en\n" \
	"Accept-Charset: iso-8859-1,*,utf-8\n" \
	"Content-type: application/x-www-form-urlencoded\n" \
	"Content-length: 58\n\n" \
	"username=%s&password=%s&submitForm=Logga+in\n\n\n\n\n",serv, serv, user, pass);

	send_func(send_msg);

	if((read(sockfd, get_msg, sizeof(get_msg))) == -1) {
		fprintf(stderr,"\tKunde inte ta emot data !\n");
		exit(1);
	}

	sleep(1);

	if(strstr(get_msg, "newPane()") == NULL) {
		if(check) fprintf(stderr,"\tTelia Broadband: KUNDE INTE LOGGA IN\n");
		return 0;	
	} else {
		if(check) printf("\tTelia Broadband: INLOGGAD\n");
		login_sw=1;
		return 1;
	}
}

void preconnect() 
{
	sprintf(send_msg, "GET /sd/init HTTP/1.0\n" \
	"Referer: http://%s/sd/init\n" \
	"Connection: Keep-Alive\n" \
	"User-Agent: Mozilla/4.73 [en] (X11; U; Linux 2.4.0 i686)\n" \
	"Host: %s\n" \
	"Accept: image/gif, image/x-xbitmap, image/jpeg, " \
	"image/pjpeg, image/png, */*\n" \
	"Accept-Encoding: gzip\n" \
	"Accept-Language: en\n" \
	"Accept-Charset: iso-8859-1,*,utf-8\n\n", serv, serv);

	send_func(send_msg);	
}

void telia_logout() 
{	
	fd_set check_sockfd;
	struct timeval c_tv;
	int retval;

	FD_ZERO(&check_sockfd);
	FD_SET(sockfd, &check_sockfd);

	c_tv.tv_sec = 1;
	c_tv.tv_usec = 0;

	sprintf(send_msg, "GET /sd/logout HTTP/1.0\n" \
	"Referer: http://%s/sd/init\n" \
	"Connection: Keep-Alive\n" \
	"User-Agent: Mozilla/4.73 [en] (X11; U; Linux 2.4.0 i686)\n" \
	"Host: %s\n" \
	"Accept: image/gif, image/x-xbitmap, image/jpeg, " \
	"image/pjpeg, image/png, */*\n" \
	"Accept-Encoding: gzip\n" \
	"Accept-Language: en\n" \
	"Accept-Charset: iso-8859-1,*,utf-8\n\n", serv, serv);

	send_func(send_msg);

	sleep(1);

	if((read(sockfd, get_msg, sizeof(get_msg))) == -1) {
		fprintf(stderr,"\tKunde inte ta emot data !\n");
		exit(1);
	}

	if((retval = select(1, &check_sockfd, NULL, NULL, &c_tv))== -1) {
		fprintf(stderr,"\t/sd/logout svarar inte, kunde inte logga ut !\n");
	}

	if(strstr(get_msg, "avslutat") == NULL) {
		fprintf(stderr,"\tTelia Broadband: KUNDE INTE LOGGA UT \n");
		exit(1);
	}

	printf("\tTelia Broadband: UTLOGGAD\n");
}


void send_func(const char *send_msg_to_server) 
{
	if((send(sockfd, send_msg_to_server, strlen(send_msg_to_server), 0)) == -1) {
		perror("send");
		exit(1);
	}
}

void create_pid()
{
	FILE *pidfd;
	
	if((pidfd=fopen(TMP_PID,"w"))==NULL) {
		fprintf(stderr,"Kunde inte skapa PID fil !\n");
		exit(1);
	}

	fprintf(pidfd,"%d",getpid());

	fclose(pidfd);
}

void quit_deamon()
{
	fprintf(stderr,"qADSL[%d]: Got SIGTERM signal, quiting !\n",getpid());
	exit(0);
}

void kill_pid()
{
	FILE *pidfd;
	char input[MAXLEN];

	if((pidfd=fopen(TMP_PID,"r"))!=NULL) {
		while((fgets(input, sizeof(input), pidfd))!=NULL) {
			kill(atoi(input),SIGTERM);
		}
	}
}
	
void read_conf()
{
	FILE *fd;
	char input[MAXLEN], tmp[MAXLEN], *p;
	int user_sw=0, host_sw=0, pass_sw=0, dea_s_sw=0, dea_t_sw=0, dea_d_sw=0;

	if((fd=fopen(conf,"r"))==NULL) {
		fprintf(stderr,"Kunde inte öppna konfigurations fil !\n");
		exit(1);
	}

	while((fgets(input, sizeof(input), fd))!= NULL) {
		if ((p = strchr(input, '\n')))
		*p = '\0';

		if((strncmp("SERV=",input,5))== 0) {
			strcpy(serv,input+5);
			host_sw=1; 
		}

		if((strncmp("USER=",input, 5))== 0) {
			strcpy(tmp,input+5);
			url_encode(tmp,user);
			user_sw=1; 
		}	

		if((strncmp("PASS=",input, 5))== 0) {
			strcpy(tmp,input+5);
			url_encode(tmp,pass);
			pass_sw=1; 
		}

		if((strncmp("DEAMON_S=",input, 9))== 0) {
			strcpy(deamon_s,input+9);
			dea_s_sw=1; 
		}

		if((strncmp("DEAMON_T=",input, 9))== 0) {
			strcpy(deamon_t,input+9);
			dea_t_sw=1; 
		}

		if((strncmp("DEAMON_D=",input, 9))== 0) {
			strcpy(deamon_d,input+9);
			dea_d_sw=1; 
		}

	}

	if((!user_sw || !pass_sw || !host_sw || !dea_s_sw || !dea_t_sw || !dea_d_sw)) {
		printf("Fel i konfigurations filen %s\n",conf);

		if(!user_sw) fprintf(stderr,"\tKunde inte hitta användarnamn i konfigurations fil !\n"); exit(1); 
		if(!pass_sw) fprintf(stderr,"\tKunde inte hitta lösenord i konfigurations fil !\n"); exit(1); 
		if(!host_sw) fprintf(stderr,"\tKunde inte hitta servernamn i konfigurations fil !\n"); exit(1); 
		if(!dea_s_sw) fprintf(stderr,"\tKunde inte hitta deamon startup choice i konfigurations fil !\n"); exit(1);
		if(!dea_t_sw) fprintf(stderr,"\tKunde inte hitta deamon type (ping/connect) i konfigurations fil !\n"); exit(1);
		if(!dea_d_sw) fprintf(stderr,"\tKunde inte hitta deamon delay i konfigurations fil !\n"); exit(1);

	}
}

void url_encode(const char *in, char *out)
{
	for(;*in;++in) {
		if(isalnum(*in)) {
			*out = *in;
			++out;
		} else {
			sprintf(out, "%%%02x", *in);
			out += 3;
		}
	}
	*out = '\0';
}

void log_login(int check)
{
        struct tm *timep;
        time_t the_time;

        (void) time(&the_time);
        timep=localtime(&the_time);
	
	if(check) {	
		if(!telia_login(1)) {
			for(i=0; i < 3; i++) {
				close(sockfd);

				if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
					perror("socket");
					exit(1);
				}

				if (connect(sockfd, (struct sockaddr *)&address, sizeof(address)) == -1) {
					fprintf(stderr,"\tKunde inte ansluta !\n\tÄr du säker på att %s är din login server ?\n", serv);
					exit(1);
				}
	
				sleep(5);
		
                                if(telia_login(1)) {
                                        login_sw=1;
                                        break; 
                                }
			}
		}
	} else {
		if(!telia_login(0)) {
			for(i=0; i < 3; i++) {
				close(sockfd);

				if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
					perror("socket");
					exit(1);
				}

				if (connect(sockfd, (struct sockaddr *)&address, sizeof(address)) == -1) {
					exit(1);
				} 

				sleep(10);

				if(telia_login(0)) {
					login_sw=1;
					break;
				}
 
			}
		}
	}

	if(!login_sw) {
		if(check) fprintf(stderr,"\tTelia Broadband: AVBRYTER !\n");
		openlog("qADSL ", LOG_PID, LOG_USER);
		syslog(LOG_ERR,"Failed to login on telia (%02d:%02d:%02d)\n",timep->tm_hour, timep->tm_min, timep->tm_sec);
		exit(1);
	}
}