Shammer's Philosophy

My private adversaria

HTTP Client ver 0.1

手始めに、引数で指定されたアドレスの80ポートへ接続し、デフォルトページを取得、表示するだけのクライアントを書いてみた。

#include <ctype.h>
#include <regex.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <netinet/in.h>

int is_valid_ipaddr(char v[]){
    regex_t preg;
    int regcompresult;
    regcompresult = regcomp(&preg, "^([0-9]|[1-9][0-9]|1[0-9][0-9]|2[0-4][0-9]|25[0-5]).([0-9]|[1-9][0-9]|1[0-9][0-9]|2[0-4][0-9]|25[0-5]).([0-9]|[1-9][0-9]|1[0-9][0-9]|2[0-4][0-9]|25[0-5]).([0-9]|[1-9][0-9]|1[0-9][0-9]|2[0-4][0-9]|25[0-5])$", REG_NOSUB | REG_EXTENDED | REG_NEWLINE);
    if( regcompresult == 0 ){
	size_t nmatch = 0;
	regmatch_t pmatch[nmatch];
	int regexecresult;
	regexecresult = regexec(&preg, v, nmatch, pmatch, 0);
	if( regexecresult == 0 ){
	    printf("%s is valid.\n", v);
	    return 0;
	} else {
	    printf("%s is not valid.\n", v);
	    return -1;
	}
    }
}

int main(int argc, char* args[]){
    if( argc == 2 ){
	int is_valid_ipaddr_result = is_valid_ipaddr(args[1]);
	if( is_valid_ipaddr_result == 0 ){
	    // HTTP Client
	    int sock, bytes_read;
	    struct sockaddr_in destination_addr;
	    unsigned short destination_port = 80;
	    char response[256];
	    sock = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
	    if( sock < 0 ){
		printf("Socket open error.\n");
		exit(1);
	    }
	    memset(&destination_addr, 0, sizeof(destination_addr));
	    destination_addr.sin_family = AF_INET;
	    destination_addr.sin_addr.s_addr = inet_addr(args[1]);
	    destination_addr.sin_port = htons(destination_port);
	    if( connect(sock, (struct sockaddr *)&destination_addr, sizeof(destination_addr)) < 0 ){
		printf("Connect error.\n");
		exit(2);
	    }
	    send( sock, "GET / HTTP/1.1\r\n", strlen("GET / HTTP/1.1\r\n"), 0);
	    send( sock, "Host: ", strlen("Host: "), 0);
	    send( sock, args[1], strlen(args[1]), 0);
	    send( sock, "\r\n\r\n", strlen("\r\n\r\n"), 0);
	    while(1) {
		bzero(response, sizeof(response));
		bytes_read = recv(sock, response, sizeof(response), 0);
		if( bytes_read > 0 ){
		    printf("%s", response);
		}
		else {
		    break;
		}
	    }
	    close(sock);
	}
	else {
	    printf("Usage: %s %s\n", args[0], "Destination_IP");
	}
    }
    return 0;
}

今後は、ポート、ホスト名、URI、HTTP Header を指定できるように拡張予定。