-
Notifications
You must be signed in to change notification settings - Fork 0
/
KCFileHandleInputStream.m
123 lines (95 loc) · 3.41 KB
/
KCFileHandleInputStream.m
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
//
// FileHandleInputStream.m
// Degrees
//
// Created by Kevin Conner on 5/22/12.
// This software is free to use.
//
#import "KCFileHandleInputStream.h"
@interface KCFileHandleInputStream ()
@property (nonatomic, strong) NSFileHandle *fileHandle;
@property (nonatomic, strong) NSMutableArray *dataParts;
@property (nonatomic, strong) NSData *currentDataPart;
@property (nonatomic, assign) NSUInteger currentDataPartBytesRead;
@end
@implementation KCFileHandleInputStream
@synthesize fileHandle = _fileHandle;
@synthesize dataParts = _dataParts;
@synthesize currentDataPart = _currentDataPart;
@synthesize currentDataPartBytesRead = _currentDataPartBytesRead;
@synthesize delegate = _delegate;
#pragma mark - Helpers
- (void)addDataPart:(NSNotification *)notification {
NSDictionary *userInfo = [notification userInfo];
NSNumber *errorNo = [userInfo objectForKey:@"NSFileHandleError"];
if (errorNo) {
[self.delegate stream:self handleEvent:NSStreamEventErrorOccurred];
return;
}
NSData *data = [userInfo objectForKey:NSFileHandleNotificationDataItem];
NSUInteger remainingBytes = data.length;
if (remainingBytes == 0) {
[self.delegate stream:self handleEvent:NSStreamEventEndEncountered];
return;
}
[self.dataParts addObject:data];
if (self.dataParts.count == 1) {
[self.delegate stream:self handleEvent:NSStreamEventHasBytesAvailable];
}
[self.fileHandle readInBackgroundAndNotify];
}
#pragma mark - Init/dealloc
- (id)initWithFileHandle:(NSFileHandle *)fileHandle {
if (self = [super init]) {
self.fileHandle = fileHandle;
self.dataParts = [NSMutableArray array];
[[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(addDataPart:) name:NSFileHandleReadCompletionNotification object:fileHandle];
[fileHandle readInBackgroundAndNotify];
}
return self;
}
- (void)dealloc {
[[NSNotificationCenter defaultCenter] removeObserver:self];
}
#pragma mark - NSInputStream
- (BOOL)hasBytesAvailable {
return self.currentDataPart != nil || 0 < self.dataParts.count;
}
- (BOOL)getBuffer:(uint8_t **)buffer length:(NSUInteger *)len {
return NO;
}
- (NSInteger)read:(uint8_t *)buffer maxLength:(NSUInteger)len {
if (self.dataParts.count == 0) {
return 0;
}
if (self.currentDataPart == nil) {
self.currentDataPart = [self.dataParts objectAtIndex:0];
self.currentDataPartBytesRead = 0;
[self.dataParts removeObjectAtIndex:0];
}
uint8_t *dataPartBuffer = (uint8_t *) self.currentDataPart.bytes + self.currentDataPartBytesRead;
NSUInteger dataPartBytesRemaining = self.currentDataPart.length - self.currentDataPartBytesRead;
if (dataPartBytesRemaining <= len) {
memcpy(buffer, dataPartBuffer, dataPartBytesRemaining);
self.currentDataPart = nil;
return dataPartBytesRemaining;
}
else {
memcpy(buffer, dataPartBuffer, len);
self.currentDataPartBytesRead += len;
return len;
}
}
- (void)open {
[self.delegate stream:self handleEvent:NSStreamEventOpenCompleted];
}
- (void)close {
[self.delegate stream:self handleEvent:NSStreamEventEndEncountered];
}
- (void)scheduleInRunLoop:(NSRunLoop *)aRunLoop forMode:(NSString *)mode {
// No-op.
}
- (void)removeFromRunLoop:(NSRunLoop *)aRunLoop forMode:(NSString *)mode {
// No-op.
}
@end