#include "hb.h"
-#define HB_URL "handbrake.fr"
-#define HB_QUERY "GET /LATEST HTTP/1.0\r\nHost: " HB_URL "\r\n\r\n"
+static void UpdateFunc( void * );
typedef struct
{
} hb_update_t;
-static void UpdateFunc( void * );
-
hb_thread_t * hb_update_init( int * build, char * version )
{
hb_update_t * data = calloc( sizeof( hb_update_t ), 1 );
HB_NORMAL_PRIORITY );
}
+
static void UpdateFunc( void * _data )
{
+
hb_update_t * data = (hb_update_t *) _data;
- hb_net_t * net;
+ char* const url = HB_PROJECT_URL_APPCAST;
+ char* const urlz = url + strlen( HB_PROJECT_URL_APPCAST ); /* marks null-term */
+ char url_host[64];
+ char url_path[128];
+ char query[256];
+
+ hb_net_t * net;
int ret;
- char buf[1024];
- char * cur, * end, * p;
+ char buf[4096];
+ char * cur, * end;
int size;
- int stable, unstable;
- char stable_str[16], unstable_str[16];
+ int i_vers;
+ char s_vers[32]; /* must be no larger than hb_handle_s.version */
int i;
- if( !( net = hb_net_open( HB_URL, 80 ) ) )
+ /* Setup hb_query and hb_query_two with the correct appcast file */
+ hb_log( "Using %s", url );
+
+ /* extract host part */
+ cur = strstr( HB_PROJECT_URL_APPCAST, "//" );
+ if( !cur || cur+2 > urlz )
+ goto error;
+ cur += 2;
+
+ end = strstr( cur, "/" );
+ if( !end || end > urlz )
+ goto error;
+
+ memset( url_host, 0, sizeof(url_host) );
+ strncpy( url_host, cur, (end-cur) );
+
+ /* extract path part */
+ memset( url_path, 0, sizeof(url_path) );
+ strncpy( url_path, end, (urlz-end) );
+
+ if( !strlen( url_path ))
+ goto error;
+
+ memset( query, 0, sizeof(query) );
+ snprintf( query, sizeof(query), "GET %s HTTP/1.0\r\nHost: %s\r\n\r\n", url_path, url_host );
+
+ /* Grab the data from the web server */
+ if( !( net = hb_net_open( url_host, 80 ) ) )
{
goto error;
}
- if( hb_net_send( net, HB_QUERY ) < 0 )
+ if( hb_net_send( net, query ) < 0 )
{
+ hb_log("Error: Unable to connect to server");
hb_net_close( &net );
goto error;
}
size = 0;
- memset( buf, 0, 1024 );
+ memset( buf, 0, 4096 );
for( ;; )
{
ret = hb_net_recv( net, &buf[size], sizeof( buf ) - size );
cur = buf;
end = &buf[sizeof( buf )];
-
+
/* Make sure we got it */
cur += 9;
if( size < 15 || strncmp( cur, "200 OK", 6 ) )
{
- /* Something went wrong */
+ hb_log("Error: We did not get a 200 OK from the server. \n");
goto error;
}
cur += 6;
if( cur >= end )
{
+ hb_log("Error: Found the end of the buffer before the end of the HTTP header information! \n");
goto error;
}
-
- stable = strtol( cur, &p, 10 );
- if( cur == p )
+
+ /*
+ * Find the <cli> tag
+ * Scan though each character of the buffer until we find that the first 4 characters of "cur" are "<cli"
+ */
+ for(i=0 ; &cur[3] < end; i++, cur++ )
{
- goto error;
+ if( cur[0] == 'c' && cur[1] == 'l' && cur[2] == 'i' && cur[3] == '>' )
+ {
+ cur += 1;
+ break;
+ }
+
+ /* If the CLI tag has not been found in the first 768 characters, or the end is reached, something bad happened.*/
+ if (( i > 768) || ( cur >= end ))
+ {
+ hb_log("Error: Did not find the <cli> tag in the expected maximum amount of characters into the file. \n");
+ goto error;
+ }
}
- cur = p + 1;
- memset( stable_str, 0, sizeof( stable_str ) );
- for( i = 0;
- i < sizeof( stable_str ) - 1 && cur < end && *cur != '\n';
- i++, cur++ )
+
+ if( cur >= end )
{
- stable_str[i] = *cur;
+ goto error;
}
-
- hb_log( "latest stable: %s, build %d", stable_str, stable );
-
- cur++;
+
+ /*
+ * Ok, The above code didn't position cur, it only found <cli so we need to shift cur along 3 places.
+ * After which, the next 10 characters are the build number
+ */
+ cur += 3;
+
if( cur >= end )
{
+ hb_log("Error: Unexpected end of buffer! Could not find the build information. \n");
goto error;
}
+
+ /* Stable HB_PROJECT_BUILD */
+ i_vers = strtol( cur, &cur, 10 );
- unstable = strtol( cur, &p, 10 );
- if( cur == p )
+ if( cur >= end )
{
+ hb_log("Error: Unexpected end of buffer! \n");
goto error;
}
- cur = p + 1;
- memset( unstable_str, 0, sizeof( unstable_str ) );
- for( i = 0;
- i < sizeof( unstable_str ) - 1 && cur < end && *cur != '\n';
- i++, cur++ )
+
+ /*
+ * The Version number is 2 places after the build, so shift cur, 2 places.
+ * Get all the characters in cur until the point where " is found.
+ */
+ cur += 2;
+
+ if( cur >= end )
{
- unstable_str[i] = *cur;
+ hb_log("Error: Unexpected end of buffer! Could not get version number. \n");
+ goto error;
}
-
- hb_log( "latest unstable: %s, build %d", unstable_str, unstable );
-
- if( HB_BUILD % 100 )
+ memset( s_vers, 0, sizeof( s_vers ) );
+ for( i = 0; i < sizeof( s_vers ) - 1 && cur < end && *cur != '"'; i++, cur++ )
{
- /* We are runnning an unstable build */
- if( unstable > HB_BUILD )
+ s_vers[i] = *cur;
+
+ /* If the CLI tag has not been found in the first 768 characters, or the end is reached, something bad happened.*/
+ if (( cur >= end ))
{
- memcpy( data->version, unstable_str, sizeof( unstable_str ) );
- *(data->build) = unstable;
+ hb_log("Error: Version number too long, or end of buffer reached. \n");
+ goto error;
}
}
- else
+
+ if( cur >= end )
{
- /* We are runnning an stable build */
- if( stable > HB_BUILD )
- {
- memcpy( data->version, stable_str, sizeof( stable_str ) );
- *(data->build) = stable;
- }
+ goto error;
+ }
+
+ /* Print the version information */
+ hb_log( "latest: %s, build %d", s_vers, i_vers );
+
+ /* Return the build information */
+ if( i_vers > HB_PROJECT_BUILD )
+ {
+ memcpy( data->version, s_vers, sizeof(s_vers) );
+ *(data->build) = i_vers;
}
error:
free( data );
return;
}
-
OSDN Git Service
