1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

int main(int argc, char **argv)
{
    int listenfd, connfd;
    char hostname[MAXLINE], port[MAXLINE];
    socklen_t clientlen;
    struct sockaddr_storage clientaddr;
    if(argc!=2){
        printf("Usage: %s <port>\n",argv[0]);
        return 0;
    }
    listenfd = Open_listenfd(argv[1]);
    while(1){
    clientlen = sizeof(clientaddr);
    connfd = Accept(listenfd, (SA *)&clientaddr, &clientlen);
    Getnameinfo((SA *) &clientaddr, clientlen, hostname, MAXLINE, port, MAXLINE, 0);
    printf("Accepted connection from (%s, %s)\n", hostname, port);
    proxy(connfd);
    Close(connfd);
    }
    return 0;
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

void proxy(int connfd){
    char buf[MAXLINE], method[MAXLINE], uri[MAXLINE], version[MAXLINE];
    char serv_req[MAXLINE];
    rio_t rio;
  
    Rio_readinitb(&rio,connfd);
    if(!Rio_readlineb(&rio,buf,MAXLINE)) return;
    fprintf(stdout,"%s",buf);
    sscanf(buf,"%s %s %s",method,uri,version);
    if(strcasecmp(method,"GET")){
        unix_error("Proxy server does not implement this method");
        return ;
    }
    URI *uri_data = (URI *)malloc(sizeof(URI));
    parse_uri(uri_data,uri);
    int serverfd = Open_clientfd(uri_data->host,uri_data->port);
    sprintf(serv_req,"GET %s HTTP/1.0\r\nHost: %s\r\n",uri_data->path,version,uri_data->host);;
    printf("%s\n",serv_req);
    Rio_writen(serverfd,serv_req,strlen(serv_req));
    readDeal_hdrs(serverfd,&rio);
    writeback_response(serverfd,connfd);
    Close(serverfd);
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143

#include <stdio.h>  
#include "csapp.h"  
#include "sbuf.h"  
/* Recommended max cache and object sizes */  
#define MAX_CACHE_SIZE 1049000  
#define MAX_OBJECT_SIZE 102400  
  
typedef struct {  
    char host[MAXLINE];  
    char port[MAXLINE];  
    char path[MAXLINE];  
}URI;  
  
/* You won't lose style points for including this long line in your code */  
static const char *user_agent_hdr = "User-Agent: Mozilla/5.0 (X11; Linux x86_64; rv:10.0.3) Gecko/20120305 Firefox/10.0.3\r\n";  
  
  
/* Reload unix_error function to avoid exit */  
static void unix_error(char *msg);  
  
/* share buf */  
sbuf_t sbuf;  
#define SBUFSIZE 20    // sbuf/线程池的大小  
#define NTHREADS 4  // 4个线程  
  
  
/* the proxy server */  
int parse_uri(URI *rebuild_uri,char raw_uri[MAXLINE]){  
    char *hostptr = strstr(raw_uri,"://");  
    char *pathptr = NULL;  
    char *portptr = NULL;  
    if(!hostptr){// 不以 http:// 形式请求  
        rebuild_uri->host[0] = 0;  
        strcpy(rebuild_uri->port,"80");  
        if(pathptr = strstr(raw_uri,"/"))  
            strcpy(rebuild_uri->path,pathptr);  
        return 1;  
    }  
    else{  
        portptr = strstr(hostptr+3,":");  
        if(portptr){  
            int port;  
            sscanf(portptr+1,"%d%s",&port,rebuild_uri->path);  
            sprintf(rebuild_uri->port,"%d",port);  
            *portptr = 0;  
        }  
        else{  
            pathptr = strstr(hostptr+3,"/");  
            strcpy(rebuild_uri->path,pathptr);  
            *pathptr = 0;  
                    }  
        strcpy(rebuild_uri->host,hostptr+3);  
    }  
    return 2;  
}  
void readDeal_hdrs(int sendfd,rio_t *rp){ // 发送请求头  
    char buf[MAXLINE];  
    sprintf(buf,"%sConnection: close\r\nProxy-Connection: close\r\n",user_agent_hdr);  
    Rio_writen(sendfd,buf,strlen(buf));  
    /* 保留其他头部信息 */  
    for(Rio_readlineb(rp,buf,MAXLINE);strcmp(buf,"\r\n");Rio_readlineb(rp,buf,MAXLINE)){  
        if(!strncmp(buf,"Host:",5) || !strncmp(buf,"Connection:",11) ||  
            !strncmp(buf,"User-Agent:",11) || !strncmp(buf,"Proxy-Connection:",17))  
            continue;  
        else  
            Rio_writen(sendfd,buf,strlen(buf));  
        printf("%s\n",buf);  
    }  
    Rio_writen(sendfd,buf,strlen(buf));  
  
}  
void writeback_response(int servfd,int clientfd){  
    rio_t rio;  
    int n;  
    char buf[MAXLINE];  
    Rio_readinitb(&rio,servfd);  
    while(n=Rio_readlineb(&rio,buf,MAXLINE))  
        Rio_writen(clientfd,buf,n);  
    }  
void proxy(int connfd){  
    char buf[MAXLINE], method[MAXLINE], uri[MAXLINE], version[MAXLINE];  
    char serv_req[MAXLINE];  
    rio_t rio;  
  
    Rio_readinitb(&rio,connfd);  
    if(!Rio_readlineb(&rio,buf,MAXLINE)) return;  
    fprintf(stdout,"%s",buf);  
    sscanf(buf,"%s %s %s",method,uri,version);  
    if(strcasecmp(method,"GET")){  
        unix_error("Proxy server does not implement this method");  
        return ;  
    }  
    URI *uri_data = (URI *)malloc(sizeof(URI));  
    parse_uri(uri_data,uri);  
    int serverfd = Open_clientfd(uri_data->host,uri_data->port);  
    sprintf(serv_req,"GET %s HTTP/1.0\r\nHost: %s\r\n",uri_data->path,uri_data->host);;  
    printf("%s\n",serv_req);  
    Rio_writen(serverfd,serv_req,strlen(serv_req));  
    readDeal_hdrs(serverfd,&rio);  
    writeback_response(serverfd,connfd);  
    Close(serverfd);  
}  
/* thread */  
void *thread(void *vargp){  
    Pthread_detach(pthread_self());  
    int connfd;  
    while(1){  
    connfd = sbuf_remove(&sbuf);  
    proxy(connfd);  
    Close(connfd);  
    }  
}  
  
int main(int argc, char **argv)  
{  
    int listenfd, connfd;  
    char hostname[MAXLINE], port[MAXLINE];  
    socklen_t clientlen;  
    struct sockaddr_storage clientaddr;  
    pthread_t tid;  
        if(argc!=2){  
        printf("Usage: %s <port>\n",argv[0]);  
        return 0;  
    }  
    listenfd = Open_listenfd(argv[1]);  
    sbuf_init(&sbuf,SBUFSIZE);  
  
    for(int i=0;i<NTHREADS;i++){ // 创建几个并发线程  
        Pthread_create(&tid,NULL,thread,NULL);  
    }  
    while(1){  
    clientlen = sizeof(clientaddr);  
    connfd = Accept(listenfd, (SA *)&clientaddr, &clientlen);  
    Getnameinfo((SA *) &clientaddr, clientlen, hostname, MAXLINE, port, MAXLINE, 0);  
    printf("Accepted connection from (%s, %s)\n", hostname, port);  
    sbuf_insert(&sbuf, connfd);  
    }  
    return 0;  
}  

static void unix_error(char *msg){  
    fprintf(stderr, "%s\n",msg);  
}