]> git.mxchange.org Git - flightgear.git/blobdiff - utils/TerraSync/terrasync.cxx
Maik Justus: Crash Fix.
[flightgear.git] / utils / TerraSync / terrasync.cxx
index 1786be923c3492a1cec48dc34d14dc757b704127..8063b3c3720bf5a193cfc2b2ffa8a4806589daf3 100644 (file)
@@ -36,6 +36,8 @@
 
 #include <iostream>
 #include <string>
+#include <deque>
+#include <map>
 
 #include <plib/netSocket.h>
 #include <plib/ul.h>
 #include <simgear/bucket/newbucket.hxx>
 #include <simgear/misc/sg_path.hxx>
 
+#ifdef HAVE_SVN_CLIENT_H
+#  ifdef HAVE_LIBSVN_CLIENT_1
+#    include <svn_auth.h>
+#    include <svn_client.h>
+#    include <svn_cmdline.h>
+#    include <svn_pools.h>
+#  else
+#    undef HAVE_SVN_CLIENT_H
+#  endif
+#endif
+
 using std::string;
 using std::cout;
 using std::endl;
 
+const char* source_base = NULL;
 const char* svn_base =
   "http://terrascenery.googlecode.com/svn/trunk/data/Scenery";
 const char* rsync_base = "scenery.flightgear.org::Scenery";
-const char* source_base = NULL;
 const char* dest_base = "terrasyncdir";
-bool use_svn = false;
-
-const char* svn_cmd = "svn checkout";
 const char* rsync_cmd = 
     "rsync --verbose --archive --delete --perms --owner --group";
 
+#ifdef HAVE_SVN_CLIENT_H
+bool use_svn = true;
+#else
+bool use_svn = false;
+const char* svn_cmd = "svn checkout";
+#endif
 
 // display usage
 static void usage( const string& prog ) {
     cout << "Usage: " << endl
          << prog << " -p <port> "
-        << "[ -R ] [ -s <rsync_source> ] -d <dest>" << endl
+        << "-R [ -s <rsync_source> ] -d <dest>" << endl
          << prog << " -p <port> "
-         << "  -S   [ -s <svn_source> ] -d <dest>" << endl;
+         << "-S [ -s <svn_source> ] -d <dest>" << endl;
+#ifdef HAVE_SVN_CLIENT_H
+    cout << "    (defaults to the built in subversion)" << endl;
+#else
+    cout << "    (defaults to rsync, using external commands)" << endl;
+#endif
+}
+
+std::deque<std::string> waitingTiles;
+typedef std::map<std::string,time_t> CompletedTiles;
+CompletedTiles completedTiles;
+
+#ifdef HAVE_SVN_CLIENT_H
+
+// Things we need for doing subversion checkout - often
+apr_pool_t *mysvn_pool = NULL;
+svn_client_ctx_t *mysvn_ctx = NULL;
+svn_opt_revision_t *mysvn_rev = NULL;
+
+static const svn_version_checklist_t mysvn_checklist[] = {
+    { "svn_subr",   svn_subr_version },
+    { "svn_client", svn_client_version },
+    { "svn_wc",     svn_wc_version },
+    { "svn_ra",     svn_ra_version },
+    { "svn_delta",  svn_delta_version },
+    { "svn_diff",   svn_diff_version },
+    { NULL, NULL }
+};
+
+// Configure our subversion session
+int mysvn_setup(void) {
+    // Are we already prepared?
+    if (mysvn_pool) return EXIT_SUCCESS;
+    // No, so initialize svn internals generally
+#ifdef _MSC_VER
+    // there is a segfault when providing an error stream.
+    //  Apparently, calling setvbuf with a nul buffer is
+    //  not supported under msvc 7.1 ( code inside svn_cmdline_init )
+    if (svn_cmdline_init("terrasync", 0) != EXIT_SUCCESS)
+        return EXIT_FAILURE;
+#else
+    if (svn_cmdline_init("terrasync", stderr) != EXIT_SUCCESS)
+        return EXIT_FAILURE;
+#endif
+    apr_pool_t *pool;
+    apr_pool_create(&pool, NULL);
+    svn_error_t *err = NULL;
+    SVN_VERSION_DEFINE(mysvn_version);
+    err = svn_ver_check_list(&mysvn_version, mysvn_checklist);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    err = svn_ra_initialize(pool);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    char *config_dir = NULL;
+    err = svn_config_ensure(config_dir, pool);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    err = svn_client_create_context(&mysvn_ctx, pool);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    err = svn_config_get_config(&(mysvn_ctx->config),
+        config_dir, pool);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    svn_config_t *cfg;
+    cfg = ( svn_config_t*) apr_hash_get(
+        mysvn_ctx->config,
+       SVN_CONFIG_CATEGORY_CONFIG,
+        APR_HASH_KEY_STRING);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    svn_auth_baton_t *ab;
+    err = svn_cmdline_setup_auth_baton(&ab,
+        TRUE, NULL, NULL, config_dir, TRUE, cfg,
+        mysvn_ctx->cancel_func, mysvn_ctx->cancel_baton, pool);
+    if (err)
+        return svn_cmdline_handle_exit_error(err, pool, "terrasync: ");
+    mysvn_ctx->auth_baton = ab;
+    mysvn_ctx->conflict_func = NULL;
+    mysvn_ctx->conflict_baton = NULL;
+    mysvn_rev = (svn_opt_revision_t*) apr_palloc(pool, 
+        sizeof(svn_opt_revision_t));
+    if (!mysvn_rev)
+        return EXIT_FAILURE;
+    mysvn_rev->kind = svn_opt_revision_head;
+    // Success if we got this far
+    mysvn_pool = pool;
+    return EXIT_SUCCESS;
+}
+
+#endif
+
+// sync one directory tree
+void sync_tree(const char* dir) {
+    int rc;
+    char command[512];
+    SGPath path( dest_base );
+
+    path.append( dir );
+    rc = path.create_dir( 0755 );
+    if (rc) {
+        cout << "Return code = " << rc << endl;
+        exit(1);
+    }
+
+    if (use_svn) {
+#ifdef HAVE_SVN_CLIENT_H
+        cout << dir << " ... ";
+       cout.flush();
+        char dest_base_dir[512];
+        snprintf( command, 512,
+            "%s/%s", source_base, dir);
+        snprintf( dest_base_dir, 512,
+            "%s/%s", dest_base, dir);
+       svn_error_t *err = NULL;
+       if (mysvn_setup() != EXIT_SUCCESS)
+           exit(1);
+       apr_pool_t *subpool = svn_pool_create(mysvn_pool);
+       err = svn_client_checkout(NULL,
+           command,
+           dest_base_dir,
+           mysvn_rev,
+           1,
+           mysvn_ctx,
+           subpool);
+       if (err) {
+           // Report errors from the checkout attempt
+           cout << "failed: " << endl
+                << err->message << endl;
+           svn_error_clear(err);
+           return;
+       } else {
+           cout << "done" << endl;
+       }
+       svn_pool_destroy(subpool);
+       return;
+#else
+
+        snprintf( command, 512,
+            "%s %s/%s %s/%s", svn_cmd,
+            source_base, dir,
+           dest_base, dir );
+#endif
+    } else {
+        snprintf( command, 512,
+            "%s %s/%s/ %s/%s/", rsync_cmd,
+            source_base, dir,
+           dest_base, dir );
+    }
+    cout << command << endl;
+    rc = system( command );
+    if (rc) {
+        cout << "Return code = " << rc << endl;
+        if (rc == 5120) exit(1);
+    }
 }
 
 
@@ -78,8 +249,14 @@ static void parse_message( const string &msg, int *lat, int *lon ) {
 
     // find GGA string and advance to start of lat
     string::size_type pos = text.find( "$GPGGA" );
+    if ( pos == string::npos )
+    {
+       *lat = -9999.0;
+       *lon = -9999.0;
+       return;
+    }
     string tmp = text.substr( pos + 7 );
-    pos = text.find( "," );
+    pos = tmp.find( "," );
     tmp = tmp.substr( pos + 1 );
     // cout << "-> " << tmp << endl;
 
@@ -123,44 +300,13 @@ static void parse_message( const string &msg, int *lat, int *lon ) {
       *lon = nowhere;
       *lat = nowhere;
     }
-
-}
-
-// sync one directory tree
-void sync_tree(char* dir) {
-    int rc;
-    char command[512];
-    SGPath path( dest_base );
-
-    path.append( dir );
-    rc = path.create_dir( 0755 );
-    if (rc) {
-        cout << "Return code = " << rc << endl;
-        exit(1);
-    }
-
-    if (use_svn) {
-        snprintf( command, 512,
-            "%s %s/%s %s/%s", svn_cmd,
-            source_base, dir,
-           dest_base, dir );
-    } else {
-      snprintf( command, 512,
-          "%s %s/%s/ %s/%s/", rsync_cmd,
-          source_base, dir,
-         dest_base, dir );
-    }
-    cout << command << endl;
-    rc = system( command );
-    if (rc) {
-        cout << "Return code = " << rc << endl;
-        if (rc == 5120) exit(1);
-    }
 }
 
 
 // sync area
 static void sync_area( int lat, int lon ) {
+    if ( lat < -90 || lat > 90 || lon < -180 || lon > 180 )
+        return;
     char NS, EW;
     int baselat, baselon;
 
@@ -185,20 +331,19 @@ static void sync_area( int lat, int lon ) {
         }
         EW = 'w';
     } else {
-            baselon = (int)(lon / 10) * 10;
-            EW = 'e';
-        }
-    
+        baselon = (int)(lon / 10) * 10;
+        EW = 'e';
+    }
+
     const char* terrainobjects[3] = { "Terrain", "Objects", 0 };
-    const char** tree;
-    char dir[512];
-  
-    for (tree = &terrainobjects[0]; *tree; tree++) {
-        snprintf( dir, 512, "%s/%c%03d%c%02d/%c%03d%c%02d",
-             *tree,
-             EW, abs(baselon), NS, abs(baselat),
-             EW, abs(lon), NS, abs(lat) );
-       sync_tree(dir);
+    
+    for (const char** tree = &terrainobjects[0]; *tree; tree++) {
+       char dir[512];
+       snprintf( dir, 512, "%s/%c%03d%c%02d/%c%03d%c%02d",
+               *tree,
+               EW, abs(baselon), NS, abs(baselat),
+               EW, abs(lon), NS, abs(lat) );
+       waitingTiles.push_back( dir );
     }
 }
 
@@ -231,10 +376,24 @@ static void sync_areas( int lat, int lon, int lat_dir, int lon_dir ) {
     }
 }
 
+void getWaitingTile() {
+    while ( !waitingTiles.empty() ) {
+       CompletedTiles::iterator ii = completedTiles.find( waitingTiles.front() );
+       time_t now = time(0);
+       if ( ii == completedTiles.end() || ii->second + 600 < now ) {
+           sync_tree(waitingTiles.front().c_str());
+           completedTiles[ waitingTiles.front() ] = now;
+           waitingTiles.pop_front();
+           break;
+       }
+       waitingTiles.pop_front();
+    }
+}
 
 int main( int argc, char **argv ) {
     int port = 5501;
     char host[256] = "";        // accept messages from anyone
+    bool testing = false;
 
     // parse arguments
     int i = 1;
@@ -253,8 +412,7 @@ int main( int argc, char **argv ) {
         } else if ( (string)argv[i] == "-S" ) {
            use_svn = true;
         } else if ( (string)argv[i] == "-T" ) {
-          sync_areas( 37, -123, 0, 0 );
-         exit(0);
+           testing = true;
         } else {
             usage( argv[0] );
             exit(-1);        
@@ -269,6 +427,12 @@ int main( int argc, char **argv ) {
        else
            source_base = rsync_base;
     }
+    
+    // We just want one grid square, no FGFS communications
+    if (testing) {
+          sync_areas( 37, -123, 0, 0 );
+         exit(0);
+    }
 
     // Must call this before any other net stuff
     netInit( &argc,argv );
@@ -294,7 +458,7 @@ int main( int argc, char **argv ) {
     int last_lat = nowhere;
     int last_lon = nowhere;
     bool recv_msg = false;
-    int synced_other = 0;
+    char synced_other = 'K';
 
     while ( true ) {
         recv_msg = false;
@@ -303,11 +467,13 @@ int main( int argc, char **argv ) {
             recv_msg = true;
 
             parse_message( msg, &lat, &lon );
-            cout << "pos in msg = " << lat << "," << lon << endl;
         }
 
         if ( recv_msg ) {
             if ( lat != last_lat || lon != last_lon ) {
+               cout << "pos in msg = " << lat << "," << lon << endl;
+               std::deque<std::string> oldRequests;
+               oldRequests.swap( waitingTiles );
                 int lat_dir, lon_dir, dist;
                 if ( last_lat == nowhere || last_lon == nowhere ) {
                     lat_dir = lon_dir = 0;
@@ -329,24 +495,28 @@ int main( int argc, char **argv ) {
                 cout << "lat_dir = " << lat_dir << "  "
                     << "lon_dir = " << lon_dir << endl;
                 sync_areas( lat, lon, lat_dir, lon_dir );
-            } else if ( last_lat == nowhere || last_lon == nowhere ) {
-               cout << "Waiting for FGFS to finish startup" << endl;
+               while ( !oldRequests.empty() ) {
+                   waitingTiles.push_back( oldRequests.front() );
+                   oldRequests.pop_front();
+               }
+           } else if ( !waitingTiles.empty() ) {
+               getWaitingTile();
            } else {
-               switch (synced_other++) {
-               case 0:
-                   sync_tree((char*) "Airports/K");
-                   break;
-               case 1:
-                   sync_tree((char*) "Airports");
-                   break;
-               default:
-                   cout << "Done non-tile syncs" << endl;
-                   break;
+               if ( last_lat == nowhere || last_lon == nowhere ) {
+                   cout << "Waiting for FGFS to finish startup" << endl;
                }
+               char c;
+               while ( !isdigit( c = synced_other++ ) && !isupper( c ) );
+
+               char dir[512];
+               snprintf( dir, 512, "Airports/%c", c );
+               waitingTiles.push_back( dir );
            }
 
             last_lat = lat;
             last_lon = lon;
+       } else if ( !waitingTiles.empty() ) {
+           getWaitingTile();
         } 
 
         ulSleep( 1 );