1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
/*
* $Id: rfioacct.c 3172 2010-02-15 08:51:54Z baud $
*/
/*
* Copyright (C) 1998-2010 by CERN/IT/PDP/DM Olof Barring
* All rights reserved
*/
#ifndef lint
static char sccsid[] = "@(#)$RCSfile: rfioacct.c,v $ $Revision: 3172 $ $Date: 2010-02-15 09:51:54 +0100 (Mon, 15 Feb 2010) $ CERN/IT/PDP/DM Olof Barring";
#endif /* not lint */
#define RFIO_KERNEL 1
#include <stdio.h>
#include <sys/types.h>
#if defined(_WIN32)
#include <pwd.h>
#include <winsock2.h>
#include <ws2tcpip.h>
#else
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#endif
#include "rfio.h"
#include "sacct.h"
void rfioacct(reqtype,uid,gid,ns,flag1,flag2,status,rc,infop,filename1,filename2)
int reqtype;
uid_t uid;
gid_t gid;
int ns;
int flag1;
int flag2;
int status;
int rc;
struct rfiostat *infop;
char *filename1,*filename2;
{
#if defined(SACCT)
int acctreclen;
struct acctrfio64ipv46 acct_rfio;
struct sockaddr_storage local_addr;
struct sockaddr_storage remote_addr;
#if defined(_AIX)
socklen_t addr_len;
#else
int addr_len;
#endif
static int ACCTRFIO_ON = -1;
char *getconfent();
char *p = NULL;
static int jid = -1;
if ( ACCTRFIO_ON == -1 ) {
if (p == NULL && (p = getconfent("ACCT", "RFIO", 0)) == NULL ||
(strcmp (p, "YES") && strcmp (p, "yes"))) ACCTRFIO_ON = 0;
else ACCTRFIO_ON = 1;
}
if (jid == -1) jid = getpid();
if ( ACCTRFIO_ON == 0 ) return;
if ( ns < 0 ) return;
memset(&acct_rfio,'\0',sizeof(acct_rfio));
acct_rfio.reqtype = (int) reqtype;
acct_rfio.uid = (int) uid;
acct_rfio.gid = (int) gid;
acct_rfio.jid = (int) jid;
acct_rfio.accept_socket = ns;
acct_rfio.flags.anonymous.flag1 = (int)flag1;
acct_rfio.flags.anonymous.flag2 = (int)flag2;
if ( infop != NULL ) {
acct_rfio.nb_read = (int) infop->readop;
acct_rfio.nb_write = (int) infop->writop;
acct_rfio.nb_ahead = (int) infop->aheadop;
acct_rfio.nb_stat = (int) infop->statop;
acct_rfio.nb_seek = (int) infop->seekop;
acct_rfio.nb_preseek = (int) infop->presop;
acct_rfio.read_size = infop->rnbr;
acct_rfio.write_size = infop->wnbr;
}
if (ns >= 0 ) {
addr_len = sizeof(local_addr);
getsockname(ns,(struct sockaddr *)&local_addr,&addr_len);
acct_rfio.af = local_addr.ss_family;
switch(acct_rfio.af) {
case AF_INET:
acct_rfio.addr.ipv4.local_addr = (int)((struct sockaddr_in *)&local_addr)->sin_addr.s_addr;
break;
case AF_INET6:
memcpy(&acct_rfio.addr.ipv6.local_addr,
&((struct sockaddr_in6 *)&local_addr)->sin6_addr.s6_addr, sizeof(acct_rfio.addr.ipv6.local_addr));
break;
}
addr_len = sizeof(remote_addr);
getpeername(ns,(struct sockaddr *)&remote_addr,&addr_len);
switch(acct_rfio.af) {
case AF_INET:
acct_rfio.addr.ipv4.remote_addr = (int)((struct sockaddr_in *)&remote_addr)->sin_addr.s_addr;
break;
case AF_INET6:
memcpy(&acct_rfio.addr.ipv6.remote_addr,
&((struct sockaddr_in6 *)&remote_addr)->sin6_addr.s6_addr, sizeof(acct_rfio.addr.ipv6.remote_addr));
break;
}
}
if (filename1 != NULL ) {
acct_rfio.len1 = min(2*MAXPATH,strlen(filename1));
strncpy(&acct_rfio.filename[0],filename1,acct_rfio.len1);
}
if (filename2 != NULL ) {
acct_rfio.len2 = min(2*MAXPATH-acct_rfio.len1,strlen(filename2));
strncpy(&acct_rfio.filename[acct_rfio.len1],filename2,acct_rfio.len2);
}
acct_rfio.status = status;
acct_rfio.rc = rc;
acctreclen = ((char *)acct_rfio.filename - (char *) &acct_rfio) + strlen(acct_rfio.filename) + 1;
wsacct(ACCTRFIO64IPV46, &acct_rfio, acctreclen);
#endif /* SACCT */
return;
}
|
