1 /*
2 
3 kHTTPd -- the next generation
4 
5 Accept connections
6 
7 */
8 /****************************************************************
9  *	This program is free software; you can redistribute it and/or modify
10  *	it under the terms of the GNU General Public License as published by
11  *	the Free Software Foundation; either version 2, or (at your option)
12  *	any later version.
13  *
14  *	This program is distributed in the hope that it will be useful,
15  *	but WITHOUT ANY WARRANTY; without even the implied warranty of
16  *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  *	GNU General Public License for more details.
18  *
19  *	You should have received a copy of the GNU General Public License
20  *	along with this program; if not, write to the Free Software
21  *	Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22  *
23  ****************************************************************/
24 
25 #include "structure.h"
26 #include "prototypes.h"
27 #include "sysctl.h"
28 
29 #include <linux/smp_lock.h>
30 
31 /*
32 
33 Purpose:
34 
35 AcceptConnections puts all "accepted" connections in the
36 "WaitForHeader" queue.
37 
38 Return value:
39 	The number of accepted connections
40 */
41 
42 
AcceptConnections(const int CPUNR,struct socket * Socket)43 int AcceptConnections(const int CPUNR, struct socket *Socket)
44 {
45 	struct http_request *NewRequest;
46 	struct socket *NewSock;
47 	int count = 0;
48 	int error;
49 
50 	EnterFunction("AcceptConnections");
51 
52 	if (atomic_read(&ConnectCount)>sysctl_khttpd_maxconnect)
53 	{
54 		LeaveFunction("AcceptConnections - to many active connections");
55 		return 0;
56 	}
57 
58 	if (Socket==NULL) return 0;
59 
60 	/*
61 	   Quick test to see if there are connections on the queue.
62 	   This is cheaper than accept() itself because this saves us
63 	   the allocation of a new socket. (Which doesn't seem to be
64 	   used anyway)
65 	*/
66    	if (Socket->sk->tp_pinfo.af_tcp.accept_queue==NULL)
67 	{
68 		return 0;
69 	}
70 
71 	error = 0;
72 	while (error>=0)
73 	{
74 		NewSock = sock_alloc();
75 		if (NewSock==NULL)
76 			break;
77 
78 
79 		NewSock->type = Socket->type;
80 		NewSock->ops = Socket->ops;
81 
82 
83 		error = Socket->ops->accept(Socket,NewSock,O_NONBLOCK);
84 
85 
86 		if (error<0)
87 		{
88 			sock_release(NewSock);
89 			break;
90 		}
91 
92 		if (NewSock->sk->state==TCP_CLOSE)
93 		{
94 			sock_release(NewSock);
95 			continue;
96 		}
97 
98 		/* Allocate a request-entry for the connection */
99 		NewRequest = kmalloc(sizeof(struct http_request),(int)GFP_KERNEL);
100 
101 		if (NewRequest == NULL)
102 		{
103 			Send50x(NewSock); 	/* Service not available. Try again later */
104 			sock_release(NewSock);
105 			break;
106 		}
107 		memset(NewRequest,0,sizeof(struct http_request));
108 
109 		NewRequest->sock = NewSock;
110 
111 		NewRequest->Next = threadinfo[CPUNR].WaitForHeaderQueue;
112 
113 		init_waitqueue_entry(&NewRequest->sleep,current);
114 
115 		add_wait_queue(NewSock->sk->sleep,&(NewRequest->sleep));
116 
117 		threadinfo[CPUNR].WaitForHeaderQueue = NewRequest;
118 
119 		atomic_inc(&ConnectCount);
120 
121 
122 		count++;
123 	}
124 
125 	LeaveFunction("AcceptConnections");
126 	return count;
127 }
128