- Added network utility "WHOIS"
git-svn-id: svn://kolibrios.org@8515 a494cfbc-eb01-0410-851d-a64ba20cac60
This commit is contained in:
superturbocat2001
parent
0c1a5475ab
commit
9c14226ab7
17
programs/network/whois/Makefile
Executable file
17
programs/network/whois/Makefile
Executable file
@ -0,0 +1,17 @@
|
||||
KTCC_DIR=../../develop/ktcc/trunk
|
||||
|
||||
NAME=whois
|
||||
|
||||
KTCC=$(KTCC_DIR)/bin/kos32-tcc
|
||||
KPACK=kpack
|
||||
|
||||
SRC=whois.c
|
||||
CFLAGS=-I $(KTCC_DIR)/libc/include
|
||||
LIBS = -lck -lnetwork
|
||||
|
||||
all:
|
||||
$(KTCC) $(CFLAGS) $(SRC) $(LIBS) -o $(NAME)
|
||||
$(KPACK) $(NAME)
|
||||
|
||||
clean:
|
||||
rm $(NAME)
|
||||
319
programs/network/whois/whois.c
Normal file
319
programs/network/whois/whois.c
Normal file
@ -0,0 +1,319 @@
|
||||
/*
|
||||
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 <stdio.h>
|
||||
#include <conio.h>
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
#include <net/socket.h>
|
||||
#include <net/network.h>
|
||||
|
||||
#define stdout (FILE*)0
|
||||
|
||||
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{
|
||||
return vfprintf(file, format, arg);
|
||||
}
|
||||
}
|
||||
|
||||
void show_help()
|
||||
{
|
||||
puts("Usage: whois <host> [-f <file>]\n");
|
||||
puts(" host Connect to server host");
|
||||
puts(" -f file Redirecting output to file\n");
|
||||
puts("Example: whois google.com -f my.txt");
|
||||
}
|
||||
|
||||
int get_whois_data(char * , char **);
|
||||
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;
|
||||
|
||||
if(argc==2){
|
||||
start_console();
|
||||
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);
|
||||
}
|
||||
}else{
|
||||
start_console();
|
||||
show_help();
|
||||
exit(0);
|
||||
}
|
||||
|
||||
get_whois_data(domain , &data);
|
||||
exit(0);
|
||||
}
|
||||
|
||||
/*
|
||||
Get the whois data of a domain
|
||||
*/
|
||||
|
||||
int get_whois_data(char *domain , char **data)
|
||||
{
|
||||
char ext[1024] , *pch , *response = NULL , *response_2 = NULL , *wch , *dt;
|
||||
|
||||
//remove "http://" and "www."
|
||||
domain = str_replace("http://" , "" , domain);
|
||||
domain = str_replace("www." , "" , domain);
|
||||
|
||||
//get the extension , com , org , edu
|
||||
dt = strdup(domain);
|
||||
if(dt == NULL){
|
||||
unix_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");
|
||||
return 1;
|
||||
}
|
||||
unix_fprintf(out, "\n\nResponse is:\n\n");
|
||||
unix_fprintf(out, "%s", response);
|
||||
|
||||
// Now analysze the response
|
||||
pch = strtok(response , "\n");
|
||||
while(pch != NULL){
|
||||
// Check if whois line
|
||||
wch = strstr(pch , "whois.");
|
||||
if(wch != NULL){
|
||||
break;
|
||||
}
|
||||
|
||||
// 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 :)
|
||||
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);
|
||||
if( whois_query(wch , domain , &response) == EXIT_FAILURE){
|
||||
unix_fprintf(out, "Whois query failed\n");
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
}else{
|
||||
unix_fprintf(out, "\nTLD whois server for %s not found\n" , ext);
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
|
||||
response_2 = strdup(response);
|
||||
|
||||
// Again search for a whois server in this response. :)
|
||||
pch = strtok(response , "\n");
|
||||
while(pch != NULL){
|
||||
// Check if whois line
|
||||
wch = strstr(pch , "whois.");
|
||||
if(wch != NULL){
|
||||
break;
|
||||
}
|
||||
//Next line please
|
||||
pch = strtok(NULL , "\n");
|
||||
}
|
||||
|
||||
/*
|
||||
If a registrar whois server is found then query it
|
||||
*/
|
||||
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);
|
||||
|
||||
if( whois_query(wch , domain , &response) == EXIT_FAILURE ){
|
||||
unix_fprintf(out, "Whois query failed");
|
||||
}else{
|
||||
unix_fprintf(out, "\n%s" , response);
|
||||
}
|
||||
}
|
||||
/*
|
||||
otherwise echo the output from the previous whois result
|
||||
*/
|
||||
else{
|
||||
unix_fprintf(out, "%s" , response_2);
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
/*
|
||||
Perform a whois query to a server and record the response
|
||||
*/
|
||||
int whois_query(char *server , char *query , char **response)
|
||||
{
|
||||
char ip[32] , message[100] , buffer[1500];
|
||||
int sock , read_size , total_size = 0;
|
||||
int WHOIS_PORT = 43;
|
||||
struct sockaddr dest;
|
||||
|
||||
sock = socket(AF_INET4 , SOCK_STREAM , IPPROTO_TCP);
|
||||
|
||||
//Prepare connection structures :)
|
||||
memset( &dest , 0 , sizeof(dest) );
|
||||
dest.sin_family = AF_INET;
|
||||
|
||||
server = str_copy(server);
|
||||
server[strcspn(server, "\r\n")] = '\0';
|
||||
|
||||
unix_fprintf(out, "\nResolving: %s ...\n" , server);
|
||||
if( hostname_to_ip(server , ip) == EXIT_FAILURE ){
|
||||
unix_fprintf(out, "Failed\n");
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
unix_fprintf(out, "Found ip: %s \n" , ip);
|
||||
|
||||
dest.sin_addr = inet_addr(ip);
|
||||
dest.sin_port = PORT(WHOIS_PORT);
|
||||
|
||||
//Now connect to remote server
|
||||
if(connect(sock , (const struct sockaddr*) &dest , sizeof(dest)) < 0){
|
||||
perror("connect failed");
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
|
||||
//Now send some data or message
|
||||
unix_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");
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
memcpy(*response + total_size , buffer , read_size);
|
||||
total_size += read_size;
|
||||
}
|
||||
|
||||
unix_fprintf(out, "Done\n");
|
||||
//fflush(stdout);
|
||||
|
||||
*response = realloc(*response , total_size + 1);
|
||||
*(*response + total_size) = '\0';
|
||||
close(sock);
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
/*
|
||||
Get the ip address of a given hostname
|
||||
*/
|
||||
int hostname_to_ip(char *hostname , char *ip)
|
||||
{
|
||||
struct addrinfo *addr_info;
|
||||
char port_str[16]; sprintf(port_str, "%d", 80);
|
||||
struct addrinfo hints;
|
||||
memset(&hints, 0, sizeof(hints));
|
||||
hints.ai_family = AF_UNSPEC; // IPv4 or IPv6 doesnt matter
|
||||
hints.ai_socktype = SOCK_STREAM; // TCP stream sockets
|
||||
if (getaddrinfo(hostname, port_str, 0, &addr_info) != 0) {
|
||||
freeaddrinfo(addr_info);
|
||||
return EXIT_FAILURE;
|
||||
}else{
|
||||
strcpy(ip, inet_ntoa(addr_info->ai_addr->sin_addr));
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
}
|
||||
/*
|
||||
Search and replace a string with another string , in a string
|
||||
*/
|
||||
char *str_replace(char *search , char *replace , char *subject)
|
||||
{
|
||||
char *p = NULL , *old = NULL , *new_subject = NULL ;
|
||||
int c = 0 , search_size;
|
||||
|
||||
search_size = strlen(search);
|
||||
|
||||
//Count how many occurences
|
||||
for(p = strstr(subject , search) ; p != NULL ; p = strstr(p + search_size , search)){
|
||||
c++;
|
||||
}
|
||||
//Final size
|
||||
c = ( strlen(replace) - search_size )*c + strlen(subject);
|
||||
|
||||
//New subject with new size
|
||||
new_subject = malloc( c );
|
||||
|
||||
//Set it to blank
|
||||
strcpy(new_subject , "");
|
||||
|
||||
//The start position
|
||||
old = subject;
|
||||
|
||||
for(p = strstr(subject , search) ; p != NULL ; p = strstr(p + search_size , search)){
|
||||
//move ahead and copy some text from original subject , from a certain position
|
||||
strncpy(new_subject + strlen(new_subject) , old , p - old);
|
||||
|
||||
//move ahead and copy the replacement text
|
||||
strcpy(new_subject + strlen(new_subject) , replace);
|
||||
|
||||
//The new start position after this search match
|
||||
old = p + search_size;
|
||||
}
|
||||
|
||||
//Copy the part after the last search match
|
||||
strcpy(new_subject + strlen(new_subject) , old);
|
||||
|
||||
return new_subject;
|
||||
}
|
||||
|
||||
char* str_copy(char *source)
|
||||
{
|
||||
char *copy = malloc(strlen(source) + 1);
|
||||
|
||||
if(copy){
|
||||
strcpy(copy, source);
|
||||
}
|
||||
return copy;
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user