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
126
|
// hostobj.m--This object manages the list of hosts that are allowed
// to connect. Use addhost to add an address without prompting. Typically
// this is only done for the localhost. When you get a new peer name
// off the network, use queryhost to see if it is already in the list
// and if not, prompt the user to see if a connection should be allowed.
//
// 1999-Feb-21 - [AFT]
//---
#import "hostobj.h"
#import <sys/types.h>
#import <sys/socket.h>
#import <netinet/in.h>
#import <netdb.h>
@implementation hostobj
- initwithdispatch:(id)adispatch
{
mydispatch = adispatch;
hostlist = [Storage newCount:0
elementSize:4
description:"[i]"];
hostmenu = [[NSMenu alloc]
initWithTitle:[NSString stringWithCString:"Hosts"]];
[hostmenu sizeToFit];
return self;
}
- (NSMenu *) gethostmenu
{
return hostmenu;;
}
- (void) addhost:(int)iaddr
{
struct hostent *host;
struct in_addr addr;
// Add to list of approved addresses
[hostlist addElement:(void *)&iaddr];
// Get actual machine name and add to hosts submenu.
addr.s_addr = iaddr;
host = gethostbyaddr((char *)&addr, sizeof(addr), AF_INET);
[[hostmenu addItemWithTitle:[NSString stringWithCString:host->h_name]
action:@selector(removehost:)
keyEquivalent:@""]
setTarget:mydispatch];
[hostmenu sizeToFit];
return;
}
- (BOOL) queryhost:(int)isuspect
{
struct hostent *host;
struct in_addr addr;
char cbuf[128];
int *ihost;
int i, iresult, ltmp, nhost;
// If the suspect IP address is in the approved list, then return true.
nhost=[hostlist count];
for(i=0; i<nhost; i++) {
ihost=[hostlist elementAt:i];
if ( isuspect == *ihost ) {
return TRUE;
}
}
// Not in approved list. Get actual machine name.
addr.s_addr = isuspect;
host = gethostbyaddr((char *)&addr, sizeof(addr), AF_INET);
ltmp = strlen(host->h_name);
memcpy(&cbuf, host->h_name, ltmp);
cbuf[ltmp]='\0';
// And prompt user to see if this is OK.
iresult = NSRunAlertPanel(@"pgview: Accept connection from",
[NSString stringWithCString:cbuf],
@"Accept",@"Reject",nil);
// If it is OK, then add to list of approved hosts.
if ( iresult ) [self addhost:isuspect];
return iresult;
}
- (void) removehost:(NSMenuItem *) sender
{
struct hostent *ahost;
char cbuf[128];
int *ihost;
int i, iresult, isuspect, nhost;
// Come here if the user has selected a host from the hosts submenu.
// The menuItem title conveniently contains the host name.
[[sender title] getCString:cbuf maxLength:128];
ahost = gethostbyname(cbuf);
if (ahost == 0) {
fprintf(stderr, "%s: unknown host", cbuf);
} else {
// We now have the host IP address. Find entry in hostlist.
memcpy(&isuspect, ahost->h_addr, 4);
nhost=[hostlist count];
for(i=0; i<nhost; i++) {
ihost=[hostlist elementAt:i];
if ( isuspect == *ihost ) {
// We now have the menuItem and the index into the hostlist. Give the
// user a chance to change mind before deleting.
iresult = NSRunAlertPanel(@"pgview: Remove system?",
[NSString stringWithCString:cbuf],
@"Yes",@"No",nil);
if ( iresult ) {
[hostlist removeElementAt:i];
[hostmenu removeItem:sender];
}
return;
}
}
printf("pgview internal error, no IP address for=%s\n",cbuf);
}
return;
}
@end
|
