c++实现web客户端

//和我的上一篇文章是一起写的，呵呵，大家给提点意见啊。

 

//:-)

 

#include <winsock.h>

#include <iostream>

#define HTTP_PORT 80       //HTTP连接的默认端口号

#define MAXSIZE 256        //自定义的每次传输数据的最大数量

using namespace std;

 

int make_socket() {

struct sockaddr_in local_addr;    //该结构体存储本地地址信息

int tempSockId;        //临时变量用来暂时存储socket描述符

tempSockId = socket(PF_INET, SOCK_STREAM, 0);

if (tempSockId == -1) {      //如果返回值为－1 则出错

return tempSockId;

}

 

local_addr.sin_family = AF_INET;

local_addr.sin_port = htons(HTTP_PORT);

local_addr.sin_addr.s_addr = inet_addr("127.0.0.1");

memset(&(local_addr.sin_zero), '/0', 8);

return tempSockId;       //返回取得的SOCKET

}

 

int sendall(int s, char *buf, int *len) {

int total = 0;        // 已经发送字节数

int bytesleft = *len;      // 还剩余多少字节数

int n;

while(total < *len) {

n = send(s, buf+total, bytesleft, 0); // 发送数据

if (n == -1) { break; }

total += n;

bytesleft -= n;

}

*len = total;        // 返回实际发送出去的字节数

return n==-1?-1:0;       // 成功发送返回0 失败－1

}

 

void main() {

 

WSADATA wsaData;

if (WSAStartup(MAKEWORD(1, 1), &wsaData) != 0) {

fprintf(stderr, "WSAStartup failed./n");

exit(1);

}

 

 

char server[100];       //用来保存用户想要连接的服务器地址

char command[100];       //用户命令

char filename[100];       //用户输入的用来保存实体内容的文件名

char buf[MAXSIZE];       //读取数据的缓存字节数组

 

FILE * to_store;       //用来存储文件内容的文件指针

int len = -1;

struct sockaddr_in remote_server;   //远程服务器的连接信息

struct hostent * host_info;     //远程服务器的地址信息

 

while(true) {

 

int mysocket = make_socket();   //构建本地socket

 

cout << "Please input the web site you want to connect or 'q' to quit:" << endl; //输出提示信息

gets(server);       //从控制台读取用户输入

//cout << server << endl;    //测试用用来输出服务器名称     

 

if (strcmp(server, "q") == 0) {   //用户输入q 退出程序

exit(0);

}

 

remote_server.sin_family = AF_INET;

remote_server.sin_port = htons(HTTP_PORT);

if ((host_info = gethostbyname(server)) == NULL) {     //通过服务器名得到连接信息

cout << "Server name error or can not be reached!" << endl;  //服务器名称错误或无法连接

cout << "*********Press any key to continue*************" << endl;//输出提示信息

char temp[1];

gets(temp);

continue;

}

remote_server.sin_addr = *((struct in_addr *)host_info->h_addr);

memset(&(remote_server.sin_zero), '/0', 8);

 

if (connect(mysocket, (struct sockaddr *)&remote_server,

sizeof(struct sockaddr)) == -1) {     //连接出错

cerr << "Connect error!" << endl;

cout << "*********Press any key to continue*************" << endl;

char temp[1];

gets(temp);

continue;

}

 

cout << "Now " << server << " is listening to your command!/n" << endl;  //连接成功

 

gets(command);

//cout << command << endl;    //测试用输出命令

 

 

len = strlen(command);

if (sendall(mysocket, command, &len) == -1) {

cout <<"sending error" << endl;  //发送数据出错

continue;

}

cout << "The following is the header" << endl; //输出提示信息

 

int readed = -1;

int i = 0;

bool flag = false;

readed = recv(mysocket, buf, 1, 0);       //从服务器端读取数据 readed为实际读到的

//readed = read(mysocket, buf, 1);    //字节数限定每次读取一个字节

while(readed == 1) {

 

if (i < 4) {

if (buf[0] == '/r' || buf[0] == '/n') {  //出现两个/r/n时 i==4

i++;

} else {

i = 0;

}

printf("%c", buf[0]);      //把http头信息打印在控制台

 

} else if (flag) {        //首次进入时

fwrite(buf, 1, 1, to_store);    //需向用户提示输入存储文件名称

} else {

cout << "Please input the filename to store the content file:" << endl;

gets(filename);

//cout << filename << endl;     //测试用输出文件名

to_store = fopen(filename, "w");

flag = true;

}

readed = recv(mysocket, buf, 1, 0);

//readed = read(mysocket, buf, 1);

}

fflush(to_store);

shutdown(mysocket, 2);

}

}