int
main(int argc, char **argv)
{
char buff[65536];
struct sockaddr_in saddr, raddr;
socklen_t rsize = sizeof (raddr);
int currentbytes = 0;
int sock;
int protocol;
int result;
security_id_t sid = -1;
if (argc < 2) {
printf("%s <protocol>\n", argv[0]);
return -1;
}
if (strcmp(argv[1], "stream") == 0)
protocol = SOCK_STREAM;
else
protocol = SOCK_DGRAM;
if (argc == 3)
sid = atoi(argv[2]);
bzero(&saddr, sizeof (struct sockaddr_in));
if (!inet_aton(hostname, &saddr.sin_addr)) {
printf("error converting address\n");
return -1;
}
saddr.sin_family = AF_INET;
saddr.sin_port = htons(port);
if ((sock = socket(saddr.sin_family, protocol, 0)) < 0) {
perror("socket");
return -1;
}
if ((result = setsockopt(sock, SOL_SOCKET, SO_REUSEADDR,
&on, sizeof (on)))) {
perror("setsockopt: ");
close(sock);
return -1;
}
if (bind(sock, (struct sockaddr *) &saddr, sizeof (saddr)) < 0) {
perror("bind");
close(sock);
return -1;
}
if (sid != -1) {
struct stat statbuf;
security_id_t outsid;
(void)fstat_secure(sock, &statbuf, &outsid);
printf("Changing socket sid from %d to %d\n", outsid, sid);
(void)fchsid(sock, sid);
}
if (protocol == SOCK_STREAM) {
int remote;
if (listen(sock, SOMAXCONN)) {
close(sock);
return -1;
}
while (readbytes > currentbytes) {
remote =
accept(sock, (struct sockaddr *) &raddr, &rsize);
if (remote < 0) {
perror("accept: ");
close(sock);
return -1;
}
for (;;) {
result = read(remote, buff, 65536);
if (result == 0) // EOF
break;
if (result < 0) {
perror("stream read failed: ");
close(sock);
return -1;
}
currentbytes += result;
printf("recv: %5d bytes, %5d total\n", result,
currentbytes);
if (currentbytes >= readbytes)
break;
}
}
} else {
for (;;) {
result = recvfrom(sock, buff, 65536, 0,
(struct sockaddr *) &raddr, &rsize);
currentbytes += result;
printf("recv: %5d bytes, %5d total\n", result,
currentbytes);
if (currentbytes >= readbytes)
break;
}
}
shutdown(sock, 2);
close(sock);
return (0);
}
int
main(int argc, char **argv)
{
struct sockaddr_in saddr, raddr;
socklen_t rsize = sizeof (raddr);
int sock;
int result;
int remote;
security_id_t sid = -1;
struct stat statbuf;
security_id_t outsid;
if (argc != 2) {
printf("%s <sid>\n", argv[0]);
return -1;
}
sid = atoi(argv[1]);
bzero(&saddr, sizeof (struct sockaddr_in));
if (!inet_aton(hostname, &saddr.sin_addr)) {
printf("error converting address\n");
return -1;
}
saddr.sin_family = AF_INET;
saddr.sin_port = htons(port);
if ((sock = socket(saddr.sin_family, SOCK_STREAM, 0)) < 0) {
perror("socket");
return -1;
}
if ((result = setsockopt(sock, SOL_SOCKET, SO_REUSEADDR,
&on, sizeof (on)))) {
perror("setsockopt: ");
close(sock);
return -1;
}
if (bind(sock, (struct sockaddr *) &saddr, sizeof (saddr)) < 0) {
perror("bind");
close(sock);
return -1;
}
if (listen(sock, SOMAXCONN)) {
close(sock);
return -1;
}
if ((result = fcntl(sock, F_SETFL, O_NONBLOCK))) {
perror("fcntl: ");
return -1;
}
(void)fstat_secure(sock, &statbuf, &outsid);
printf("Changing socket sid from %d to %d\n", outsid, sid);
(void)fchsid(sock, sid);
remote = accept(sock, (struct sockaddr *) &raddr, &rsize);
printf("Changing socket sid back to %d\n", outsid);
(void)fchsid(sock, outsid);
printf("remote == %d\n", remote);
shutdown(sock, 2);
close(sock);
if (remote == -1)
return 1;
return (0);
}
extern procps_status_t * procps_scan(int save_user_arg0
#ifdef CONFIG_SELINUX
, int use_selinux , security_id_t *sid
#endif
)
{
static DIR *dir;
struct dirent *entry;
static procps_status_t ret_status;
char *name;
int n;
char status[32];
char buf[1024];
FILE *fp;
procps_status_t curstatus;
int pid;
long tasknice;
struct stat sb;
if (!dir) {
dir = opendir("/proc");
if(!dir)
bb_error_msg_and_die("Can't open /proc");
}
for(;;) {
if((entry = readdir(dir)) == NULL) {
closedir(dir);
dir = 0;
return 0;
}
name = entry->d_name;
if (!(*name >= '0' && *name <= '9'))
continue;
memset(&curstatus, 0, sizeof(procps_status_t));
pid = atoi(name);
curstatus.pid = pid;
sprintf(status, "/proc/%d", pid);
if(stat(status, &sb))
continue;
my_getpwuid(curstatus.user, sb.st_uid, sizeof(curstatus.user));
sprintf(status, "/proc/%d/stat", pid);
if((fp = fopen(status, "r")) == NULL)
continue;
#ifdef CONFIG_SELINUX
if(use_selinux)
{
if(fstat_secure(fileno(fp), &sb, sid))
continue;
}
else
#endif
name = fgets(buf, sizeof(buf), fp);
fclose(fp);
if(name == NULL)
continue;
name = strrchr(buf, ')'); /* split into "PID (cmd" and "<rest>" */
if(name == 0 || name[1] != ' ')
continue;
*name = 0;
sscanf(buf, "%*s (%15c", curstatus.short_cmd);
n = sscanf(name+2,
"%c %d "
"%*s %*s %*s %*s " /* pgrp, session, tty, tpgid */
"%*s %*s %*s %*s %*s " /* flags, min_flt, cmin_flt, maj_flt, cmaj_flt */
#ifdef FEATURE_CPU_USAGE_PERCENTAGE
"%lu %lu "
#else
"%*s %*s "
#endif
"%*s %*s %*s " /* cutime, cstime, priority */
"%ld "
"%*s %*s %*s " /* timeout, it_real_value, start_time */
"%*s " /* vsize */
"%ld",
curstatus.state, &curstatus.ppid,
#ifdef FEATURE_CPU_USAGE_PERCENTAGE
&curstatus.utime, &curstatus.stime,
#endif
&tasknice,
&curstatus.rss);
#ifdef FEATURE_CPU_USAGE_PERCENTAGE
if(n != 6)
#else
if(n != 4)
#endif
continue;
if (curstatus.rss == 0 && curstatus.state[0] != 'Z')
curstatus.state[1] = 'W';
else
curstatus.state[1] = ' ';
if (tasknice < 0)
curstatus.state[2] = '<';
else if (tasknice > 0)
curstatus.state[2] = 'N';
else
curstatus.state[2] = ' ';
#ifdef PAGE_SHIFT
curstatus.rss <<= (PAGE_SHIFT - 10); /* 2**10 = 1kb */
#else
curstatus.rss *= (getpagesize() >> 10); /* 2**10 = 1kb */
#endif
if(save_user_arg0) {
sprintf(status, "/proc/%d/cmdline", pid);
if((fp = fopen(status, "r")) == NULL)
continue;
if((n=fread(buf, 1, sizeof(buf)-1, fp)) > 0) {
if(buf[n-1]=='\n')
buf[--n] = 0;
name = buf;
while(n) {
if(((unsigned char)*name) < ' ')
*name = ' ';
name++;
n--;
}
*name = 0;
if(buf[0])
curstatus.cmd = strdup(buf);
/* if NULL it work true also */
}
fclose(fp);
}
return memcpy(&ret_status, &curstatus, sizeof(procps_status_t));
}
}
