Initial commit of Seafile server core.

2025-09-02 07:54:27 +00:00 · 2016-08-10 14:53:33 +08:00
commit 2643119433
352 changed files with 85573 additions and 0 deletions
--- a/server/processors/recvcommit-v2-proc.c
+++ b/server/processors/recvcommit-v2-proc.c
@@ -0,0 +1,180 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+
+#include "common.h"
+
+#include <fcntl.h>
+
+#include <ccnet.h>
+#include "net.h"
+#include "utils.h"
+
+#include "seafile-session.h"
+#include "recvcommit-v2-proc.h"
+#include "processors/objecttx-common.h"
+#include "seaf-utils.h"
+
+#include "log.h"
+
+enum {
+    INIT,
+    RECV_OBJECT
+};
+
+typedef struct {
+    guint32 writer_id;
+    gboolean registered;
+} RecvcommitPriv;
+
+#define GET_PRIV(o)  \
+   (G_TYPE_INSTANCE_GET_PRIVATE ((o), SEAFILE_TYPE_RECVCOMMIT_V2_PROC, RecvcommitPriv))
+
+#define USE_PRIV \
+    RecvcommitPriv *priv = GET_PRIV(processor);
+
+static int recv_commit_start (CcnetProcessor *processor, int argc, char **argv);
+static void handle_update (CcnetProcessor *processor,
+                           char *code, char *code_msg,
+                           char *content, int clen);
+static void
+write_done_cb (OSAsyncResult *res, void *cb_data);
+
+
+G_DEFINE_TYPE (SeafileRecvcommitV2Proc, seafile_recvcommit_v2_proc, CCNET_TYPE_PROCESSOR)
+
+static void
+release_resource (CcnetProcessor *processor)
+{
+    USE_PRIV;
+
+    if (priv->registered)
+        seaf_obj_store_unregister_async_write (seaf->commit_mgr->obj_store,
+                                               priv->writer_id);
+
+    CCNET_PROCESSOR_CLASS (seafile_recvcommit_v2_proc_parent_class)->release_resource (processor);
+}
+
+static void
+seafile_recvcommit_v2_proc_class_init (SeafileRecvcommitV2ProcClass *klass)
+{
+    CcnetProcessorClass *proc_class = CCNET_PROCESSOR_CLASS (klass);
+
+    proc_class->name = "recvcommit-v2-proc";
+    proc_class->start = recv_commit_start;
+    proc_class->handle_update = handle_update;
+    proc_class->release_resource = release_resource;
+
+    g_type_class_add_private (klass, sizeof (RecvcommitPriv));
+}
+
+static void
+seafile_recvcommit_v2_proc_init (SeafileRecvcommitV2Proc *processor)
+{
+}
+
+static int
+recv_commit_start (CcnetProcessor *processor, int argc, char **argv)
+{
+    USE_PRIV;
+    char *session_token;
+
+    if (argc != 2) {
+        ccnet_processor_send_response (processor, SC_BAD_ARGS, SS_BAD_ARGS, NULL, 0);
+        ccnet_processor_done (processor, FALSE);
+        return -1;
+    }
+
+    session_token = argv[1];
+    if (seaf_token_manager_verify_token (seaf->token_mgr,
+                                         processor->peer_id,
+                                         session_token, NULL) == 0) {
+        ccnet_processor_send_response (processor, SC_OK, SS_OK, NULL, 0);
+        processor->state = RECV_OBJECT;
+        priv->writer_id =
+            seaf_obj_store_register_async_write (seaf->commit_mgr->obj_store,
+                                                 write_done_cb,
+                                                 processor);
+        priv->registered = TRUE;
+        return 0;
+    } else {
+        ccnet_processor_send_response (processor, 
+                                       SC_ACCESS_DENIED, SS_ACCESS_DENIED,
+                                       NULL, 0);
+        ccnet_processor_done (processor, FALSE);
+        return -1;
+    }
+}
+
+static void
+write_done_cb (OSAsyncResult *res, void *cb_data)
+{
+    CcnetProcessor *processor = cb_data;
+
+    if (!res->success) {
+        ccnet_processor_send_response (processor, SC_BAD_OBJECT, SS_BAD_OBJECT,
+                                       NULL, 0);
+        seaf_warning ("[recvcommit] Failed to write commit object.\n");
+        ccnet_processor_done (processor, FALSE);
+    }
+    /* FIXME: need to send ACK if success. */
+}
+
+static int
+save_commit (CcnetProcessor *processor, ObjectPack *pack, int len)
+{
+    USE_PRIV;
+
+    return seaf_obj_store_async_write (seaf->commit_mgr->obj_store,
+                                       priv->writer_id,
+                                       pack->id,
+                                       pack->object,
+                                       len - 41);
+}
+
+static void
+receive_commit (CcnetProcessor *processor, char *content, int clen)
+{
+    ObjectPack *pack = (ObjectPack *)content;
+
+    if (clen < sizeof(ObjectPack)) {
+        seaf_warning ("[recvcommit] invalid object id.\n");
+        goto bad;
+    }
+
+    g_debug ("[recvcommit] recv commit object %s\n", pack->id);
+
+    if (save_commit (processor, pack, clen) < 0) {
+        goto bad;
+    }
+
+    return;
+
+bad:
+    ccnet_processor_send_response (processor, SC_BAD_OBJECT, SS_BAD_OBJECT,
+                                   NULL, 0);
+    seaf_warning ("[recvcommit] Failed to write commit object.\n");
+    ccnet_processor_done (processor, FALSE);
+}
+
+static void handle_update (CcnetProcessor *processor,
+                           char *code, char *code_msg,
+                           char *content, int clen)
+{
+    switch (processor->state) {
+    case RECV_OBJECT:
+        if (strncmp(code, SC_OBJECT, 3) == 0) {
+            receive_commit (processor, content, clen);
+        } else if (strncmp(code, SC_END, 3) == 0) {
+            g_debug ("[recvcommit] Recv commit end.\n");
+            ccnet_processor_done (processor, TRUE);
+        } else {
+            seaf_warning ("[recvcommit] Bad update: %s %s\n", code, code_msg);
+            ccnet_processor_send_response (processor,
+                                           SC_BAD_UPDATE_CODE, SS_BAD_UPDATE_CODE,
+                                           NULL, 0);
+            ccnet_processor_done (processor, FALSE);
+        }
+        break;
+    default:
+        g_return_if_reached ();
+    }
+}