Make sure a signature is present to ensure accurate error reporting

Fixes #6
0xced · Aug 9, 2013 · 1f0559e · 1f0559e
1 parent ba452f4
commit 1f0559e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/XCDYouTubeVideoPlayerViewController/XCDYouTubeVideoPlayerViewController.m b/XCDYouTubeVideoPlayerViewController/XCDYouTubeVideoPlayerViewController.m
@@ -224,9 +224,10 @@ - (NSURL *) videoURLWithData:(NSData *)data error:(NSError **)error
 		NSDictionary *stream = DictionaryWithQueryString(streamQuery, queryEncoding);
 		NSString *type = stream[@"type"];
 		NSString *urlString = stream[@"url"];
-		if (urlString && [AVURLAsset isPlayableExtendedMIMEType:type])
+		NSString *signature = stream[@"sig"];
+		if (urlString && signature && [AVURLAsset isPlayableExtendedMIMEType:type])
 		{
-			NSURL *streamURL = [NSURL URLWithString:[NSString stringWithFormat:@"%@&signature=%@", urlString, stream[@"sig"]]];
+			NSURL *streamURL = [NSURL URLWithString:[NSString stringWithFormat:@"%@&signature=%@", urlString, signature]];
 			streamURLs[@([stream[@"itag"] integerValue])] = streamURL;
 		}
 	}