Changeset 1554 for zzuf/trunk/src/libzzuf.c

Timestamp:

01/04/07 02:09:04 (6 years ago)

Author:

sam

Message:

• Implement -P / --protect.

File:

• zzuf/trunk/src/libzzuf.c (modified) (5 diffs)

zzuf/trunk/src/libzzuf.c

@@ -28 +28 @@
 #include <unistd.h>
 #include <stdlib.h>
+#include <string.h>
 #include <fcntl.h>
 #include <regex.h>
@@ -46 +47 @@
 int   _zz_signal   = 0;
 
+/* Global tables */
+int   _zz_protect[256];
+
 /* Local variables */
 static regex_t * re_include = NULL;
@@ -51 +55 @@
 
 /* Local prototypes */
+static void _zz_protect_init(char const *);
 static void _zz_fd_init(void);
 static void _zz_fd_fini(void);
@@ -75 +80 @@
         _zz_ratio = 5.0f;
 
+    tmp = getenv("ZZUF_PROTECT");
+    if(tmp && *tmp)
+        _zz_protect_init(tmp);
+
     tmp = getenv("ZZUF_INCLUDE");
     if(tmp && *tmp)
@@ -112 +121 @@
 {
     _zz_fd_fini();
+}
+
+/* Byte list stuff */
+static void _zz_protect_init(char const *list)
+{
+    static char const hex[] = "0123456789abcdef0123456789ABCDEF";
+    char const *tmp;
+    int a, b;
+
+    memset(_zz_protect, 0, 256 * sizeof(int));
+
+    for(tmp = list, a = b = -1; *tmp; tmp++)
+    {
+        int new;
+
+        if(*tmp == '\\' && tmp[1] == '\0')
+            new = '\\';
+        else if(*tmp == '\\')
+        {
+            tmp++;
+            if(*tmp == 'n')
+                new = '\n';
+            else if(*tmp == 'r')
+                new = '\r';
+            else if(*tmp == 't')
+                new = '\t';
+            else if(*tmp == '0')
+                new = '\0';
+            else if((*tmp == 'x' || *tmp == 'X')
+                     && tmp[1] && strchr(hex, tmp[1])
+                     && tmp[2] && strchr(hex, tmp[2]))
+            {
+                new = ((strchr(hex, tmp[1]) - hex) & 0xf) << 4;
+                new |= (strchr(hex, tmp[2]) - hex) & 0xf;
+                tmp += 2;
+            }
+            else
+                new = (unsigned char)*tmp; /* XXX: OK for \\, but what else? */
+        }
+        else
+            new = (unsigned char)*tmp;
+
+        if(a != -1 && b == '-' && a <= new)
+        {
+            while(a <= new)
+                _zz_protect[a++] = 1;
+            a = b = -1;
+        }
+        else
+        {
+            if(a != -1)
+                _zz_protect[a] = 1;
+            a = b;
+            b = new;
+        }
+    }
+
+    if(a != -1)
+        _zz_protect[a] = 1;
+    if(b != -1)
+        _zz_protect[b] = 1;
 }