Uses slightly less CPU now. Still too much.
[ITFoundation.git] / ITInetServerSocket.m
1 //
2 //  ITInetServerSocket.m
3 //  ITFoundation
4 //
5 //  Created by Alexander Strange on Thu Feb 13 2003.
6 //  Copyright (c) 2003 __MyCompanyName__. All rights reserved.
7 //
8
9 #import "ITInetServerSocket.h"
10 #import "ITInetSocket.h"
11 #import <sys/types.h>
12 #import <sys/time.h>
13 #import <arpa/inet.h>
14 #import <netinet/in.h>
15 #import <sys/socket.h>
16 #import <netdb.h>
17 #import <fcntl.h>
18 #import <unistd.h>
19
20 /* Too bad Objective-C doesn't have class variables... */
21 static NSMutableSet *servsockets;
22 static NSTimer *timer;
23 @interface ITInetServerSocket(Private)
24 +(void)registerSocket:(ITInetServerSocket*)sock;
25 +(void)unregisterSocket:(ITInetServerSocket*)sock;
26 -(short)lookupPortForServiceType:(NSString*)name;
27 -(void)setupConnection;
28 -(void)stopConnection;
29 -(void)setupRendezvousAdvertising;
30 -(void)stopRendezvousAdvertising;
31 +(void)setupTimer;
32 +(void)stopTimer;
33 +(void)globalTimerFunc:(NSTimer*)timer;
34 -(void)timerFunc;
35 @end
36
37 @implementation ITInetServerSocket
38 + (void)initialize
39 {
40     servsockets = [[NSMutableSet alloc] init];
41     [self setupTimer];
42 }
43
44 - (id)init
45 {
46     if (self = [super init])
47            {
48            sockfd = -1;
49            delegate = nil;
50            clients = [[NSMutableSet alloc] init];
51            service = nil;
52            port = 0;
53            rndType = rndName = nil;
54            }
55     return self;
56 }
57
58 - (id)initWithDelegate:(id)d
59 {
60     if (self = [super init])
61            {
62            sockfd = -1;
63            delegate = [d retain];
64            clients = [[NSMutableSet alloc] init];
65            service = nil;
66            port = 0;
67            rndType = rndName = nil;
68            }
69     return self;
70 }
71
72 - (void)dealloc
73 {
74     [self stopConnection];
75     [clients release];
76     [delegate release];
77     [rndName release];
78     [rndType release];
79 }
80
81 - (BOOL)start
82 {
83     if (!rndName || !rndType || !port) return NO;
84     [ITInetServerSocket registerSocket:self];
85     return YES;
86 }
87
88 - (int)sockfd
89 {
90     return sockfd;
91 }
92
93 - (NSSet*)clients
94 {
95     return clients;
96 }
97
98 - (id)delegate
99 {
100     return delegate;
101 }
102
103 - (short)port
104 {
105     return port;
106 }
107
108 - (void)stop
109 {
110     [ITInetServerSocket unregisterSocket:self];
111 }
112
113 - (void)setServiceType:(NSString*)type useForPort:(BOOL)p
114 {
115     rndType = [type retain];
116     if (p) {
117            port = [self lookupPortForServiceType:type];
118     }
119 }
120
121 - (void)setServiceName:(NSString*)name
122 {
123     rndName = [name retain];
124 }
125
126 - (void)setPort:(short)p
127 {
128     port = p;
129 }
130 @end
131
132 @implementation ITInetServerSocket(Private)
133 +(void)registerSocket:(ITInetServerSocket*)sock
134 {
135     [sock setupConnection];
136     [servsockets addObject:sock];
137 }
138
139 +(void)unregisterSocket:(ITInetServerSocket*)sock
140 {
141     [sock stopConnection];
142     [servsockets removeObject:sock];
143 }
144
145 -(short)lookupPortForServiceType:(NSString*)name
146 {
147     const char *_name = [name cString];
148     struct addrinfo hints,*res;
149     short p;
150
151     hints.ai_flags = AI_PASSIVE;
152     hints.ai_family = PF_INET;
153     hints.ai_socktype = SOCK_STREAM;
154     hints.ai_protocol = IPPROTO_TCP;
155     hints.ai_addrlen = 0;
156     hints.ai_canonname = NULL;
157     hints.ai_addr = NULL;
158     hints.ai_next = NULL;
159
160     getaddrinfo(NULL,_name,&hints,&res);
161     p = ntohs(((struct sockaddr_in *)res->ai_addr)->sin_port);
162     freeaddrinfo(res);
163     return p;
164 }
165
166 -(void)setupConnection
167 {
168     struct sockaddr_in sa;
169
170     sockfd = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
171     sa.sin_addr.s_addr = INADDR_ANY;
172     sa.sin_family = AF_INET;
173     sa.sin_port = htons(port);
174     bind(sockfd,(struct sockaddr *)&sa,sizeof(sa));
175     listen(sockfd, SOMAXCONN);
176     fcntl(sockfd,F_SETFL,O_NONBLOCK);
177     [self setupRendezvousAdvertising];
178 }
179
180 - (void)stopConnection
181 {
182     [self stopRendezvousAdvertising];
183     [clients makeObjectsPerformSelector:@selector(disconnect)];
184     shutdown(sockfd,2);
185     close(sockfd);
186     sockfd = -1;
187 }
188
189 - (void)setupRendezvousAdvertising
190 {
191     service = [[NSNetService alloc] initWithDomain:@"" type:[NSString stringWithFormat:@"_%@._tcp.",rndType] name:rndName port:htons(port)];
192     [service publish];
193 }
194
195 - (void)stopRendezvousAdvertising
196 {
197     [service stop];
198     [service release];
199     service = nil;
200 }
201
202 + (void)setupTimer
203 {
204     if (!timer) timer = [NSTimer timerWithTimeInterval:0 target:self selector:@selector(globalTimerFunc:) userInfo:nil repeats:YES];
205     [[NSRunLoop currentRunLoop] addTimer:timer forMode:NSDefaultRunLoopMode];
206 }
207
208 + (void)stopTimer
209 {
210     [timer invalidate];
211     [timer release];
212     timer = nil;
213 }
214
215 + (void)globalTimerFunc:(NSTimer*)timer
216 {
217     [servsockets makeObjectsPerformSelector:@selector(timerFunc)];
218 }
219
220 - (void)timerFunc
221 {
222     if (sockfd != -1)
223            {
224            struct sockaddr_in csa;
225            int csalen;
226            signed int cfd;
227            cfd = accept(sockfd,(struct sockaddr*)&csa,&csalen);
228            if (cfd == -1) {
229                   if (errno == EWOULDBLOCK) ;
230                   else {perror("Too bad I haven't implemented error checking yet");}
231            }
232            else {
233                   ITInetSocket *csocket = [[[ITInetSocket alloc] initWithFD:cfd delegate:self] autorelease];
234                   [clients addObject:csocket];
235                   [delegate newClientJoined:csocket];
236            }
237            }
238 }
239 @end