Add nitotv cookies

1 files changed, 67 insertions, 0 deletions

diff --git a/methods/http.cc b/methods/http.cc
index 16c0de611..508cf5727 100644
--- a/methods/http.cc
+++ b/methods/http.cc
@@ -35,6 +35,7 @@
 #include <stddef.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <sys/fcntl.h>
 #include <sys/select.h>
 #include <sys/sysctl.h>
 #include <sys/stat.h>
@@ -50,6 +51,8 @@
 #include <lockdown.h>
 #include <CoreFoundation/CoreFoundation.h>
 #include <CFNetwork/CFNetwork.h>
+#include <TargetConditionals.h>
+
 extern "C" CFDictionaryRef SCDynamicStoreCopyProxies(void *);
 									/*}}}*/
 using namespace std;
@@ -163,6 +166,45 @@ std::unique_ptr<ServerState> HttpMethod::CreateServerState(URI const &uri)/*{{{*
 void HttpMethod::RotateDNS()						/*{{{*/
 {
 }
+
+#if TARGET_OS_TV
+char * createCookie() {
+    FILE *fp;
+    long lSize;
+    char *buffer;
+    const char *tp = "/var/mobile/Documents/nitoStoreToken";
+    if (access(tp, R_OK)) return NULL;
+    fp = fopen ( tp , "rb" );
+    if( !fp ) {
+        perror(tp);
+        return NULL;
+    }
+    fseek( fp , 0L , SEEK_END);
+    lSize = ftell( fp );
+    rewind( fp );
+
+    /* allocate memory for entire content */
+    buffer = static_cast<char*>(calloc( 1, lSize+1 ));
+    if( !buffer ) {
+        fclose(fp);
+        return NULL;
+    }
+
+    /* copy the file into the buffer */
+    if( 1!=fread( buffer , lSize, 1 , fp) )
+    {
+        fclose(fp);
+        free(buffer);
+        return NULL;
+    }
+    /* do your work here, buffer is a string contains the whole text */
+
+    fclose(fp);
+
+    return buffer;
+}
+#endif
+
 									/*}}}*/
 BaseHttpMethod::DealWithHeadersResult HttpMethod::DealWithHeaders(FetchResult &Res, RequestState &Req)/*{{{*/
 {
@@ -253,6 +295,17 @@ int HttpMethod::Loop()
       }
 
       CFStringRef sr = CFStringCreateWithCString(kCFAllocatorDefault, urs.c_str(), se);
+#if TARGET_OS_TV
+      bool isNito = false;
+      if (sr) {
+          CFStringRef nitoBaseURL = CFSTR("https://nito.tv/");
+          size_t nitoURLLength = CFStringGetLength(nitoBaseURL);
+          size_t srLength = CFStringGetLength(sr);
+          if (srLength >= nitoURLLength && CFStringCompareWithOptions(sr, nitoBaseURL, CFRangeMake(0, nitoURLLength), kCFCompareCaseInsensitive) == kCFCompareEqualTo) {
+              isNito = true;
+          }
+      }
+#endif
       CFURLRef ur = CFURLCreateWithString(kCFAllocatorDefault, sr, NULL);
       CFRelease(sr);
       CFHTTPMessageRef hm = CFHTTPMessageCreateRequest(kCFAllocatorDefault, CFSTR("GET"), ur, kCFHTTPVersion1_1);
@@ -290,6 +343,20 @@ int HttpMethod::Loop()
 
       CFHTTPMessageSetHeaderFieldValue(hm, CFSTR("User-Agent"), CFSTR("Telesphoreo APT-HTTP/1.0.592"));
 
+#if TARGET_OS_TV
+      if (isNito) {
+          char *cookie = createCookie();
+          if (cookie) {
+              size_t cookie_len = strlen(cookie);
+              CFStringRef cookieString = CFStringCreateWithBytesNoCopy(kCFAllocatorDefault, (unsigned char *)cookie, cookie_len, kCFStringEncodingUTF8, false, kCFAllocatorDefault);
+              if (cookieString) {
+                  CFHTTPMessageSetHeaderFieldValue(hm, CFSTR("Set-Cookie"), cookieString);
+                  CFRelease(cookieString);
+              }
+          }
+      }
+#endif
+
       CFReadStreamRef rs = CFReadStreamCreateForHTTPRequest(kCFAllocatorDefault, hm);
       CFRelease(hm);