SendController.m
上传用户:ccvv88
上传日期:2020-08-12
资源大小:97k
文件大小:10k
源码类别:

iPhone

开发平台:

Objective-C

  1. /*
  2.     File:       SendController.m
  3.     Contains:   Manages the Send tab.
  4.     Written by: DTS
  5.     Copyright:  Copyright (c) 2009 Apple Inc. All Rights Reserved.
  6.     Disclaimer: IMPORTANT: This Apple software is supplied to you by Apple Inc.
  7.                 ("Apple") in consideration of your agreement to the following
  8.                 terms, and your use, installation, modification or
  9.                 redistribution of this Apple software constitutes acceptance of
  10.                 these terms.  If you do not agree with these terms, please do
  11.                 not use, install, modify or redistribute this Apple software.
  12.                 In consideration of your agreement to abide by the following
  13.                 terms, and subject to these terms, Apple grants you a personal,
  14.                 non-exclusive license, under Apple's copyrights in this
  15.                 original Apple software (the "Apple Software"), to use,
  16.                 reproduce, modify and redistribute the Apple Software, with or
  17.                 without modifications, in source and/or binary forms; provided
  18.                 that if you redistribute the Apple Software in its entirety and
  19.                 without modifications, you must retain this notice and the
  20.                 following text and disclaimers in all such redistributions of
  21.                 the Apple Software. Neither the name, trademarks, service marks
  22.                 or logos of Apple Inc. may be used to endorse or promote
  23.                 products derived from the Apple Software without specific prior
  24.                 written permission from Apple.  Except as expressly stated in
  25.                 this notice, no other rights or licenses, express or implied,
  26.                 are granted by Apple herein, including but not limited to any
  27.                 patent rights that may be infringed by your derivative works or
  28.                 by other works in which the Apple Software may be incorporated.
  29.                 The Apple Software is provided by Apple on an "AS IS" basis. 
  30.                 APPLE MAKES NO WARRANTIES, EXPRESS OR IMPLIED, INCLUDING
  31.                 WITHOUT LIMITATION THE IMPLIED WARRANTIES OF NON-INFRINGEMENT,
  32.                 MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE, REGARDING
  33.                 THE APPLE SOFTWARE OR ITS USE AND OPERATION ALONE OR IN
  34.                 COMBINATION WITH YOUR PRODUCTS.
  35.                 IN NO EVENT SHALL APPLE BE LIABLE FOR ANY SPECIAL, INDIRECT,
  36.                 INCIDENTAL OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
  37.                 TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  38.                 DATA, OR PROFITS; OR BUSINESS INTERRUPTION) ARISING IN ANY WAY
  39.                 OUT OF THE USE, REPRODUCTION, MODIFICATION AND/OR DISTRIBUTION
  40.                 OF THE APPLE SOFTWARE, HOWEVER CAUSED AND WHETHER UNDER THEORY
  41.                 OF CONTRACT, TORT (INCLUDING NEGLIGENCE), STRICT LIABILITY OR
  42.                 OTHERWISE, EVEN IF APPLE HAS BEEN ADVISED OF THE POSSIBILITY OF
  43.                 SUCH DAMAGE.
  44. */
  45. #import "SendController.h"
  46. #import "AppDelegate.h"
  47. @interface SendController ()
  48. // Properties that don't need to be seen by the outside world.
  49. @property (nonatomic, readonly) BOOL              isSending;
  50. @property (nonatomic, retain)   NSNetService *    netService;
  51. @property (nonatomic, retain)   NSOutputStream *  networkStream;
  52. @property (nonatomic, retain)   NSInputStream *   fileStream;
  53. @property (nonatomic, readonly) uint8_t *         buffer;
  54. @property (nonatomic, assign)   size_t            bufferOffset;
  55. @property (nonatomic, assign)   size_t            bufferLimit;
  56. @end
  57. @implementation SendController
  58. #pragma mark * Status management
  59. // These methods are used by the core transfer code to update the UI.
  60. - (void)_sendDidStart
  61. {
  62.     self.statusLabel.text = @"Sending";
  63.     self.cancelButton.enabled = YES;
  64.     [self.activityIndicator startAnimating];
  65.     [[AppDelegate sharedAppDelegate] didStartNetworking];
  66. }
  67. - (void)_updateStatus:(NSString *)statusString
  68. {
  69.     assert(statusString != nil);
  70.     self.statusLabel.text = statusString;
  71. }
  72. - (void)_sendDidStopWithStatus:(NSString *)statusString
  73. {
  74.     if (statusString == nil) {
  75.         statusString = @"Send succeeded";
  76.     }
  77.     self.statusLabel.text = statusString;
  78.     self.cancelButton.enabled = NO;
  79.     [self.activityIndicator stopAnimating];
  80.     [[AppDelegate sharedAppDelegate] didStopNetworking];
  81. }
  82. #pragma mark * Core transfer code
  83. // This is the code that actually does the networking.
  84. @synthesize netService    = _netService;
  85. @synthesize networkStream = _networkStream;
  86. @synthesize fileStream    = _fileStream;
  87. @synthesize bufferOffset  = _bufferOffset;
  88. @synthesize bufferLimit   = _bufferLimit;
  89. // Because buffer is declared as an array, you have to use a custom getter.  
  90. // A synthesised getter doesn't compile.
  91. - (uint8_t *)buffer
  92. {
  93.     return self->_buffer;
  94. }
  95. - (BOOL)isSending
  96. {
  97.     return (self.networkStream != nil);
  98. }
  99. - (void)_startSend:(NSString *)filePath
  100. {
  101.     NSOutputStream *    output;
  102.     BOOL                success;
  103.     
  104.     assert(filePath != nil);
  105.     
  106.     assert(self.networkStream == nil);      // don't tap send twice in a row!
  107.     assert(self.fileStream == nil);         // ditto
  108.     // Open a stream for the file we're going to send.
  109.     
  110.     self.fileStream = [NSInputStream inputStreamWithFileAtPath:filePath];
  111.     assert(self.fileStream != nil);
  112.     
  113.     [self.fileStream open];
  114.     
  115.     // Open a stream to the server, finding the server via Bonjour.  Then configure 
  116.     // the stream for async operation.
  117.     
  118.     self.netService = [[[NSNetService alloc] initWithDomain:@"local." type:@"_x-SNSUpload._tcp." name:@"Test"] autorelease];
  119.     assert(self.netService != nil);
  120.     success = [self.netService getInputStream:NULL outputStream:&output];
  121.     assert(success);
  122.     
  123.     self.networkStream = output;
  124.     
  125.     // -[NSNetService getInputStream:outputStream:] currently returns the stream 
  126.     // with a reference that we have to release (something that's counter to the 
  127.     // standard Cocoa memory management rules <rdar://problem/6868813>).
  128.     
  129.     [output release];
  130.     
  131.     self.networkStream.delegate = self;
  132.     [self.networkStream scheduleInRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
  133.     
  134.     [self.networkStream open];
  135.     
  136.     // Tell the UI we're sending.
  137.     
  138.     [self _sendDidStart];
  139. }
  140. - (void)_stopSendWithStatus:(NSString *)statusString
  141. {
  142.     if (self.networkStream != nil) {
  143.         self.networkStream.delegate = nil;
  144.         [self.networkStream removeFromRunLoop:[NSRunLoop currentRunLoop] forMode:NSDefaultRunLoopMode];
  145.         [self.networkStream close];
  146.         self.networkStream = nil;
  147.     }
  148.     if (self.netService != nil) {
  149.         [self.netService stop];
  150.         self.netService = nil;
  151.     }
  152.     if (self.fileStream != nil) {
  153.         [self.fileStream close];
  154.         self.fileStream = nil;
  155.     }
  156.     self.bufferOffset = 0;
  157.     self.bufferLimit  = 0;
  158.     [self _sendDidStopWithStatus:statusString];
  159. }
  160. - (void)stream:(NSStream *)aStream handleEvent:(NSStreamEvent)eventCode
  161.     // An NSStream delegate callback that's called when events happen on our 
  162.     // network stream.
  163. {
  164.     #pragma unused(aStream)
  165.     assert(aStream == self.networkStream);
  166.     switch (eventCode) {
  167.         case NSStreamEventOpenCompleted: {
  168.             [self _updateStatus:@"Opened connection"];
  169.         } break;
  170.         case NSStreamEventHasBytesAvailable: {
  171.             assert(NO);     // should never happen for the output stream
  172.         } break;
  173.         case NSStreamEventHasSpaceAvailable: {
  174.             [self _updateStatus:@"Sending"];
  175.             
  176.             // If we don't have any data buffered, go read the next chunk of data.
  177.             
  178.             if (self.bufferOffset == self.bufferLimit) {
  179.                 NSInteger   bytesRead;
  180.                 
  181.                 bytesRead = [self.fileStream read:self.buffer maxLength:kSendBufferSize];
  182.                 
  183.                 if (bytesRead == -1) {
  184.                     [self _stopSendWithStatus:@"File read error"];
  185.                 } else if (bytesRead == 0) {
  186.                     [self _stopSendWithStatus:nil];
  187.                 } else {
  188.                     self.bufferOffset = 0;
  189.                     self.bufferLimit  = bytesRead;
  190.                 }
  191.             }
  192.             
  193.             // If we're not out of data completely, send the next chunk.
  194.             
  195.             if (self.bufferOffset != self.bufferLimit) {
  196.                 NSInteger   bytesWritten;
  197.                 bytesWritten = [self.networkStream write:&self.buffer[self.bufferOffset] maxLength:self.bufferLimit - self.bufferOffset];
  198.                 assert(bytesWritten != 0);
  199.                 if (bytesWritten == -1) {
  200.                     [self _stopSendWithStatus:@"Network write error"];
  201.                 } else {
  202.                     self.bufferOffset += bytesWritten;
  203.                 }
  204.             }
  205.         } break;
  206.         case NSStreamEventErrorOccurred: {
  207.             [self _stopSendWithStatus:@"Stream open error"];
  208.         } break;
  209.         case NSStreamEventEndEncountered: {
  210.             // ignore
  211.         } break;
  212.         default: {
  213.             assert(NO);
  214.         } break;
  215.     }
  216. }
  217. #pragma mark * Actions
  218. - (IBAction)sendAction:(UIView *)sender
  219. {
  220.     assert( [sender isKindOfClass:[UIView class]] );
  221.     if ( ! self.isSending ) {
  222.         NSString *  filePath;
  223.         
  224.         // Use the tag on the UIButton to determine which image to send.
  225.         
  226.         filePath = [[AppDelegate sharedAppDelegate] pathForTestImage:sender.tag];
  227.         assert(filePath != nil);
  228.         
  229.         [self _startSend:filePath];
  230.     }
  231. }
  232. - (IBAction)cancelAction:(id)sender
  233. {
  234.     #pragma unused(sender)
  235.     [self _stopSendWithStatus:@"Cancelled"];
  236. }
  237. #pragma mark * View controller boilerplate
  238. @synthesize statusLabel       = _statusLabel;
  239. @synthesize activityIndicator = _activityIndicator;
  240. @synthesize cancelButton        = _stopButton;
  241. - (void)viewDidLoad
  242. {
  243.     [super viewDidLoad];
  244.     assert(self.statusLabel != nil);
  245.     assert(self.activityIndicator != nil);
  246.     assert(self.cancelButton != nil);
  247.     
  248.     self.activityIndicator.hidden = YES;
  249.     self.statusLabel.text = @"Tap a picture to start the send";
  250.     self.cancelButton.enabled = NO;
  251. }
  252. - (void)viewDidUnload
  253. {
  254.     [super viewDidUnload];
  255.     self.statusLabel = nil;
  256.     self.activityIndicator = nil;
  257.     self.cancelButton = nil;
  258. }
  259. - (void)dealloc
  260. {
  261.     [self _stopSendWithStatus:@"Stopped"];
  262.     [self->_statusLabel release];
  263.     [self->_activityIndicator release];
  264.     [self->_stopButton release];
  265.     [super dealloc];
  266. }
  267. @end