Remove isamd. It's not been in use for a long time and isamb is better
[idzebra-moved-to-github.git] / index / zserver.c
index 7d992ae..4a44d06 100644 (file)
@@ -1,5 +1,5 @@
-/* $Id: zserver.c,v 1.101 2003-01-13 10:56:09 oleg Exp $
-   Copyright (C) 1995,1996,1997,1998,1999,2000,2001,2002
+/* $Id: zserver.c,v 1.118 2004-08-04 08:35:24 adam Exp $
+   Copyright (C) 1995,1996,1997,1998,1999,2000,2001,2002,2003,2004
    Index Data Aps
 
 This file is part of the Zebra server.
@@ -20,20 +20,23 @@ Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
 02111-1307, USA.
 */
 
-
-
 #include <stdio.h>
 #include <assert.h>
 #include <fcntl.h>
 #ifdef WIN32
 #include <io.h>
 #include <process.h>
+#include <sys/locking.h>
 #else
 #include <unistd.h>
 #endif
 
+#include <errno.h>
 #include <yaz/log.h>
 #include <yaz/ill.h>
+#include <yaz/yaz-util.h>
+
+#include <sys/types.h>
 
 #include "zserver.h"
 
@@ -67,7 +70,7 @@ bend_initresult *bend_init (bend_initrequest *q)
     q->implementation_name = "Zebra Information Server";
     q->implementation_version = "Zebra " ZEBRAVER;
 
-    logf (LOG_DEBUG, "bend_init");
+    yaz_log (LOG_DEBUG, "bend_init");
 
     sob = statserv_getcontrol ();
     if (!(zh = zebra_open (sob->handle)))
@@ -76,6 +79,7 @@ bend_initresult *bend_init (bend_initrequest *q)
        r->errcode = 1;
        return r;
     }
+    r->handle = zh;
     if (q->auth)
     {
        if (q->auth->which == Z_IdAuthentication_open)
@@ -90,15 +94,20 @@ bend_initresult *bend_init (bend_initrequest *q)
            }
            xfree (openpass);
        }
+       else if (q->auth->which == Z_IdAuthentication_idPass)
+       {
+           Z_IdPass *idPass = q->auth->u.idPass;
+
+           user = idPass->userId;
+           passwd = idPass->password;
+       }
     }
     if (zebra_auth (zh, user, passwd))
     {
        r->errcode = 222;
        r->errstring = user;
-       zebra_close (zh);
        return r;
     }
-    r->handle = zh;
     if (q->charneg_request) /* characater set and langauge negotiation? */
     {
         char **charsets = 0;
@@ -107,60 +116,54 @@ bend_initresult *bend_init (bend_initrequest *q)
         int num_langs = 0;
         int selected = 0;
         int i;
-        char *right_name;
+        NMEM nmem = nmem_create();
 
-        NMEM nmem = nmem_create ();
         yaz_log (LOG_LOG, "character set and language negotiation");
 
         yaz_get_proposal_charneg (nmem, q->charneg_request,
                                   &charsets, &num_charsets,
                                   &langs, &num_langs, &selected);
-               
-        right_name = (char *)nmem_malloc(nmem, 32);
-               
+        
         for (i = 0; i < num_charsets; i++)
         {
-           *right_name = '\0';
-               
+            const char *right_name = "";
            /*
             * FIXME! It is like rudiment :-))
             * We have to support this short names of character sets,
             * because a lot servers in Russia to use own in during
             * character set and language negotiation still.
             */
-
+            
             if (!yaz_matchstr(charsets[i], "win")) {
-                   strcpy(right_name, "WINDOWS-1251");
-               } else if (!yaz_matchstr(charsets[i], "koi")) {
-                   strcpy(right_name, "KOI8-R");
+                right_name = "WINDOWS-1251";
+            } else if (!yaz_matchstr(charsets[i], "koi")) {
+                right_name = "KOI8-R";
             } else if (!yaz_matchstr(charsets[i], "iso")) {
-                   strcpy(right_name, "ISO-8859-5");
+                right_name = "ISO-8859-5";
             } else if (!yaz_matchstr(charsets[i], "dos")) {
-                   strcpy(right_name, "CP866");
-               } else if (!yaz_matchstr(charsets[i], "uni")) {
-                   strcpy(right_name, "UTF-8");
-               } else {
-                   strcpy(right_name, charsets[i]);
-        }
-               
-        logf (LOG_LOG, "charset %d %s (right name is %s)", i,
-           charsets[i], right_name);
-            
-        if (odr_set_charset (q->decode, "UTF-8", right_name) == 0)
-        {
-           odr_set_charset (q->stream, right_name, "UTF-8");
-            if (selected)
-               zebra_record_encoding (zh, right_name);
+                right_name = "CP866";
+            } else if (!yaz_matchstr(charsets[i], "uni")) {
+                right_name = "UTF-8";
+            } else {
+                right_name = charsets[i];
+            }
+            if (odr_set_charset (q->decode, "UTF-8", right_name) == 0)
+            {
+                yaz_log (LOG_LOG, "charset %d %s (proper name %s): OK", i,
+                         charsets[i], right_name);
+                odr_set_charset (q->stream, right_name, "UTF-8");
+                if (selected)
+                    zebra_record_encoding (zh, right_name);
                q->charneg_response =
-                   yaz_set_response_charneg (q->stream, right_name,
+                   yaz_set_response_charneg (q->stream, charsets[i],
                                               0, selected);
                break;
             } else {
-               yaz_log(LOG_LOG, "odr_set_charset(): charset name %s unsupported",
-                       right_name);
+                yaz_log (LOG_LOG, "charset %d %s (proper name %s): unsupported", i,
+                         charsets[i], right_name);
             }
         }
-        nmem_destroy (nmem);
+        nmem_destroy(nmem);
     }
     return r;
 }
@@ -263,11 +266,11 @@ int bend_search (void *handle, bend_search_rr *r)
         zebra_result (zh, &r->errcode, &r->errstring);
         return 0;
     }
-    logf (LOG_LOG, "ResultSet '%s'", r->setname);
+    yaz_log (LOG_LOG, "ResultSet '%s'", r->setname);
     switch (r->query->which)
     {
     case Z_Query_type_1: case Z_Query_type_101:
-       zebra_search_rpn (zh, r->decode, r->stream, r->query->u.type_1,
+       zebra_search_RPN (zh, r->stream, r->query->u.type_1,
                          r->setname, &r->hits);
         zebra_result (zh, &r->errcode, &r->errstring);
         if (!r->errcode)
@@ -326,6 +329,11 @@ static int bend_scan (void *handle, bend_scan_rr *r)
         zebra_result (zh, &r->errcode, &r->errstring);
         return 0;
     }
+    if (r->step_size != 0 && *r->step_size != 0) {
+       r->errcode = 205; /* "Only zero step size supported for Scan" */
+       r->errstring = 0;
+        return 0;
+    }
     r->entries = (struct scan_entry *)
        odr_malloc (r->stream, sizeof(*r->entries) * r->num_entries);
     zebra_scan (zh, r->stream, r->term,
@@ -375,6 +383,10 @@ int bend_delete (void *handle, bend_delete_rr *rr)
 
 static int es_admin_request (ZebraHandle zh, Z_AdminEsRequest *r)
 {
+    if (r->toKeep->databaseName)
+    {
+       yaz_log(LOG_LOG, "adm request database %s", r->toKeep->databaseName);
+    }
     switch (r->toKeep->which)
     {
     case Z_ESAdminOriginPartToKeep_reIndex:
@@ -385,10 +397,11 @@ static int es_admin_request (ZebraHandle zh, Z_AdminEsRequest *r)
        break;
     case Z_ESAdminOriginPartToKeep_drop:
        yaz_log(LOG_LOG, "adm-drop");
+       zebra_drop_database (zh, r->toKeep->databaseName);
        break;
     case Z_ESAdminOriginPartToKeep_create:
        yaz_log(LOG_LOG, "adm-create %s", r->toKeep->databaseName);
-       zebra_admin_create (zh, r->toKeep->databaseName);
+       zebra_create_database (zh, r->toKeep->databaseName);
        break;
     case Z_ESAdminOriginPartToKeep_import:
        yaz_log(LOG_LOG, "adm-import");
@@ -400,6 +413,9 @@ static int es_admin_request (ZebraHandle zh, Z_AdminEsRequest *r)
        break;
     case Z_ESAdminOriginPartToKeep_commit:
        yaz_log(LOG_LOG, "adm-commit");
+       if (r->toKeep->databaseName)
+           zebra_select_database(zh, r->toKeep->databaseName);
+       zebra_commit(zh);
        break;
     case Z_ESAdminOriginPartToKeep_shutdown:
        yaz_log(LOG_LOG, "shutdown");
@@ -412,10 +428,6 @@ static int es_admin_request (ZebraHandle zh, Z_AdminEsRequest *r)
     default:
        yaz_log(LOG_LOG, "unknown admin");
     }
-    if (r->toKeep->databaseName)
-    {
-       yaz_log(LOG_LOG, "database %s", r->toKeep->databaseName);
-    }
     return 0;
 }
 
@@ -524,27 +536,26 @@ int bend_esrequest (void *handle, bend_esrequest_rr *rr)
            if (notToKeep)
            {
                int i;
-                zebra_begin_trans (zh);
+                zebra_begin_trans (zh, 1);
                for (i = 0; i < notToKeep->num; i++)
                {
                    Z_External *rec = notToKeep->elements[i]->record;
                     struct oident *oident = 0;
-                    Odr_oct *recid = notToKeep->elements[i]->u.opaque;
+                    Odr_oct *opaque_recid = 0;
+                   SYSNO sysno = 0;
 
-                    if (!recid)
-                    {
-                        rr->errcode = 224;
-                        rr->errstring = "record Id not supplied";
-                        break;
-                    }
-                    if (notToKeep->elements[i]->which !=
-                        Z_IUSuppliedRecords_elem_opaque)
-                    {
-                        rr->errcode = 224;
-                        rr->errstring = "only opaque record ID supported";
-                        break;
+                   if (notToKeep->elements[i]->u.opaque)
+                   {
+                       switch(notToKeep->elements[i]->which)
+                       {
+                       case Z_IUSuppliedRecords_elem_opaque:
+                           opaque_recid = notToKeep->elements[i]->u.opaque;
+                           break; /* OK, recid already set */
+                       case Z_IUSuppliedRecords_elem_number:
+                           sysno = *notToKeep->elements[i]->u.number;
+                           break;
+                       }
                     }
-                        
                    if (rec->direct_reference)
                    {
                        oident = oid_getentbyoid(rec->direct_reference);
@@ -595,7 +606,7 @@ int bend_esrequest (void *handle, bend_esrequest_rr *rr)
                             action = 3;
                         if (*toKeep->action ==
                             Z_IUOriginPartToKeep_specialUpdate)
-                            action = 1;
+                            action = 4;
 
                         if (!action)
                         {
@@ -603,31 +614,76 @@ int bend_esrequest (void *handle, bend_esrequest_rr *rr)
                             rr->errstring = "unsupported ES Update action";
                             break;
                         }
-                        else
-                        {
+                        else if (opaque_recid)
+                       {
                             int r = zebra_admin_exchange_record (
-                                zh, toKeep->databaseName,
+                                zh,
                                 rec->u.octet_aligned->buf,
                                 rec->u.octet_aligned->len,
-                                recid->buf, recid->len,
+                                opaque_recid->buf, opaque_recid->len,
                                 action);
-                            if (r && *toKeep->action ==
-                                Z_IUOriginPartToKeep_specialUpdate)
-                            {
-                                r = zebra_admin_exchange_record (
-                                    zh, toKeep->databaseName,
-                                    rec->u.octet_aligned->buf,
-                                    rec->u.octet_aligned->len,
-                                    recid->buf, recid->len,
-                                    2);
-                            }
                             if (r)
                             {
                                 rr->errcode = 224;
                                 rr->errstring = "record exchange failed";
                                 break;
                             }
-                        }
+                       }
+                       else
+                       {
+                           int r = -1;
+                           switch(action) {
+                           case 1:
+                               r = zebra_insert_record(
+                                   zh,
+                                   0, /* recordType */
+                                   &sysno,
+                                   0, /* match */
+                                   0, /* fname */
+                                   rec->u.octet_aligned->buf,
+                                   rec->u.octet_aligned->len,
+                                   0);
+                               if (r)
+                               {
+                                   rr->errcode = 224;
+                                   rr->errstring = "insert_record failed";
+                               }
+                               break;
+                           case 2:
+                           case 4:
+                               r = zebra_update_record(
+                                   zh,
+                                   0, /* recordType */
+                                   &sysno,
+                                   0, /* match */
+                                   0, /* fname */
+                                   rec->u.octet_aligned->buf,
+                                   rec->u.octet_aligned->len,
+                                   1);
+                               if (r)
+                               {
+                                   rr->errcode = 224;
+                                   rr->errstring = "update_record failed";
+                               }
+                               break;
+                           case 3:
+                               r = zebra_delete_record(
+                                   zh,
+                                   0, /* recordType */
+                                   &sysno,
+                                   0, /* match */
+                                   0, /* fname */
+                                   rec->u.octet_aligned->buf,
+                                   rec->u.octet_aligned->len,
+                                   0);
+                               if (r)
+                               {
+                                   rr->errcode = 224;
+                                   rr->errstring = "delete_record failed";
+                               }
+                               break;
+                           }                           
+                       }
                     }
                }
                 zebra_end_trans (zh);
@@ -646,33 +702,57 @@ int bend_esrequest (void *handle, bend_esrequest_rr *rr)
 
 static void bend_start (struct statserv_options_block *sob)
 {
+    if (sob->handle)
+       zebra_stop((ZebraService) sob->handle);
+    sob->handle = zebra_start(sob->configname);
+    if (!sob->handle)
+    {
+       yaz_log (LOG_FATAL, "Failed to read config `%s'", sob->configname);
+       exit (1);
+    }
 #ifdef WIN32
     
 #else
     if (!sob->inetd) 
     {
-        char *pidfile = "zebrasrv.pid";
-        int fd = creat (pidfile, 0666);
+       char pidfname[4096];
+        struct flock area;
+       int fd;
+
+       zebra_pidfname(sob->handle, pidfname);
+
+        fd = open (pidfname, O_EXCL|O_WRONLY|O_CREAT, 0666);
         if (fd == -1)
-            logf (LOG_WARN|LOG_ERRNO, "creat %s", pidfile);
+        {
+            if (errno != EEXIST)
+            {
+                yaz_log(LOG_FATAL|LOG_ERRNO, "lock file %s", pidfname);
+                exit(1);
+            }
+            fd = open(pidfname, O_RDWR, 0666);
+            if (fd == -1)
+            {
+                yaz_log(LOG_FATAL|LOG_ERRNO, "lock file %s", pidfname);
+                exit(1);
+            }
+        }
+        area.l_type = F_WRLCK;
+        area.l_whence = SEEK_SET;
+        area.l_len = area.l_start = 0L;
+        if (fcntl (fd, F_SETLK, &area) == -1)
+        {
+            yaz_log(LOG_ERRNO|LOG_FATAL, "Zebra server already running");
+            exit(1);
+        }
         else
         {
            char pidstr[30];
        
            sprintf (pidstr, "%ld", (long) getpid ());
            write (fd, pidstr, strlen(pidstr));
-           close (fd);
         }
     }
 #endif
-    if (sob->handle)
-       zebra_stop((ZebraService) sob->handle);
-    sob->handle = zebra_start(sob->configname);
-    if (!sob->handle)
-    {
-       yaz_log (LOG_FATAL, "Failed to read config `%s'", sob->configname);
-       exit (1);
-    }
 }
 
 static void bend_stop(struct statserv_options_block *sob)
@@ -680,14 +760,12 @@ static void bend_stop(struct statserv_options_block *sob)
 #ifdef WIN32
 
 #else
-    /*
-     * Fixme! Why commets?, because the first chaild process to invoke bend_stop().
-     *
-     
-    if (!sob->inetd) 
-        unlink ("zebrasrv.pid");
-    */
-    
+    if (!sob->inetd && sob->handle) 
+    {
+       char pidfname[4096];
+       zebra_pidfname(sob->handle, pidfname);
+        unlink (pidfname);
+    }
 #endif
     if (sob->handle)
     {