Small bugfixes for http-push.c

[git.git] / http-push.c

diff --git a/http-push.c b/http-push.c
index c10067c..0b90fb9 100644 (file)
--- a/http-push.c
+++ b/http-push.c
@@ -7,7 +7,7 @@
 
 #include <curl/curl.h>
 #include <curl/easy.h>
-#include "expat.h"
+#include <expat.h>
 
 static const char http_push_usage[] =
 "git-http-push [--complete] [--force] [--verbose] <url> <ref> [<ref>...]\n";
@@ -28,6 +28,15 @@ static const char http_push_usage[] =
 #define NO_CURL_EASY_DUPHANDLE
 #endif
 
+#ifndef XML_STATUS_OK
+enum XML_Status {
+  XML_STATUS_OK = 1,
+  XML_STATUS_ERROR = 0
+};
+#define XML_STATUS_OK    1
+#define XML_STATUS_ERROR 0
+#endif
+
 #define RANGE_HEADER_SIZE 30
 
 /* DAV method names and request body templates */
@@ -47,6 +56,7 @@ static int active_requests = 0;
 static int data_received;
 static int pushing = 0;
 static int aborted = 0;
+static char remote_dir_exists[256];
 
 #ifdef USE_CURL_MULTI
 static int max_requests = -1;
@@ -650,6 +660,7 @@ static void finish_request(struct transfer_request *request)
                if (request->http_code == 404) {
                        request->state = NEED_PUSH;
                } else if (request->curl_result == CURLE_OK) {
+                       remote_dir_exists[request->sha1[0]] = 1;
                        request->state = COMPLETE;
                } else {
                        fprintf(stderr, "HEAD %s failed, aborting (%d/%ld)\n",
@@ -661,6 +672,7 @@ static void finish_request(struct transfer_request *request)
        } else if (request->state == RUN_MKCOL) {
                if (request->curl_result == CURLE_OK ||
                    request->http_code == 405) {
+                       remote_dir_exists[request->sha1[0]] = 1;
                        start_put(request);
                } else {
                        fprintf(stderr, "MKCOL %s failed, aborting (%d/%ld)\n",
@@ -728,11 +740,12 @@ void process_curl_messages(void)
                               slot->curl != curl_message->easy_handle)
                                slot = slot->next;
                        if (slot != NULL) {
+                               int curl_result = curl_message->data.result;
                                curl_multi_remove_handle(curlm, slot->curl);
                                active_requests--;
                                slot->done = 1;
                                slot->in_use = 0;
-                               slot->curl_result = curl_message->data.result;
+                               slot->curl_result = curl_result;
                                curl_easy_getinfo(slot->curl,
                                                  CURLINFO_HTTP_CODE,
                                                  &slot->http_code);
@@ -767,7 +780,10 @@ void process_request_queue(void)
                        start_check(request);
                        curl_multi_perform(curlm, &num_transfers);
                } else if (pushing && request->state == NEED_PUSH) {
-                       start_mkcol(request);
+                       if (remote_dir_exists[request->sha1[0]])
+                               start_put(request);
+                       else
+                               start_mkcol(request);
                        curl_multi_perform(curlm, &num_transfers);
                }
                request = request->next;
@@ -1230,7 +1246,7 @@ struct active_lock *lock_remote(char *file, long timeout)
        in_buffer.posn = 0;
        in_buffer.buffer = in_data;
 
-       new_lock = xmalloc(sizeof(*new_lock));
+       new_lock = xcalloc(1, sizeof(*new_lock));
        new_lock->owner = NULL;
        new_lock->token = NULL;
        new_lock->timeout = -1;
@@ -1599,6 +1615,8 @@ int main(int argc, char **argv)
                break;
        }
 
+       memset(remote_dir_exists, 0, 256);
+
        curl_global_init(CURL_GLOBAL_ALL);
 
 #ifdef USE_CURL_MULTI