mirror of
https://git.rtems.org/rtems-libbsd/
synced 2025-06-10 19:52:26 +08:00
9449f151d0
The sources can be obtained via: http://opensource.apple.com/tarballs/mDNSResponder/mDNSResponder-544.tar.gz
275 lines
10 KiB
C
275 lines
10 KiB
C
/* -*- Mode: C; tab-width: 4 -*-
|
|
*
|
|
* Copyright (c) 2002-2003 Apple Computer, Inc. All rights reserved.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include "dns_sd.h"
|
|
|
|
@interface RegistrationController : NSObject
|
|
{
|
|
IBOutlet NSTableColumn *typeColumn;
|
|
IBOutlet NSTableColumn *nameColumn;
|
|
IBOutlet NSTableColumn *portColumn;
|
|
IBOutlet NSTableColumn *domainColumn;
|
|
IBOutlet NSTableColumn *textColumn;
|
|
|
|
IBOutlet NSTableView *serviceDisplayTable;
|
|
|
|
IBOutlet NSTextField *serviceTypeField;
|
|
IBOutlet NSTextField *serviceNameField;
|
|
IBOutlet NSTextField *servicePortField;
|
|
IBOutlet NSTextField *serviceDomainField;
|
|
IBOutlet NSTextField *serviceTextField;
|
|
|
|
NSMutableArray *srvtypeKeys;
|
|
NSMutableArray *srvnameKeys;
|
|
NSMutableArray *srvportKeys;
|
|
NSMutableArray *srvdomainKeys;
|
|
NSMutableArray *srvtextKeys;
|
|
|
|
NSMutableDictionary *registeredDict;
|
|
}
|
|
|
|
- (IBAction)registerService:(id)sender;
|
|
- (IBAction)unregisterService:(id)sender;
|
|
|
|
- (IBAction)addNewService:(id)sender;
|
|
- (IBAction)removeSelected:(id)sender;
|
|
|
|
@end
|
|
|
|
void reg_reply
|
|
(
|
|
DNSServiceRef sdRef,
|
|
DNSServiceFlags flags,
|
|
DNSServiceErrorType errorCode,
|
|
const char *name,
|
|
const char *regtype,
|
|
const char *domain,
|
|
void *context
|
|
)
|
|
{
|
|
// registration reply
|
|
printf("Got a reply from the server with error %d\n", errorCode);
|
|
return;
|
|
}
|
|
|
|
static void myCFSocketCallBack(CFSocketRef cfs, CFSocketCallBackType CallBackType, CFDataRef address, const void *data, void *context)
|
|
{
|
|
DNSServiceProcessResult((DNSServiceRef)context);
|
|
}
|
|
|
|
static void addDNSServiceRefToRunLoop(DNSServiceRef ref)
|
|
{
|
|
int s = DNSServiceRefSockFD(ref);
|
|
CFSocketContext myCFSocketContext = { 0, ref, NULL, NULL, NULL };
|
|
CFSocketRef c = CFSocketCreateWithNative(kCFAllocatorDefault, s, kCFSocketReadCallBack, myCFSocketCallBack, &myCFSocketContext);
|
|
CFRunLoopSourceRef rls = CFSocketCreateRunLoopSource(kCFAllocatorDefault, c, 0);
|
|
CFRunLoopAddSource(CFRunLoopGetCurrent(), rls, kCFRunLoopDefaultMode);
|
|
CFRelease(rls);
|
|
}
|
|
|
|
|
|
@implementation RegistrationController
|
|
|
|
- (void)registerDefaults
|
|
{
|
|
NSMutableDictionary *regDict = [NSMutableDictionary dictionary];
|
|
|
|
NSArray *typeArray = [NSArray arrayWithObjects:@"_ftp._tcp.", @"_ssh._tcp.", @"_tftp._tcp.", @"_http._tcp.", @"_printer._tcp.", @"_afpovertcp._tcp.", nil];
|
|
NSArray *nameArray = [NSArray arrayWithObjects:@"My ftp Server", @"My Computer", @"Testing Boot Image", @"A Web Server", @"Steve's Printer", @"Company AppleShare Server", nil];
|
|
NSArray *portArray = [NSArray arrayWithObjects:@"21", @"22", @"69", @"80", @"515", @"548", nil];
|
|
NSArray *domainArray = [NSArray arrayWithObjects:@"", @"", @"", @"", @"", @"", nil];
|
|
NSArray *textArray = [NSArray arrayWithObjects:@"", @"", @"image=mybootimage", @"path=/index.html", @"rn=lpt1", @"Vol=Public", nil];
|
|
|
|
[regDict setObject:typeArray forKey:@"SrvTypeKeys"];
|
|
[regDict setObject:nameArray forKey:@"SrvNameKeys"];
|
|
[regDict setObject:portArray forKey:@"SrvPortKeys"];
|
|
[regDict setObject:domainArray forKey:@"SrvDomainKeys"];
|
|
[regDict setObject:textArray forKey:@"SrvTextKeys"];
|
|
|
|
[[NSUserDefaults standardUserDefaults] registerDefaults:regDict];
|
|
}
|
|
|
|
- (id)init
|
|
{
|
|
srvtypeKeys = [[NSMutableArray array] retain]; //Define arrays for Type, Domain, and Name
|
|
srvnameKeys = [[NSMutableArray array] retain];
|
|
srvportKeys = [[NSMutableArray array] retain];
|
|
srvdomainKeys = [[NSMutableArray array] retain];
|
|
srvtextKeys = [[NSMutableArray array] retain];
|
|
|
|
registeredDict = [[NSMutableDictionary alloc] init];
|
|
|
|
[self registerDefaults];
|
|
return [super init];
|
|
}
|
|
|
|
- (void)awakeFromNib //BrowserController startup procedure
|
|
{
|
|
[srvtypeKeys addObjectsFromArray:[[NSUserDefaults standardUserDefaults] arrayForKey:@"SrvTypeKeys"]];
|
|
[srvnameKeys addObjectsFromArray:[[NSUserDefaults standardUserDefaults] arrayForKey:@"SrvNameKeys"]];
|
|
[srvportKeys addObjectsFromArray:[[NSUserDefaults standardUserDefaults] arrayForKey:@"SrvPortKeys"]];
|
|
[srvdomainKeys addObjectsFromArray:[[NSUserDefaults standardUserDefaults] arrayForKey:@"SrvDomainKeys"]];
|
|
[srvtextKeys addObjectsFromArray:[[NSUserDefaults standardUserDefaults] arrayForKey:@"SrvTextKeys"]];
|
|
|
|
[serviceDisplayTable reloadData]; //Reload (redraw) data in fields
|
|
|
|
}
|
|
|
|
|
|
|
|
- (IBAction)registerService:(id)sender
|
|
{
|
|
int selectedRow = [serviceDisplayTable selectedRow];
|
|
CFMachPortContext context;
|
|
DNSServiceRef dns_client;
|
|
|
|
if (selectedRow < 0) {
|
|
return;
|
|
}
|
|
|
|
NSString *key = [srvtypeKeys objectAtIndex:selectedRow];
|
|
if ([registeredDict objectForKey:key]) { printf("Already registered\n"); return; }
|
|
|
|
context.version = 1;
|
|
context.info = 0;
|
|
context.retain = NULL;
|
|
context.release = NULL;
|
|
context.copyDescription = NULL;
|
|
unsigned char txtbuffer[300];
|
|
strncpy(txtbuffer+1, [[srvtextKeys objectAtIndex:selectedRow] UTF8String], sizeof(txtbuffer)-1);
|
|
txtbuffer[0] = strlen(txtbuffer+1);
|
|
|
|
DNSServiceErrorType err = DNSServiceRegister
|
|
(
|
|
&dns_client, 0, 0,
|
|
[[srvnameKeys objectAtIndex:selectedRow] UTF8String],
|
|
[key UTF8String],
|
|
[[srvdomainKeys objectAtIndex:selectedRow] UTF8String],
|
|
NULL, htons([[srvportKeys objectAtIndex:selectedRow] intValue]),
|
|
txtbuffer[0]+1, txtbuffer,
|
|
reg_reply,
|
|
nil
|
|
);
|
|
if (err)
|
|
printf("DNSServiceRegister failed %d\n", err);
|
|
else
|
|
{
|
|
addDNSServiceRefToRunLoop(dns_client);
|
|
[registeredDict setObject:[NSNumber numberWithUnsignedInt:(unsigned int)dns_client] forKey:key];
|
|
}
|
|
}
|
|
|
|
- (IBAction)unregisterService:(id)sender
|
|
{
|
|
int selectedRow = [serviceDisplayTable selectedRow];
|
|
NSString *key = [srvtypeKeys objectAtIndex:selectedRow];
|
|
|
|
NSNumber *refPtr = [registeredDict objectForKey:key];
|
|
DNSServiceRef ref = (DNSServiceRef)[refPtr unsignedIntValue];
|
|
|
|
if (ref) {
|
|
DNSServiceRefDeallocate(ref);
|
|
[registeredDict removeObjectForKey:key];
|
|
}
|
|
}
|
|
|
|
-(void)tableView:(NSTableView *)theTableView setObjectValue:(id)object forTableColumn:(NSTableColumn *)tableColumn row:(int)row
|
|
{
|
|
if (row<0) return;
|
|
}
|
|
|
|
- (int)numberOfRowsInTableView:(NSTableView *)theTableView //Begin mandatory TableView methods
|
|
{
|
|
return [srvtypeKeys count];
|
|
}
|
|
|
|
- (id)tableView:(NSTableView *)theTableView objectValueForTableColumn:(NSTableColumn *)theColumn row:(int)rowIndex
|
|
{
|
|
if (theColumn == typeColumn) {
|
|
return [srvtypeKeys objectAtIndex:rowIndex];
|
|
}
|
|
if (theColumn == nameColumn) {
|
|
return [srvnameKeys objectAtIndex:rowIndex];
|
|
}
|
|
if (theColumn == portColumn) {
|
|
return [srvportKeys objectAtIndex:rowIndex];
|
|
}
|
|
if (theColumn == domainColumn) {
|
|
return [srvdomainKeys objectAtIndex:rowIndex];
|
|
}
|
|
if (theColumn == textColumn) {
|
|
return [srvtextKeys objectAtIndex:rowIndex];
|
|
}
|
|
|
|
return(0);
|
|
} //End of mandatory TableView methods
|
|
|
|
- (IBAction)removeSelected:(id)sender
|
|
{
|
|
// remove the selected row and force a refresh
|
|
|
|
int selectedRow = [serviceDisplayTable selectedRow];
|
|
|
|
if (selectedRow) {
|
|
|
|
[srvtypeKeys removeObjectAtIndex:selectedRow];
|
|
[srvnameKeys removeObjectAtIndex:selectedRow];
|
|
[srvportKeys removeObjectAtIndex:selectedRow];
|
|
[srvdomainKeys removeObjectAtIndex:selectedRow];
|
|
[srvtextKeys removeObjectAtIndex:selectedRow];
|
|
|
|
[[NSUserDefaults standardUserDefaults] setObject:srvtypeKeys forKey:@"SrvTypeKeys"];
|
|
[[NSUserDefaults standardUserDefaults] setObject:srvnameKeys forKey:@"SrvNameKeys"];
|
|
[[NSUserDefaults standardUserDefaults] setObject:srvportKeys forKey:@"SrvPortKeys"];
|
|
[[NSUserDefaults standardUserDefaults] setObject:srvdomainKeys forKey:@"SrvDomainKeys"];
|
|
[[NSUserDefaults standardUserDefaults] setObject:srvtextKeys forKey:@"SrvTextKeys"];
|
|
|
|
[serviceDisplayTable reloadData];
|
|
}
|
|
}
|
|
|
|
- (IBAction)addNewService:(id)sender
|
|
{
|
|
// add new entries from the edit fields to the arrays for the defaults
|
|
|
|
if ([[serviceTypeField stringValue] length] && [[serviceNameField stringValue] length] && [[serviceDomainField stringValue] length]&& [[servicePortField stringValue] length]) {
|
|
[srvtypeKeys addObject:[serviceTypeField stringValue]];
|
|
[srvnameKeys addObject:[serviceNameField stringValue]];
|
|
[srvportKeys addObject:[servicePortField stringValue]];
|
|
[srvdomainKeys addObject:[serviceDomainField stringValue]];
|
|
[srvtextKeys addObject:[serviceTextField stringValue]];
|
|
|
|
[[NSUserDefaults standardUserDefaults] setObject:srvtypeKeys forKey:@"SrvTypeKeys"];
|
|
[[NSUserDefaults standardUserDefaults] setObject:srvnameKeys forKey:@"SrvNameKeys"];
|
|
[[NSUserDefaults standardUserDefaults] setObject:srvportKeys forKey:@"SrvPortKeys"];
|
|
[[NSUserDefaults standardUserDefaults] setObject:srvdomainKeys forKey:@"SrvDomainKeys"];
|
|
[[NSUserDefaults standardUserDefaults] setObject:srvtextKeys forKey:@"SrvTextKeys"];
|
|
|
|
[serviceDisplayTable reloadData];
|
|
} else {
|
|
NSBeep();
|
|
}
|
|
|
|
}
|
|
|
|
@end
|
|
|
|
int main(int argc, const char *argv[])
|
|
{
|
|
return NSApplicationMain(argc, argv);
|
|
}
|