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;
173 return [m_subtags count];
178 @implementation ECTagInt8
180 @synthesize uint8Value = m_val;
182 + (id)tagFromInt8:(uint8_t) value withName:(ECTagNames) name {
183 ECTagInt8 *tag = [[ECTagInt8 alloc] init];
186 tag->m_type = EC_TAGTYPE_UINT8;
192 + (id)tagFromBuffer:(uint8_t **) buffer {
193 ECTagInt8 *tag = [[ECTagInt8 alloc] init];
194 tag->m_val = **buffer;
196 tag->m_type = EC_TAGTYPE_UINT8;
203 - (void)writeToSocket:(NSOutputStream *) socket {
204 [super writeToSocket:socket];
206 [socket write:&m_val maxLength:sizeof(m_val)];
211 @implementation ECTagInt16
213 @synthesize uint16Value = m_val;
215 + (id)tagFromInt16:(uint16_t) value withName:(ECTagNames) name {
216 ECTagInt16 *tag = [[ECTagInt16 alloc] init];
219 tag->m_type = EC_TAGTYPE_UINT16;
225 + (id)tagFromBuffer:(uint8_t **) buffer {
226 ECTagInt16 *tag = [[ECTagInt16 alloc] init];
228 tag->m_val = ntohs(*((uint16_t *)(*buffer)));
230 tag->m_type = EC_TAGTYPE_UINT16;
240 @implementation ECTagInt32
242 @synthesize uint32Value = m_val;
244 + (id)tagFromInt32:(uint32_t) value withName:(ECTagNames) name {
245 ECTagInt32 *tag = [[ECTagInt32 alloc] init];
248 tag->m_type = EC_TAGTYPE_UINT32;
254 + (id)tagFromBuffer:(uint8_t **) buffer {
255 ECTagInt32 *tag = [[ECTagInt32 alloc] init];
257 tag->m_val = ntohl(*((uint32_t *)(*buffer)));
259 tag->m_type = EC_TAGTYPE_UINT32;
269 @implementation ECTagInt64
271 @synthesize uint64Value = m_val;
273 + (id)tagFromInt64:(uint64_t) value withName:(ECTagNames) name {
274 ECTagInt64 *tag = [[ECTagInt64 alloc] init];
277 tag->m_type = EC_TAGTYPE_UINT64;
284 + (id)tagFromBuffer:(uint8_t **) buffer {
285 ECTagInt64 *tag = [[ECTagInt64 alloc] init];
287 uint32 val32 = *((uint32_t *)(*buffer));
291 val32 = *((uint32_t *)(*buffer));
295 tag->m_val = (hi << 32) | lo;
297 tag->m_type = EC_TAGTYPE_UINT64;
303 - (void)writeToSocket:(NSOutputStream *) socket {
304 [super writeToSocket:socket];
306 uint32_t val32 = m_val >> 32;
307 val32 = htonl(val32);
308 [socket write:(uint8_t *)&val32 maxLength:sizeof(val32)];
310 val32 = m_val & 0xffffffff;
311 val32 = htonl(val32);
312 [socket write:(uint8_t *)&val32 maxLength:sizeof(val32)];
318 @implementation ECTagData
320 - (void)writeToSocket:(NSOutputStream *) socket {
321 [super writeToSocket:socket];
323 [socket write:(const uint8_t *)[m_data bytes] maxLength:m_size];
331 + (id)tagFromBuffer:(uint8_t **) buffer withLenght:(int) length {
332 ECTagData *tag = [[ECTagData alloc] init];
334 tag->m_data = [NSData dataWithBytes: *buffer length: length];
335 [tag->m_data retain];
343 @implementation ECTagMD5
345 + (id)tagFromString:(NSString *) string withName:(ECTagNames) name {
346 ECTagMD5 *tag = [[ECTagMD5 alloc] init];
349 unsigned char md5data[16];
351 CC_MD5_Update(&ctx, [string UTF8String], [string length]);
352 CC_MD5_Final(md5data, &ctx);
354 tag->m_data = [NSData dataWithBytes: md5data length: sizeof(md5data)];
355 [tag->m_data retain];
357 tag->m_type = EC_TAGTYPE_HASH16;
363 + (id)tagFromBuffer:(uint8_t **) buffer {
364 ECTagMD5 *tag = [[ECTagMD5 alloc] init];
367 tag->m_val.lo = *((uint64_t *)(*buffer));
369 tag->m_val.hi = *((uint64_t *)(*buffer));
375 - (MD5Data)getMD5Data {
377 uint8_t *data_ptr = (uint8_t *)[m_data bytes];
379 uint64_t hi = *(uint64_t *)data_ptr;
381 uint64_t lo = *(uint64_t *)data_ptr;
382 MD5Data md5 = {hi, lo};
388 - (NSString *)stringKey {
389 NSString *s = [NSString stringWithFormat:@"%qx%qx", m_val.hi, m_val.lo];
395 @implementation ECTagString
397 @synthesize stringValue = m_val;
399 + tagFromString:(NSString *) string withName:(ECTagNames) name {
400 ECTagString *tag = [[ECTagString alloc] init];
402 const char *rawstring = [string UTF8String];
403 tag->m_size = strlen(rawstring) + 1;
404 tag->m_data = [NSData dataWithBytes: rawstring length: tag->m_size];
405 [tag->m_data retain];
406 tag->m_type = EC_TAGTYPE_STRING;
412 + (id)tagFromBuffer:(uint8_t **) buffer {
413 ECTagString *tag = [[ECTagString alloc] init];
415 tag->m_val = [NSString stringWithCString:(char *)(*buffer) encoding:NSUTF8StringEncoding];
416 *buffer += [tag->m_val length] + 1;
423 @implementation ECPacket
425 @synthesize opcode = m_opcode;
427 + (id)packetWithOpcode:(ec_opcode_t) opcode {
428 ECPacket *p = [[ECPacket alloc] init];
430 [p initWithOpcode:opcode];
436 - (void)initWithOpcode:(ec_opcode_t) opcode {
440 // allow notification push to my client
441 m_flags |= EC_FLAG_NOTIFY | EC_FLAG_ACCEPTS;
446 + (id)packetFromBuffer:(uint8_t *) buffer withLength:(int)length {
450 ECPacket *p = [[ECPacket alloc] init];
451 uint8_t *data = buffer;
453 p->m_flags = ntohl(*((uint32_t *)data));
455 uint32_t packet_size = ntohl(*((uint32_t *)data));
457 if ( packet_size > 1024*1024 ) {
460 p->m_opcode = (ec_opcode_t)(*data);
463 uint16_t tag_count = ntohs(*((uint16_t *)data));
466 p->m_subtags = [[NSMutableArray alloc] init];
467 [p->m_subtags retain];
468 uint8_t *start_ptr = data;
469 for(int i = 0; i < tag_count; i++) {
470 ECTag *tag = [ECTag tagFromBuffer:&data withLenght:(length - (data - start_ptr))];
471 // some tags are not supported yet
473 [p->m_subtags addObject:tag];
481 - (void)writeToSocket:(NSOutputStream *) socket {
482 // 1 (command) + 2 (tag count)
483 int packet_size = [self getSize] + 1 + 2;
485 // No need for zlib on client side
486 // if ( packet_size > MaxUncompressedPacket ) {
487 // m_flags |= (Int32)ECFlags.EC_FLAG_ZLIB;
489 uint32_t val32 = htonl(m_flags);
490 [socket write:(uint8_t *)&val32 maxLength:sizeof(val32)];
491 if ( m_flags & EC_FLAG_ACCEPTS ) {
492 [socket write:(uint8_t *)&val32 maxLength:sizeof(val32)];
494 val32 = htonl(packet_size);
495 [socket write:(uint8_t *)&val32 maxLength:sizeof(val32)];
496 [socket write:(uint8_t *)&m_opcode maxLength:sizeof(m_opcode)];
498 if ( [m_subtags count] ) {
499 [self writeSubtagsToSocket:socket];
502 [socket write:(uint8_t *)&val16 maxLength:sizeof(val16)];
508 @implementation ECLoginPacket
510 + (id)loginPacket:(NSString *) password withVersion:(NSString *) version {
511 ECLoginPacket *p = [[ECLoginPacket alloc] init];
513 [p initWithOpcode:EC_OP_AUTH_REQ];
515 ECTagMD5 *passtag = [ECTagMD5 tagFromString: password withName:EC_TAG_PASSWD_HASH];
516 [p->m_subtags addObject:passtag];
518 ECTagString *version_tag = [ECTagString tagFromString:version withName:EC_TAG_CLIENT_VERSION];
519 [p->m_subtags addObject:version_tag];
521 ECTagString *client_name_tag = [ECTagString tagFromString:@"cocoa-frontend" withName:EC_TAG_CLIENT_NAME];
522 [p->m_subtags addObject:client_name_tag];
524 ECTagInt64 *proto_version_tag = [ECTagInt64 tagFromInt64:EC_CURRENT_PROTOCOL_VERSION withName:EC_TAG_PROTOCOL_VERSION];
525 [p->m_subtags addObject:proto_version_tag];
534 @implementation ECRemoteConnection
536 @synthesize error = m_error;
538 + (id)remoteConnection {
539 ECRemoteConnection *p = [[ECRemoteConnection alloc] init];
542 // rx buffer can be resized as needed
544 p->m_rxbuf = [NSMutableData dataWithLength:1024];
548 // client only transmit commands, which are
549 // quite small in size. "big enough" buffer will do the trick
551 // p->m_txbuf = [NSMutableData dataWithLength:1024];
552 // [p->m_txbuf retain];
569 - (void)connectToAddress:(NSString *) hostname withPort:(int)trgport {
572 // NSHost *host = [NSHost hostWithName:hostname];
573 NSHost *host = [NSHost hostWithAddress:hostname];
575 [NSStream getStreamsToHost:host port:trgport inputStream:&m_istream outputStream:&m_ostream];
580 [m_istream setDelegate:self];
581 [m_ostream setDelegate:self];
583 [m_istream scheduleInRunLoop:[NSRunLoop currentRunLoop]
584 forMode:NSDefaultRunLoopMode];
586 [m_ostream scheduleInRunLoop:[NSRunLoop currentRunLoop]
587 forMode:NSDefaultRunLoopMode];
592 m_remaining_size = 8;
596 - (void)sendLogin:(NSString *) password {
597 m_login_requested = true;
599 ECLoginPacket *p = [ECLoginPacket loginPacket:password withVersion:@"0.1"];
603 - (void)sendPacket:(ECPacket *) packet {
604 NSOutputStream *memstream = [NSOutputStream outputStreamToMemory];
607 [packet writeToSocket:memstream];
608 id data = [memstream propertyForKey:NSStreamDataWrittenToMemoryStreamKey];
610 m_tx_size = [data length];
611 NSLog(@"[EC] sending packet %d bytes\n", m_tx_size);
612 m_tx_ptr = [m_ostream write:(const uint8_t *)[data bytes] maxLength:[data length]];
613 NSLog(@"[EC] %d bytes sent\n", m_tx_ptr);
614 if ( m_tx_ptr == m_tx_size ) {
617 m_txbuf = (NSData *)data;
620 NSStreamStatus stream_status = [m_ostream streamStatus];
621 switch ( stream_status ) {
622 case NSStreamStatusNotOpen:
623 case NSStreamStatusClosed:
624 case NSStreamStatusError:
625 NSLog(@"[EC] error in output stream\n");
630 // NSLog(@"[EC] status in output stream=%d\n", stream_status);
634 - (void)stream:(NSStream *)stream handleEvent:(NSStreamEvent)eventCode {
636 case NSStreamEventOpenCompleted:
638 NSLog(@"[EC] open complete\n");
641 case NSStreamEventErrorOccurred:
643 NSError *e = [stream streamError];
644 NSString *description = [e localizedDescription];
645 NSLog(@"[EC] socket error [%s]\n", [description UTF8String]);
648 case NSStreamEventHasBytesAvailable:
650 uint8_t *data_ptr = (uint8_t *)[m_rxbuf mutableBytes];
651 unsigned int len = [m_rxbuf length];
653 len = [(NSInputStream *)stream read:data_ptr + m_rx_size maxLength:len];
655 NSLog(@"[EC] receiving %d bytes, %d in total, %d remaining\n", len, m_rx_size, m_remaining_size);
659 // Remote side must be closed connection
662 if ( [delegate respondsToSelector:@selector(handleError)] ) {
663 [delegate handleError];
667 int packet_offset = 0;
668 while ( total_len != 0 ) {
669 len = ( m_remaining_size > total_len ) ? total_len : m_remaining_size;
672 // are we still waiting for flags and size?
673 if ( m_rx_size < 8 ) {
674 if ( (m_rx_size + len) >= 8 ) {
675 // got flags and packet size - may proceed
676 //uint32_t flags = *(((uint32_t *)[m_rxbuf mutableBytes]) + 0);
677 uint32_t val32 = *((uint32_t *)(data_ptr + 4 + packet_offset));
679 int delta = 8 - m_rx_size;
681 m_remaining_size = ntohl(val32) - (len - delta);
683 NSLog(@"[EC] rx got flags+size, remaining count %d\n", m_remaining_size);
686 m_remaining_size -= len;
689 m_remaining_size -= len;
692 if ( m_remaining_size == 0 ) {
694 // full packet received, call handler
696 uint8_t *packet_start = data_ptr + packet_offset;
697 int packet_length = [m_rxbuf length] - packet_offset;
698 ECPacket *packet = [ECPacket packetFromBuffer:packet_start withLength:packet_length];
699 packet_offset += m_rx_size;
701 if ( m_login_requested ) {
702 m_login_requested = false;
703 m_login_ok = packet.opcode == EC_OP_AUTH_OK;
704 NSLog(@"[EC] server reply to login request: %@\n", m_login_ok ? @"login OK" : @"login FAILED");
707 NSLog(@"[EC] calling delegate\n");
709 if ( [delegate respondsToSelector:@selector(handlePacket:)] ) {
710 [delegate handlePacket: packet];
713 m_remaining_size = 8;
720 case NSStreamEventHasSpaceAvailable:
722 if ( m_txbuf != nil ) {
723 int remaining = [m_txbuf length] - m_tx_ptr;
725 const uint8_t *txdata = ((const uint8_t *)[m_txbuf bytes]) + m_tx_ptr;
726 int txcount = [m_ostream write:txdata maxLength:remaining];
739 - (void)setDelegate:(id)val