Python module revbump.
[doldaconnect.git] / daemon / reqstat.c
1 /*
2  *  Dolda Connect - Modular multiuser Direct Connect-style client
3  *  Copyright (C) 2007 Fredrik Tolf <fredrik@dolda2000.com>
4  *  
5  *  This program is free software; you can redistribute it and/or modify
6  *  it under the terms of the GNU General Public License as published by
7  *  the Free Software Foundation; either version 2 of the License, or
8  *  (at your option) any later version.
9  *  
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU General Public License for more details.
14  *  
15  *  You should have received a copy of the GNU General Public License
16  *  along with this program; if not, write to the Free Software
17  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
18 */
19
20 #include <stdlib.h>
21 #include <stdio.h>
22 #include <unistd.h>
23 #include <errno.h>
24 #include <time.h>
25
26 #ifdef HAVE_CONFIG_H
27 #include <config.h>
28 #endif
29 #include "transfer.h"
30 #include "module.h"
31 #include "log.h"
32
33 struct trdata {
34     size_t startpos;
35 };
36
37 static char *fn = NULL;
38
39 void filelog(char *format, ...)
40 {
41     FILE *out;
42     va_list args;
43     char *b, *t, *p;
44     time_t now;
45     
46     if(fn == NULL)
47         return;
48     if((out = fopen(fn, "a")) == NULL) {
49         flog(LOG_WARNING, "could not open reqstat log file `%s': %s", fn, strerror(errno));
50         return;
51     }
52     now = time(NULL);
53     va_start(args, format);
54     b = vsprintf2(format, args);
55     va_end(args);
56     t = ctime(&now);
57     if((p = strchr(t, '\n')) != NULL)
58         *p = 0;
59     fprintf(out, "%s: %s\n", t, b);
60     free(b);
61     fclose(out);
62 }
63
64 void request(struct transfer *transfer, struct trdata *data)
65 {
66     filelog("request %ls", transfer->path);
67 }
68
69 void start(struct transfer *transfer, struct trdata *data)
70 {
71     filelog("start %ls at %zi", transfer->path, data->startpos);
72 }
73
74 void finish(struct transfer *transfer, struct trdata *data)
75 {
76     filelog("finish %ls at %zi, total %zi", transfer->path, transfer->curpos, transfer->curpos - data->startpos);
77 }
78
79 static int chattr(struct transfer *transfer, wchar_t *attrib, struct trdata *data)
80 {
81     if(!wcscmp(attrib, L"state")) {
82         if(transfer->state == TRNS_MAIN)
83             data->startpos = transfer->curpos;
84         start(transfer, data);
85     } else if(!wcscmp(attrib, L"path")) {
86         if(transfer->path[0] != L'/') {
87             CBUNREG(transfer, trans_ac, data);
88             CBUNREG(transfer, trans_destroy, data);
89             free(data);
90         }
91         request(transfer, data);
92     }
93     return(0);
94 }
95
96 static int destroy(struct transfer *transfer, struct trdata *data)
97 {
98     if(transfer->curpos > data->startpos)
99         finish(transfer, data);
100     free(data);
101     return(0);
102 }
103
104 static int reg(struct transfer *transfer, void *uudata)
105 {
106     struct trdata *data;
107     
108     if(transfer->dir != TRNSD_UP)
109         return(0);
110     data = memset(smalloc(sizeof(*data)), 0, sizeof(*data));
111     CBREG(transfer, trans_ac, (int (*)(struct transfer *, wchar_t *, void *))chattr, NULL, data);
112     CBREG(transfer, trans_destroy, (int (*)(struct transfer *, void *))destroy, NULL, data);
113     return(0);
114 }
115
116 static int chfile(struct configvar *var, void *uudata)
117 {
118     if(fn != NULL)
119         free(fn);
120     if(var->val.str[0] == L'\0') {
121         fn = NULL;
122     } else {
123         if((fn = icwcstombs(var->val.str, NULL)) == NULL)
124             flog(LOG_WARNING, "could not convert reqstat filename `%ls' into local charset: %s", var->val.str, strerror(errno));
125     }
126     return(0);
127 }
128
129 static int init(int hup)
130 {
131     if(!hup) {
132         GCBREG(newtransfercb, reg, NULL);
133         chfile(confgetvar("reqstat", "file"), NULL);
134         CBREG(confgetvar("reqstat", "file"), conf_update, chfile, NULL, NULL);
135     }
136     return(0);
137 }
138
139 static struct configvar myvars[] = {
140     /** The name of a file to log upload request information to. If
141      * unspecified, upload requests will not be logged. */
142     {CONF_VAR_STRING, "file", {.str = L""}},
143     {CONF_VAR_END}
144 };
145
146 static struct module me = {
147     .conf = {
148         .vars = myvars
149     },
150     .init = init,
151     .name = "reqstat"
152 };
153
154 MODULE(me)