add sqmail-4.2.29asqmail-4.2

1 files changed, 122 insertions, 0 deletions

diff --git a/src/control.c b/src/control.c
new file mode 100644
index 0000000..2558225
--- /dev/null
+++ b/src/control.c
@@ -0,0 +1,122 @@
+#include <unistd.h>
+#include "open.h"
+#include "getln.h"
+#include "stralloc.h"
+#include "buffer.h"
+#include "logmsg.h"
+#include "control.h"
+#include "alloc.h"
+#include "scan.h"
+#include "error.h"
+
+static char inbuf[2048];
+static stralloc line = {0};
+static stralloc me = {0};
+static int meok = 0;
+
+/** @file  control.c
+*/
+
+static void striptrailingwhitespace(stralloc *sa)
+{
+  while (sa->len > 0)
+    switch (sa->s[sa->len - 1]) {
+      case '\n': case ' ': case '\t':
+        --sa->len;
+        break;
+      default:
+        return;
+    }
+}
+
+int control_init(void)
+{
+  int r;
+
+  r = control_readline(&me,"control/me");
+  if (r == 1) meok = 1;
+  return r;
+}
+
+int control_rldef(stralloc *sa,char *fn,int flagme,char *def)
+{
+  int r;
+
+  r = control_readline(sa,fn);
+  if (r) return r;
+  if (flagme) if (meok) return stralloc_copy(sa,&me) ? 1 : -1;
+  if (def) return stralloc_copys(sa,def) ? 1 : -1;
+  return r;
+}
+
+int control_readline(stralloc *sa,char *fn)
+{
+  buffer b;
+  int fd;
+  int match;
+
+  fd = open_read(fn);
+  if (fd == -1) { if (errno == ENOENT) return 0; return -1; }
+ 
+  buffer_init(&b,read,fd,inbuf,sizeof(inbuf));
+
+  if (getln(&b,sa,&match,'\n') == -1) { close(fd); return -1; }
+
+  striptrailingwhitespace(sa);
+
+  close(fd);
+  return 1;
+}
+
+int control_readint(int *i,char *fn)
+{
+  unsigned long u;
+
+  switch (control_readline(&line,fn)) {
+    case 0: return 0;
+    case -1: return -1;
+  }
+  if (!stralloc_0(&line)) return -1;
+  if (!scan_ulong(line.s,&u)) return 0;
+  *i = u;
+
+  return 1;
+}
+
+int control_readfile(stralloc *sa,char *fn,int flagme)
+{
+  buffer b;
+  int fd;
+  int match;
+
+  if (!stralloc_copys(sa,"")) return -1;
+
+  fd = open_read(fn);
+  if (fd == -1) {
+    if (errno == ENOENT) {
+      if (flagme && meok) {
+        if (!stralloc_copy(sa,&me)) return -1;
+        if (!stralloc_0(sa)) return -1;
+        return 1;
+      }
+      return 0;
+    }
+    return -1;
+  }
+
+  buffer_init(&b,read,fd,inbuf,sizeof(inbuf));
+
+  for (;;) {
+    if (getln(&b,&line,&match,'\n') == -1) break;
+    if (!match && !line.len) { close(fd); return 1; }
+    striptrailingwhitespace(&line);
+    if (!stralloc_0(&line)) break;
+    if (line.s[0])
+      if (line.s[0] != '#')
+        if (!stralloc_cat(sa,&line)) break;
+    if (!match) { close(fd); return 1; }
+  }
+
+  close(fd);
+  return -1;
+}