WebSVN – Kolibri OS – Diff – /programs/network/whois/whois.c

 WHOIS port for KolibriOS (Adapted by turbocat2001).
 The main code is taken from the site:
 https://www.binarytides.com/whois-client-code-in-c-with-linux-sockets/
 */
+#include
+int errno;
 #include
-#include
 #include
 #include
 #include
 #include
-#include
-#define stdout (FILE*)0
+#include
 FILE *out=stdout;
 #ifdef DEBUG
 FILE *out=stderr;
-#endif
-#define EXIT_SUCCESS 0
-#define EXIT_FAILURE 1
-#define perror(STR) debug_printf(STR);
-int unix_fprintf(FILE* file, const char* format, ...)
-{
-    va_list	arg;
-    va_start (arg, format);
-    if(file==stdout){
-		return vprintf(format, arg);
-	}else if(file==stderr){
-		char log_board[300];
-		vsnprintf(log_board, sizeof log_board, format, arg);
-        debug_out_str(log_board);
-		return 0;
-	}else{
+#endif
 		return vfprintf(file, format, arg);
 #define EXIT_SUCCESS 0
 #define EXIT_FAILURE 1
 int hostname_to_ip(char * , char *);
 int whois_query(char * , char * , char **);
 char *str_replace(char *search , char *replace , char *subject );
 char* str_copy(char*);
-void start_console()
-{
-	con_init_console_dll();
-	con_set_title("Whois");
 int main(int argc , char *argv[])
+{
 	networklib_init();
 	char *domain , *data = NULL;
-	int f_flag=0;
 	int f_flag=0;
 	if(argc==2){
-		start_console();
+	if(argc==2){
 		domain=strdup(argv[1]);
+	}
 	else if(!strcmp(argv[2], "-f") && argc==4){
 		domain=strdup(argv[1]);
 		if((out=fopen(argv[3],"w"))==NULL){
-			printf("Error writing to file: '%s' !\n", argv[3]);
-			exit(0);
+			printf("Error writing to file: '%s' !\n", argv[3]);
 			exit(0);
 	}else{
+	}else{
+		show_help();
-		start_console();
+		exit(0);
 		show_help();
-		exit(0);
+	if(out==stdout){
 		con_set_title("Whois");
+	}
 	domain = str_replace("http://" , "" , domain);
 	domain = str_replace("www." , "" , domain);
 	//get the extension , com , org , edu
+	dt = strdup(domain);
 	dt = strdup(domain);
 	if(dt == NULL){
-		unix_fprintf(out, "strdup failed");
+		fprintf(out, "strdup failed");
+	}
 	pch = (char*)strtok(dt , ".");
 	while(pch != NULL){
 		strcpy(ext , pch);
 		pch = strtok(NULL , ".");
-	}
+	}
 	// This will tell the whois server for the particular TLD like com , org
 	if( whois_query("whois.iana.org" , ext , &response) == EXIT_FAILURE){
-		unix_fprintf(out, "Whois query failed");
+		fprintf(out, "Whois query failed");
         return 1;
 	}
-	unix_fprintf(out, "\n\nResponse is:\n\n");
+	fprintf(out, "\n\nResponse is:\n\n");
-	unix_fprintf(out, "%s", response);
+	fprintf(out, "%s", response);
 	// Now analysze the response
 	pch = strtok(response , "\n");
 		// Next line please
 		pch = strtok(NULL , "\n");
+	}
 	// Now we have the TLD whois server in wch , query again
 	//This will provide minimal whois information along with the parent whois server of the specific domain :)
+	wch = strdup(wch);
 	free(response);
     //This should not be necessary , but segmentation fault without this , why ?
 	response = NULL;
 	if(wch != NULL){
-		unix_fprintf(out, "\nTLD Whois server is : %s" , wch);
+		fprintf(out,"\nTLD Whois server is : %s" , wch);
-        if( whois_query(wch , domain , &response) == EXIT_FAILURE){
+		if( whois_query(wch , domain , &response) == EXIT_FAILURE){
-			unix_fprintf(out, "Whois query failed\n");
+			fprintf(out, "Whois query failed\n");
             return EXIT_FAILURE;
+		}
 	}else{
-		unix_fprintf(out, "\nTLD whois server for %s not found\n" , ext);
+		fprintf(out, "\nTLD whois server for %s not found\n" , ext);
 		return EXIT_SUCCESS;
+	}
     */
 	if(wch){
 		// Now we have the registrar whois server , this has the direct full information of the particular domain
 		// so lets query again
-		unix_fprintf(out, "\nRegistrar Whois server is : %s" , wch);
+		fprintf(out, "\nRegistrar Whois server is : %s" , wch);
 		if( whois_query(wch , domain , &response) == EXIT_FAILURE ){
-			unix_fprintf(out, "Whois query failed");
+			fprintf(out, "Whois query failed");
 		}else{
-            unix_fprintf(out, "\n%s" , response);
+            fprintf(out, "\n%s" , response);
+        }
+	}
 	/*
         otherwise echo the output from the previous whois result
     */
 	else{
-		unix_fprintf(out, "%s" , response_2);
+		fprintf(out, "%s" , response_2);
+	}
 	return 0;
 	sock = socket(AF_INET4 , SOCK_STREAM , IPPROTO_TCP);
     //Prepare connection structures :)
-    memset( &dest , 0 , sizeof(dest) );
+    memset(&dest , 0 , sizeof(dest) );
-    dest.sin_family = AF_INET;
 	dest.sin_family = AF_INET;
-    server = str_copy(server);
+    server = str_copy(server);
-    server[strcspn(server, "\r\n")] = '\0';
     server[strcspn(server, "\r\n")] = '\0';
-	unix_fprintf(out, "\nResolving: %s ...\n" , server);
+	fprintf(out, "\nResolving: %s ...\n" , server);
-	if( hostname_to_ip(server , ip) == EXIT_FAILURE ){
+	if(hostname_to_ip(server , ip) == EXIT_FAILURE ){
-        unix_fprintf(out, "Failed\n");
+        fprintf(out, "Failed\n");
         return EXIT_FAILURE;
-	}
+	}
 	unix_fprintf(out, "Found ip: %s \n" , ip);
+	fprintf(out, "Found ip: %s \n" , ip);
     dest.sin_addr = inet_addr(ip);
 	dest.sin_port = PORT(WHOIS_PORT);
 ;	//Now connect to remote server
-	//Now connect to remote server
+	if(connect(sock , (const struct sockaddr*) &dest , sizeof(dest)) < 0){
-	if(connect(sock , (const struct sockaddr*) &dest , sizeof(dest)) < 0){
+		perror("connect failed");
-		perror("connect failed");
+		perror(strerror(errno));
         return EXIT_FAILURE;
+	}
 	//Now send some data or message
-	unix_fprintf(out, "\nQuerying for: %s ...\n" , query);
+	fprintf(out, "\nQuerying for: %s ...\n" , query);
 	sprintf(message , "%s\r\n" , query);
 	if( send(sock , message , strlen(message) , 0) < 0){
 		perror("send failed");
         return EXIT_FAILURE;
 	}
 	//Now receive the response
 	while((read_size = recv(sock, buffer, sizeof(buffer), 0))){
 		*response = realloc(*response , read_size + total_size);
 		if(*response == NULL){
-			unix_fprintf(out, "realloc failed");
+			fprintf(out, "realloc failed");
-            return EXIT_FAILURE;
             return EXIT_FAILURE;
 		memcpy(*response + total_size , buffer , read_size);
-		total_size += read_size;
+		memcpy(*response + total_size , buffer , read_size);
 		total_size += read_size;

Subversion Repositories Kolibri OS

(root)/programs/network/whois/whois.c – Rev 8515 → 8686