Fixed HTTP-client query-string handling bug.
[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 #ifdef HAVE_XATTR
34 #include <attr/xattr.h>
35 #endif
36
37 struct trdata {
38     size_t startpos;
39 };
40
41 static char *fn = NULL;
42
43 void filelog(char *format, ...)
44 {
45     FILE *out;
46     va_list args;
47     char *b, *t, *p;
48     time_t now;
49     
50     if(fn == NULL)
51         return;
52     if((out = fopen(fn, "a")) == NULL) {
53         flog(LOG_WARNING, "could not open reqstat log file `%s': %s", fn, strerror(errno));
54         return;
55     }
56     now = time(NULL);
57     va_start(args, format);
58     b = vsprintf2(format, args);
59     va_end(args);
60     t = ctime(&now);
61     if((p = strchr(t, '\n')) != NULL)
62         *p = 0;
63     fprintf(out, "%s: %s\n", t, b);
64     free(b);
65     fclose(out);
66 }
67
68 #ifdef HAVE_XATTR
69 void xainc(wchar_t *file, char *an, off_t inc)
70 {
71     char buf[32];
72     ssize_t al;
73     off_t val;
74     char *fn;
75     
76     if(file[0] != L'/')
77         return;
78     if((fn = icswcstombs(file, NULL, NULL)) == NULL) {
79         flog(LOG_WARNING, "could not convert filename %ls into local charset: %s", file, strerror(errno));
80         return;
81     }
82     if((al = getxattr(fn, an, buf, sizeof(buf) - 1)) < 0) {
83         if(errno != ENOATTR) {
84             flog(LOG_WARNING, "could not get xattr %s on %s: %s", an, fn, strerror(errno));
85             return;
86         }
87         val = 0;
88     } else {
89         buf[al] = 0;
90         val = strtoll(buf, NULL, 10);
91     }
92     val += inc;
93     al = snprintf(buf, sizeof(buf), "%ji", (intmax_t)val);
94     if(setxattr(fn, an, buf, al, 0) < 0)
95         flog(LOG_WARNING, "could not set xattr %s on %s: %s", an, fn, strerror(errno));
96 }
97 #endif
98
99 void request(struct transfer *transfer, struct trdata *data)
100 {
101     filelog("request %ls", transfer->path);
102 #ifdef HAVE_XATTR
103     if(confgetint("reqstat", "xa"))
104         xainc(transfer->path, "user.dc-req", 1);
105 #endif
106 }
107
108 void start(struct transfer *transfer, struct trdata *data)
109 {
110     filelog("start %ls at %zi", transfer->path, data->startpos);
111 #ifdef HAVE_XATTR
112     if(confgetint("reqstat", "xa"))
113         xainc(transfer->path, "user.dc-started", 1);
114 #endif
115 }
116
117 void finish(struct transfer *transfer, struct trdata *data)
118 {
119     filelog("finish %ls at %zi, total %zi", transfer->path, transfer->curpos, transfer->curpos - data->startpos);
120 #ifdef HAVE_XATTR
121     if(confgetint("reqstat", "xa"))
122         xainc(transfer->path, "user.dc-bytes", transfer->curpos - data->startpos);
123 #endif
124 }
125
126 static int chattr(struct transfer *transfer, wchar_t *attrib, struct trdata *data)
127 {
128     if(!wcscmp(attrib, L"state")) {
129         if(transfer->state == TRNS_MAIN)
130             data->startpos = transfer->curpos;
131         start(transfer, data);
132     } else if(!wcscmp(attrib, L"path")) {
133         if(transfer->path[0] != L'/') {
134             CBUNREG(transfer, trans_ac, data);
135             CBUNREG(transfer, trans_destroy, data);
136             free(data);
137         }
138         request(transfer, data);
139     }
140     return(0);
141 }
142
143 static int destroy(struct transfer *transfer, struct trdata *data)
144 {
145     if(transfer->curpos > data->startpos)
146         finish(transfer, data);
147     free(data);
148     return(0);
149 }
150
151 static int reg(struct transfer *transfer, void *uudata)
152 {
153     struct trdata *data;
154     
155     if(transfer->dir != TRNSD_UP)
156         return(0);
157     data = memset(smalloc(sizeof(*data)), 0, sizeof(*data));
158     CBREG(transfer, trans_ac, (int (*)(struct transfer *, wchar_t *, void *))chattr, NULL, data);
159     CBREG(transfer, trans_destroy, (int (*)(struct transfer *, void *))destroy, NULL, data);
160     return(0);
161 }
162
163 static int chfile(struct configvar *var, void *uudata)
164 {
165     if(fn != NULL)
166         free(fn);
167     if(var->val.str[0] == L'\0') {
168         fn = NULL;
169     } else {
170         if((fn = icwcstombs(var->val.str, NULL)) == NULL)
171             flog(LOG_WARNING, "could not convert reqstat filename `%ls' into local charset: %s", var->val.str, strerror(errno));
172     }
173     return(0);
174 }
175
176 static int init(int hup)
177 {
178     if(!hup) {
179         GCBREG(newtransfercb, reg, NULL);
180         chfile(confgetvar("reqstat", "file"), NULL);
181         CBREG(confgetvar("reqstat", "file"), conf_update, chfile, NULL, NULL);
182     }
183     return(0);
184 }
185
186 static struct configvar myvars[] = {
187     /** The name of a file to log upload request information to. If
188      * unspecified, upload requests will not be logged. */
189     {CONF_VAR_STRING, "file", {.str = L""}},
190     /** If set to true, upload statistics of files will be accumulated
191      * in counters stored in extends attributes (see attr(5)) on the
192      * files themselves. The data accumulated is the number of
193      * requests, the number of successfully started uploads and the
194      * number of uploaded bytes. */
195     {CONF_VAR_BOOL, "xa", {.num = 0}},
196     {CONF_VAR_END}
197 };
198
199 static struct module me = {
200     .conf = {
201         .vars = myvars
202     },
203     .init = init,
204     .name = "reqstat"
205 };
206
207 MODULE(me)