#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;
+ 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;
}
-
- // 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_PROJECT_BUILD */
+ i_vers = 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;
-
- // 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( cur >= end )
+ memset( s_vers, 0, sizeof( s_vers ) );
+ for( i = 0; i < sizeof( s_vers ) - 1 && cur < end && *cur != '"'; i++, cur++ )
{
- goto error;
+ 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 ))
+ {
+ hb_log("Error: Version number too long, or end of buffer reached. \n");
+ goto error;
+ }
}
-
- hb_log( "latest stable: %s, build %d", stable_str, stable );
-
- // END OF STABLE INFO ###################################################
-
-
- // 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"
- */
- for(i =0 ; &cur[3] < end; i++, cur++ )
- {
- if( cur[0] == '<' && cur[1] == 'c' && cur[2] == 'l' && cur[3] == 'i' )
- {
- 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.
- */
-
- cur += 13;
-
- if( cur >= end )
- {
- goto error;
- }
-
- unstable = strtol( cur, &p, 10 );
-
- if( cur >= end )
- {
- 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 )
+ if( cur >= end )
{
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 );
-
- // END OF UNSTABLE INFO ###################################################
+ /* Print the version information */
+ hb_log( "latest: %s, build %d", s_vers, i_vers );
- /*
- * Handle the update checking as normal.
- * This code is unchanged.
- */
- if( HB_BUILD % 100 )
+ /* Return the build information */
+ if( i_vers > HB_PROJECT_BUILD )
{
- /* We are runnning an unstable build */
- if( unstable > HB_BUILD )
- {
- memcpy( data->version, unstable_str, sizeof( unstable_str ) );
- *(data->build) = unstable;
- }
- }
- else
- {
- /* We are runnning an stable build */
- if( stable > HB_BUILD )
- {
- memcpy( data->version, stable_str, sizeof( stable_str ) );
- *(data->build) = stable;
- }
+ memcpy( data->version, s_vers, sizeof(s_vers) );
+ *(data->build) = i_vers;
}
error:
free( data );
return;
}
-
OSDN Git Service
