getstats.c 4.4 KB
Newer Older
1
/* Synchronet C-exported statistics retrieval functions */
2
3
4
5
6

/****************************************************************************
 * @format.tab-size 4		(Plain Text/Source Code File Header)			*
 * @format.use-tabs true	(see http://www.synchro.net/ptsc_hdr.html)		*
 *																			*
7
 * Copyright Rob Swindell - http://www.synchro.net/copyright.html			*
8
9
10
11
12
13
14
15
16
17
18
19
20
21
 *																			*
 * This program is free software; you can redistribute it and/or			*
 * modify it under the terms of the GNU General Public License				*
 * as published by the Free Software Foundation; either version 2			*
 * of the License, or (at your option) any later version.					*
 * See the GNU General Public License for more details: gpl.txt or			*
 * http://www.fsf.org/copyleft/gpl.html										*
 *																			*
 * For Synchronet coding style and modification guidelines, see				*
 * http://www.synchro.net/source.html										*
 *																			*
 * Note: If this box doesn't appear square, then you need to fix your tabs.	*
 ****************************************************************************/

22
23
24
#include "getstats.h"
#include "nopen.h"
#include "smblib.h"
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

/****************************************************************************/
/* Reads data from dsts.dab into stats structure                            */
/* If node is zero, reads from ctrl\dsts.dab, otherwise from each node		*/
/****************************************************************************/
BOOL DLLCALL getstats(scfg_t* cfg, char node, stats_t* stats)
{
    char str[MAX_PATH+1];
    int file;

    sprintf(str,"%sdsts.dab",node ? cfg->node_path[node-1] : cfg->ctrl_dir);
    if((file=nopen(str,O_RDONLY))==-1) {
        return(FALSE); 
	}
    lseek(file,4L,SEEK_SET);    /* Skip update time/date */
40
	/* TODO: Direct read of unpacked struct */
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
    read(file,stats,sizeof(stats_t));
    close(file);
	return(TRUE);
}

/****************************************************************************/
/* Returns the number of files in the directory 'dirnum'                    */
/****************************************************************************/
long DLLCALL getfiles(scfg_t* cfg, uint dirnum)
{
	char str[256];
	long l;

	if(dirnum>=cfg->total_dirs)	/* out of range */
		return(0);
	sprintf(str,"%s%s.ixb",cfg->dir[dirnum]->data_dir, cfg->dir[dirnum]->code);
57
	l=(long)flength(str);
58
59
60
61
62
63
64
65
66
67
	if(l>0L)
		return(l/F_IXBSIZE);
	return(0);
}

/****************************************************************************/
/* Returns total number of posts in a sub-board 							*/
/****************************************************************************/
ulong DLLCALL getposts(scfg_t* cfg, uint subnum)
{
68
69
70
	if(cfg->sub[subnum]->misc & SUB_NOVOTING) {
		char path[MAX_PATH + 1];
		off_t l;
71

72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
		SAFEPRINTF2(path, "%s%s.sid", cfg->sub[subnum]->data_dir, cfg->sub[subnum]->code);
		l = flength(path);
		if(l < sizeof(idxrec_t))
			return 0;
		return l / sizeof(idxrec_t);
	}
	smb_t smb = {{0}};
	SAFEPRINTF2(smb.file, "%s%s", cfg->sub[subnum]->data_dir, cfg->sub[subnum]->code);
	smb.retry_time = cfg->smb_retry_time;
	smb.subnum = subnum;
	if(smb_open_index(&smb) != SMB_SUCCESS)
		return 0;
	size_t result = smb_msg_count(&smb, (1 << SMB_MSG_TYPE_NORMAL) | (1 << SMB_MSG_TYPE_POLL));
	smb_close(&smb);
	return result;
87
}
88
89
90
91
92
93
94

BOOL inc_sys_upload_stats(scfg_t* cfg, ulong files, ulong bytes)
{
	char	str[MAX_PATH+1];
	int		file;
	uint32_t	val;

Rob Swindell's avatar
Rob Swindell committed
95
	SAFEPRINTF(str,"%sdsts.dab",cfg->ctrl_dir);
96
97
98
	if((file=nopen(str,O_RDWR))==-1) 
		return(FALSE);

Rob Swindell's avatar
Rob Swindell committed
99
100
	(void)lseek(file,20L,SEEK_SET);   /* Skip timestamp, logons and logons today */
	(void)read(file,&val,4);        /* Uploads today         */
101
	val+=files;
Rob Swindell's avatar
Rob Swindell committed
102
103
104
	(void)lseek(file,-4L,SEEK_CUR);
	(void)write(file,&val,4);
	(void)read(file,&val,4);        /* Upload bytes today    */
105
	val+=bytes;
Rob Swindell's avatar
Rob Swindell committed
106
107
	(void)lseek(file,-4L,SEEK_CUR);
	(void)write(file,&val,4);
108
109
110
111
112
113
114
115
116
117
	close(file);
	return(TRUE);
}

BOOL inc_sys_download_stats(scfg_t* cfg, ulong files, ulong bytes)
{
	char	str[MAX_PATH+1];
	int		file;
	uint32_t	val;

Rob Swindell's avatar
Rob Swindell committed
118
	SAFEPRINTF(str,"%sdsts.dab",cfg->ctrl_dir);
119
120
121
	if((file=nopen(str,O_RDWR))==-1) 
		return(FALSE);

Rob Swindell's avatar
Rob Swindell committed
122
123
	(void)lseek(file,28L,SEEK_SET);   /* Skip timestamp, logons and logons today */
	(void)read(file,&val,4);        /* Downloads today         */
124
	val+=files;
Rob Swindell's avatar
Rob Swindell committed
125
126
127
	(void)lseek(file,-4L,SEEK_CUR);
	(void)write(file,&val,4);
	(void)read(file,&val,4);        /* Download bytes today    */
128
	val+=bytes;
Rob Swindell's avatar
Rob Swindell committed
129
130
	(void)lseek(file,-4L,SEEK_CUR);
	(void)write(file,&val,4);
131
132
133
	close(file);
	return(TRUE);
}