4 #import <CommonCrypto/CommonDigest.h>
8 @synthesize tagName = m_name;
9 @synthesize tagType = m_type;
10 @synthesize subtags = m_subtags;
12 + (id)tagFromBuffer:(uint8_t **) buffer withLenght:(int) length {
13 ECTag *tag = nil; //[[ECTag alloc] init];
16 NSLog(@"[EC] buffer for tag is too short");
20 uint16_t name16 = *(uint16_t *)(*buffer);
21 name16 = ntohs(name16);
22 bool have_subtags = (name16 & 1) != 0;
24 ECTagNames tag_name = (ECTagNames)(name16);
25 *buffer += sizeof(name16);
27 uint8_t type8 = *(*buffer);
28 ECTagTypes tag_type = (ECTagTypes)type8;
29 *buffer += sizeof(type8);
31 uint32_t size32 = *(uint32_t *)(*buffer);
32 size32 = ntohl(size32);
33 *buffer += sizeof(uint32_t);
35 NSMutableArray *subtags = have_subtags ? [ECTag readSubtags:buffer withLenght:(length-3)] : nil;
38 case EC_TAGTYPE_UINT8:
39 tag = [ECTagInt8 tagFromBuffer:buffer];
41 case EC_TAGTYPE_UINT16:
42 tag = [ECTagInt16 tagFromBuffer:buffer];
44 case EC_TAGTYPE_UINT32:
45 tag = [ECTagInt32 tagFromBuffer:buffer];
47 case EC_TAGTYPE_UINT64:
48 tag = [ECTagInt64 tagFromBuffer:buffer];
50 case EC_TAGTYPE_HASH16:
51 tag = [ECTagMD5 tagFromBuffer:buffer];
53 case EC_TAGTYPE_STRING:
54 tag = [ECTagString tagFromBuffer:buffer];
60 tag->m_name = tag_name;
61 tag->m_subtags = subtags;
66 + (NSMutableArray *)readSubtags:(uint8_t **) buffer withLenght:(int) length {
68 uint16_t count16 = *(uint16_t *)(*buffer);
69 count16 = ntohs(count16);
70 *buffer += sizeof(count16);
71 NSMutableArray *array = [[NSMutableArray alloc] init];
73 for(int i = 0; i < count16; i++) {
74 id tag = [ECTag tagFromBuffer:buffer withLenght:length];
76 [array addObject:tag];
83 - (void)writeToSocket:(NSOutputStream *) socket {
84 uint16_t name16 = (uint16_t)m_name;
86 uint8_t type8 = (uint8_t)m_type;
87 if ( [m_subtags count] ) {
90 name16 = htons(name16);
91 [socket write:(uint8_t *)&name16 maxLength:sizeof(name16)];
92 [socket write:&type8 maxLength:sizeof(type8)];
93 uint32_t size32 = [self getSize];
94 size32 = htonl(size32);
95 [socket write:(uint8_t *)&size32 maxLength:sizeof(size32)];
98 - (void)writeSubtagsToSocket:(NSOutputStream *) socket {
99 uint16_t count16 = [m_subtags count];
100 count16 = htons(count16);
101 [socket write:(uint8_t *)&count16 maxLength:sizeof(count16)];
102 for (ECTag *t in m_subtags) {
103 [t writeToSocket:socket];
108 int total_size = m_size;
109 for (ECTag *t in m_subtags) {
110 total_size += [t getSize];
111 // name + type + size
112 total_size += (2 + 1 + 4);
113 if ([t->m_subtags count]) {
120 - (id)tagByName:(ECTagNames) tagname {
122 for (ECTag *t in m_subtags) {
123 if (t.tagName == tagname) {
131 - (void)initSubtags {
132 m_subtags = [NSMutableArray array];
141 - (uint64_t)tagInt64ByName: (ECTagNames) tagname {
142 ECTag *st = [self tagByName: tagname];
147 switch ([st getSize]) {
149 ECTagInt8 *t = (ECTagInt8 *)st;
150 value = t.uint8Value;
154 ECTagInt16 *t = (ECTagInt16 *)st;
155 value = t.uint16Value;
159 ECTagInt32 *t = (ECTagInt32 *)st;
160 value = t.uint32Value;
164 ECTagInt64 *t = (ECTagInt64 *)st;
165 value = t.uint64Value;
175 @implementation ECTagInt8
177 @synthesize uint8Value = m_val;
179 + (id)tagFromInt8:(uint8_t) value withName:(ECTagNames) name {
180 ECTagInt8 *tag = [[ECTagInt8 alloc] init];
183 tag->m_type = EC_TAGTYPE_UINT8;
189 + (id)tagFromBuffer:(uint8_t **) buffer {
190 ECTagInt8 *tag = [[ECTagInt8 alloc] init];
191 tag->m_val = **buffer;
193 tag->m_type = EC_TAGTYPE_UINT8;
200 - (void)writeToSocket:(NSOutputStream *) socket {
201 [super writeToSocket:socket];
203 [socket write:&m_val maxLength:sizeof(m_val)];
208 @implementation ECTagInt16
210 @synthesize uint16Value = m_val;
212 + (id)tagFromInt16:(uint16_t) value withName:(ECTagNames) name {
213 ECTagInt16 *tag = [[ECTagInt16 alloc] init];
216 tag->m_type = EC_TAGTYPE_UINT16;
222 + (id)tagFromBuffer:(uint8_t **) buffer {
223 ECTagInt16 *tag = [[ECTagInt16 alloc] init];
225 tag->m_val = ntohs(*((uint16_t *)(*buffer)));
227 tag->m_type = EC_TAGTYPE_UINT16;
237 @implementation ECTagInt32
239 @synthesize uint32Value = m_val;
241 + (id)tagFromInt32:(uint32_t) value withName:(ECTagNames) name {
242 ECTagInt32 *tag = [[ECTagInt32 alloc] init];
245 tag->m_type = EC_TAGTYPE_UINT32;
251 + (id)tagFromBuffer:(uint8_t **) buffer {
252 ECTagInt32 *tag = [[ECTagInt32 alloc] init];
254 tag->m_val = ntohl(*((uint32_t *)(*buffer)));
256 tag->m_type = EC_TAGTYPE_UINT32;
266 @implementation ECTagInt64
268 @synthesize uint64Value = m_val;
270 + (id)tagFromInt64:(uint64_t) value withName:(ECTagNames) name {
271 ECTagInt64 *tag = [[ECTagInt64 alloc] init];
274 tag->m_type = EC_TAGTYPE_UINT64;
281 + (id)tagFromBuffer:(uint8_t **) buffer {
282 ECTagInt64 *tag = [[ECTagInt64 alloc] init];
284 uint32 val32 = *((uint32_t *)(*buffer));
288 val32 = *((uint32_t *)(*buffer));
292 tag->m_val = (hi << 32) | lo;
294 tag->m_type = EC_TAGTYPE_UINT64;
300 - (void)writeToSocket:(NSOutputStream *) socket {
301 [super writeToSocket:socket];
303 uint32_t val32 = m_val >> 32;
304 val32 = htonl(val32);
305 [socket write:(uint8_t *)&val32 maxLength:sizeof(val32)];
307 val32 = m_val & 0xffffffff;
308 val32 = htonl(val32);
309 [socket write:(uint8_t *)&val32 maxLength:sizeof(val32)];
315 @implementation ECTagData
317 - (void)writeToSocket:(NSOutputStream *) socket {
318 [super writeToSocket:socket];
320 [socket write:(const uint8_t *)[m_data bytes] maxLength:m_size];
328 + (id)tagFromBuffer:(uint8_t **) buffer withLenght:(int) length {
329 ECTagData *tag = [[ECTagData alloc] init];
331 tag->m_data = [NSData dataWithBytes: *buffer length: length];
332 [tag->m_data retain];
340 @implementation ECTagMD5
342 + (id)tagFromString:(NSString *) string withName:(ECTagNames) name {
343 ECTagMD5 *tag = [[ECTagMD5 alloc] init];
346 unsigned char md5data[16];
348 CC_MD5_Update(&ctx, [string UTF8String], [string length]);
349 CC_MD5_Final(md5data, &ctx);
351 tag->m_data = [NSData dataWithBytes: md5data length: sizeof(md5data)];
352 [tag->m_data retain];
354 tag->m_type = EC_TAGTYPE_HASH16;
360 + (id)tagFromBuffer:(uint8_t **) buffer {
361 ECTagMD5 *tag = [[ECTagMD5 alloc] init];
364 tag->m_val.lo = *((uint64_t *)(*buffer));
366 tag->m_val.hi = *((uint64_t *)(*buffer));
372 - (MD5Data)getMD5Data {
374 uint8_t *data_ptr = (uint8_t *)[m_data bytes];
376 uint64_t hi = *(uint64_t *)data_ptr;
378 uint64_t lo = *(uint64_t *)data_ptr;
379 MD5Data md5 = {hi, lo};
385 - (NSString *)stringKey {
386 NSString *s = [NSString stringWithFormat:@"%qx%qx", m_val.hi, m_val.lo];
392 @implementation ECTagString
394 @synthesize stringValue = m_val;
396 + tagFromString:(NSString *) string withName:(ECTagNames) name {
397 ECTagString *tag = [[ECTagString alloc] init];
399 const char *rawstring = [string UTF8String];
400 tag->m_size = strlen(rawstring) + 1;
401 tag->m_data = [NSData dataWithBytes: rawstring length: tag->m_size];
402 [tag->m_data retain];
403 tag->m_type = EC_TAGTYPE_STRING;
409 + (id)tagFromBuffer:(uint8_t **) buffer {
410 ECTagString *tag = [[ECTagString alloc] init];
412 tag->m_val = [NSString stringWithCString:(char *)(*buffer) encoding:NSUTF8StringEncoding];
413 *buffer += [tag->m_val length] + 1;
420 @implementation ECPacket
422 @synthesize opcode = m_opcode;
424 + (id)packetWithOpcode:(ec_opcode_t) opcode {
425 ECPacket *p = [[ECPacket alloc] init];
427 [p initWithOpcode:opcode];
433 - (void)initWithOpcode:(ec_opcode_t) opcode {
437 // allow notification push to my client
438 m_flags |= EC_FLAG_NOTIFY | EC_FLAG_ACCEPTS;
443 + (id)packetFromBuffer:(uint8_t *) buffer withLength:(int)length {
447 ECPacket *p = [[ECPacket alloc] init];
448 uint8_t *data = buffer;
450 p->m_flags = ntohl(*((uint32_t *)data));
452 uint32_t packet_size = ntohl(*((uint32_t *)data));
454 if ( packet_size > 1024*1024 ) {
457 p->m_opcode = (ec_opcode_t)(*data);
460 uint16_t tag_count = ntohs(*((uint16_t *)data));
463 p->m_subtags = [[NSMutableArray alloc] init];
464 [p->m_subtags retain];
465 uint8_t *start_ptr = data;
466 for(int i = 0; i < tag_count; i++) {
467 ECTag *tag = [ECTag tagFromBuffer:&data withLenght:(length - (data - start_ptr))];
468 // some tags are not supported yet
470 [p->m_subtags addObject:tag];
478 - (void)writeToSocket:(NSOutputStream *) socket {
479 // 1 (command) + 2 (tag count)
480 int packet_size = [self getSize] + 1 + 2;
482 // No need for zlib on client side
483 // if ( packet_size > MaxUncompressedPacket ) {
484 // m_flags |= (Int32)ECFlags.EC_FLAG_ZLIB;
486 uint32_t val32 = htonl(m_flags);
487 [socket write:(uint8_t *)&val32 maxLength:sizeof(val32)];
488 if ( m_flags & EC_FLAG_ACCEPTS ) {
489 [socket write:(uint8_t *)&val32 maxLength:sizeof(val32)];
491 val32 = htonl(packet_size);
492 [socket write:(uint8_t *)&val32 maxLength:sizeof(val32)];
493 [socket write:(uint8_t *)&m_opcode maxLength:sizeof(m_opcode)];
495 if ( [m_subtags count] ) {
496 [self writeSubtagsToSocket:socket];
499 [socket write:(uint8_t *)&val16 maxLength:sizeof(val16)];
505 @implementation ECLoginPacket
507 + (id)loginPacket:(NSString *) password withVersion:(NSString *) version {
508 ECLoginPacket *p = [[ECLoginPacket alloc] init];
510 [p initWithOpcode:EC_OP_AUTH_REQ];
512 ECTagMD5 *passtag = [ECTagMD5 tagFromString: password withName:EC_TAG_PASSWD_HASH];
513 [p->m_subtags addObject:passtag];
515 ECTagString *version_tag = [ECTagString tagFromString:version withName:EC_TAG_CLIENT_VERSION];
516 [p->m_subtags addObject:version_tag];
518 ECTagString *client_name_tag = [ECTagString tagFromString:@"cocoa-frontend" withName:EC_TAG_CLIENT_NAME];
519 [p->m_subtags addObject:client_name_tag];
521 ECTagInt64 *proto_version_tag = [ECTagInt64 tagFromInt64:EC_CURRENT_PROTOCOL_VERSION withName:EC_TAG_PROTOCOL_VERSION];
522 [p->m_subtags addObject:proto_version_tag];
531 @implementation ECRemoteConnection
533 @synthesize error = m_error;
535 + (id)remoteConnection {
536 ECRemoteConnection *p = [[ECRemoteConnection alloc] init];
539 // rx buffer can be resized as needed
541 p->m_rxbuf = [NSMutableData dataWithLength:1024];
545 // client only transmit commands, which are
546 // quite small in size. "big enough" buffer will do the trick
548 // p->m_txbuf = [NSMutableData dataWithLength:1024];
549 // [p->m_txbuf retain];
566 - (void)connectToAddress:(NSString *) hostname withPort:(int)trgport {
569 // NSHost *host = [NSHost hostWithName:hostname];
570 NSHost *host = [NSHost hostWithAddress:hostname];
572 [NSStream getStreamsToHost:host port:trgport inputStream:&m_istream outputStream:&m_ostream];
577 [m_istream setDelegate:self];
578 [m_ostream setDelegate:self];
580 [m_istream scheduleInRunLoop:[NSRunLoop currentRunLoop]
581 forMode:NSDefaultRunLoopMode];
583 [m_ostream scheduleInRunLoop:[NSRunLoop currentRunLoop]
584 forMode:NSDefaultRunLoopMode];
589 m_remaining_size = 8;
593 - (void)sendLogin:(NSString *) password {
594 m_login_requested = true;
596 ECLoginPacket *p = [ECLoginPacket loginPacket:password withVersion:@"0.1"];
600 - (void)sendPacket:(ECPacket *) packet {
601 NSOutputStream *memstream = [NSOutputStream outputStreamToMemory];
604 [packet writeToSocket:memstream];
605 id data = [memstream propertyForKey:NSStreamDataWrittenToMemoryStreamKey];
607 m_tx_size = [data length];
608 NSLog(@"[EC] sending packet %d bytes\n", m_tx_size);
609 m_tx_ptr = [m_ostream write:(const uint8_t *)[data bytes] maxLength:[data length]];
610 NSLog(@"[EC] %d bytes sent\n", m_tx_ptr);
611 if ( m_tx_ptr == m_tx_size ) {
614 m_txbuf = (NSData *)data;
617 NSStreamStatus stream_status = [m_ostream streamStatus];
618 switch ( stream_status ) {
619 case NSStreamStatusNotOpen:
620 case NSStreamStatusClosed:
621 case NSStreamStatusError:
622 NSLog(@"[EC] error in output stream\n");
627 // NSLog(@"[EC] status in output stream=%d\n", stream_status);
631 - (void)stream:(NSStream *)stream handleEvent:(NSStreamEvent)eventCode {
633 case NSStreamEventOpenCompleted:
635 NSLog(@"[EC] open complete\n");
638 case NSStreamEventErrorOccurred:
640 NSError *e = [stream streamError];
641 NSString *description = [e localizedDescription];
642 NSLog(@"[EC] socket error [%s]\n", [description UTF8String]);
645 case NSStreamEventHasBytesAvailable:
647 uint8_t *data_ptr = (uint8_t *)[m_rxbuf mutableBytes];
648 unsigned int len = [m_rxbuf length];
650 len = [(NSInputStream *)stream read:data_ptr + m_rx_size maxLength:len];
651 NSLog(@"[EC] receiving %d bytes, %d in total, %d remaining\n", len, m_rx_size, m_remaining_size);
654 // Remote side must be closed connection
658 int packet_offset = 0;
659 while ( total_len != 0 ) {
660 len = ( m_remaining_size > total_len ) ? total_len : m_remaining_size;
663 // are we still waiting for flags and size?
664 if ( m_rx_size < 8 ) {
665 if ( (m_rx_size + len) >= 8 ) {
666 // got flags and packet size - may proceed
667 //uint32_t flags = *(((uint32_t *)[m_rxbuf mutableBytes]) + 0);
668 uint32_t val32 = *((uint32_t *)(data_ptr + 4 + packet_offset));
670 int delta = 8 - m_rx_size;
672 m_remaining_size = ntohl(val32) - (len - delta);
673 NSLog(@"[EC] rx got flags+size, remaining count %d\n", m_remaining_size);
675 m_remaining_size -= len;
678 m_remaining_size -= len;
681 if ( m_remaining_size == 0 ) {
683 // full packet received, call handler
685 uint8_t *packet_start = data_ptr + packet_offset;
686 int packet_length = [m_rxbuf length] - packet_offset;
687 ECPacket *packet = [ECPacket packetFromBuffer:packet_start withLength:packet_length];
688 packet_offset += m_rx_size;
690 if ( m_login_requested ) {
691 m_login_requested = false;
692 m_login_ok = packet.opcode == EC_OP_AUTH_OK;
693 NSLog(@"[EC] server reply: %@\n", m_login_ok ? @"login OK" : @"login FAILED");
695 NSLog(@"[EC] calling delegate\n");
696 if ( [delegate respondsToSelector:@selector(handlePacket:)] ) {
697 [delegate handlePacket: packet];
700 m_remaining_size = 8;
707 case NSStreamEventHasSpaceAvailable:
709 if ( m_txbuf != nil ) {
710 int remaining = [m_txbuf length] - m_tx_ptr;
712 const uint8_t *txdata = ((const uint8_t *)[m_txbuf bytes]) + m_tx_ptr;
713 int txcount = [m_ostream write:txdata maxLength:remaining];
726 - (void)setDelegate:(id)val