#include "hb.h"
-#define HB_URL "handbrake.fr"
-#define HB_QUERY "GET /appcast.xml 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;
+ hb_update_t * data = (hb_update_t *) _data;
+
+ char * hb_query, * hb_query_two;
+ 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;
+
+ /* Setup hb_query and hb_query_two with the correct appcast file */
+ if( HB_BUILD % 100 )
+ {
+ hb_log("Using http://handbrake.fr/appcast_unstable.xml (primary)");
+ hb_log("Using http://handbrake.fr/appcast.xml (secondary)");
+ }
+ else
+ {
+ hb_log("Using http://handbrake.fr/appcast.xml (primary)");
+ hb_log("Using http://handbrake.fr/appcast_unstable.xml (secondary)");
+ }
+
+ hb_query = "GET /appcast.xml HTTP/1.0\r\nHost: handbrake.fr\r\n\r\n";
+ hb_query_two = "GET /appcast_unstable.xml HTTP/1.0\r\nHost: handbrake.fr\r\n\r\n";
- if( !( net = hb_net_open( HB_URL, 80 ) ) )
+ /* Grab the data from the web server */
+ if( !( net = hb_net_open( "handbrake.fr", 80 ) ) )
{
goto error;
}
- if( hb_net_send( net, HB_QUERY ) < 0 )
+ if( hb_net_send( net, hb_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;
}
-
- // FIND THE STABLE VERSION INFORMATION ###################################################
-
- /*
- * Find the <cli-stable> 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++ )
- {
- if( cur[0] == '<' && cur[1] == 'c' && cur[2] == 'l' && cur[3] == 'i' )
- {
+ /*
+ * 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++ )
+ {
+ 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 510 characters, or the end is reached, something bad happened.
- if (( i > 510) || ( cur >= end ))
- {
- goto error;
- }
- }
+ /* 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;
+ }
+ }
- if( cur >= end )
+ if( cur >= end )
{
goto error;
}
- /*
- * Ok, The above code didn't position cur, it only found <cli so we need to shift cur along 11 places.
- * After which, the next 10 characters are the build number
- */
- cur += 11;
+ /*
+ * 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 )
+ if( cur >= end )
{
+ hb_log("Error: Unexpected end of buffer! Could not find the build information. \n");
goto error;
}
- stable = strtol( cur, &cur, 10 );
-
- if( cur >= end )
+ /* Stable HB_BUILD */
+ stable = strtol( cur, &cur, 10 );
+
+ if( cur >= end )
{
+ hb_log("Error: Unexpected end of buffer! \n");
goto error;
}
- /*
- * 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;
+ /*
+ * 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 )
+ if( cur >= end )
{
+ hb_log("Error: Unexpected end of buffer! Could not get version number. \n");
goto error;
}
- memset( stable_str, 0, sizeof( stable_str ) );
- for( i = 0; i < sizeof( stable_str ) - 1 && cur < end && *cur != '"'; i++, cur++ )
- {
- stable_str[i] = *cur;
+ memset( stable_str, 0, sizeof( stable_str ) );
+ for( i = 0; i < sizeof( stable_str ) - 1 && cur < end && *cur != '"'; i++, cur++ )
+ {
+ stable_str[i] = *cur;
- // If the version number is longer than 7 characters, or the end is reached, something has gone wrong.
- if (( i > 7) || ( cur >= end ))
- {
- goto error;
- }
- }
+ /* If the CLI tag has not been found in the first 768 characters, or the end is reached, something bad happened.*/
+ if (( i > 7) || ( cur >= end ))
+ {
+ hb_log("Error: Version number too long, or end of buffer reached. \n");
+ goto error;
+ }
+ }
- if( cur >= end )
+ if( cur >= end )
{
goto error;
}
- hb_log( "latest stable: %s, build %d", stable_str, stable );
+ /* HANDLE THE SECOND APPCAST FILE NOW */
+ /* Grab the data from the web server */
+ if( !( net = hb_net_open( "handbrake.fr", 80 ) ) )
+ {
+ goto error;
+ }
- // END OF STABLE INFO ###################################################
+ if( hb_net_send( net, hb_query_two ) < 0 )
+ {
+ hb_log("Error: Unable to connect to server");
+ hb_net_close( &net );
+ goto error;
+ }
+
+ size = 0;
+ memset( buf, 0, 4096 );
+ for( ;; )
+ {
+ ret = hb_net_recv( net, &buf[size], sizeof( buf ) - size );
+ if( ret < 1 )
+ {
+ hb_net_close( &net );
+ break;
+ }
+ size += ret;
+ }
+
+ 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;
- // FIND THE UNSTABLE INFO ###############################################
- /*
- * Find the <cli-unstable> tag
- * Scan though each character of the buffer until we find that the first 4 characters of "cur" are "<cli"
- */
+ /* Find the end of the headers and the beginning of the content */
+ for( ; &cur[3] < end; cur++ )
+ {
+ if( cur[0] == '\r' && cur[1] == '\n' &&
+ cur[2] == '\r' && cur[3] == '\n' )
+ {
+ cur += 4;
+ break;
+ }
+ }
- for(i =0 ; &cur[3] < end; i++, cur++ )
- {
- if( cur[0] == '<' && cur[1] == 'c' && cur[2] == 'l' && cur[3] == 'i' )
- {
+ if( cur >= end )
+ {
+ hb_log("Error: Found the end of the buffer before the end of the HTTP header information! \n");
+ goto error;
+ }
+
+ /*
+ * 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++ )
+ {
+ if( cur[0] == 'c' && cur[1] == 'l' && cur[2] == 'i' && cur[3] == '>' )
+ {
cur += 1;
break;
- }
+ }
- // If the second CLI tag is more than 25 characters forward, or the end is reached, something went wrong.
- if (( i > 25) || ( cur >= end ))
- {
- goto error;
- }
- }
-
- /*
- * Now we need to handle the unstable build information
- * Unstable build number is 29 Characters after the last position used.
- */
+ // 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 += 13;
-
- if( cur >= end )
+ if( cur >= end )
{
goto error;
- }
+ }
- unstable = strtol( cur, &p, 10 );
-
- if( cur >= end )
+ /*
+ * 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;
}
- /*
- * Now we need to get the unstable version number.
- * First move the cur pointer 12 places.
- * Then iterate over cur until " is found. Thats the end of the version number.
- */
- cur += 12;
-
- if( cur >= end )
+ /* UnStable HB_BUILD */
+ unstable = strtol( cur, &cur, 10 );
+
+ if( cur >= end )
{
+ hb_log("Error: Unexpected end of buffer! \n");
goto error;
}
- memset( unstable_str, 0, sizeof( unstable_str ) );
- for( i = 0; i < sizeof( unstable_str ) - 1 && cur < end && *cur != '"'; i++, cur++ )
- {
- unstable_str[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 )
+ {
+ hb_log("Error: Unexpected end of buffer! Could not get version number. \n");
+ goto error;
+ }
+ memset( unstable_str, 0, sizeof( unstable_str ) );
+ for( i = 0; i < sizeof( unstable_str ) - 1 && cur < end && *cur != '"'; i++, cur++ )
+ {
+ unstable_str[i] = *cur;
- // If the version number is greater than 7 chars or the end is reached, something went wrong.
- if (( i > 7) || ( cur >= end ))
- {
- goto error;
- }
- }
-
- hb_log( "latest unstable: %s, build %d", unstable_str, unstable );
+ // If the version number is longer than 7 characters, or the end is reached, something has gone wrong.
+ if (( i > 7) || ( cur >= end ))
+ {
+ hb_log("Error: Version number too long, or end of buffer reached. \n");
+ goto error;
+ }
+ }
- // END OF UNSTABLE INFO ###################################################
-
- /*
- * Handle the update checking as normal.
- * This code is unchanged.
- */
- if( HB_BUILD % 100 )
+ if( cur >= end )
+ {
+ goto error;
+ }
+
+ /* Print the version information */
+ hb_log( "latest stable: %s, build %d", stable_str, stable );
+ hb_log( "latest unstable: %s, build %d", unstable_str, unstable );
+
+ /* Return the build information */
+ if( HB_BUILD % 100 )
{
/* We are runnning an unstable build */
if( unstable > HB_BUILD )
free( data );
return;
}
-
OSDN Git Service
