/* -*- Mode: C; tab-width: 4 -*-
*
* Copyright (c) 2004 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 <stdio.h> // For printf()
#include <string.h> // For strcpy()
#include <Events.h> // For WaitNextEvent()
#include <OpenTransport.h>
#include <OpenTptInternet.h>
#include <SIOUX.h> // For SIOUXHandleOneEvent()
#include "dns_sd.h"
typedef union { UInt8 b[2]; UInt16 NotAnInteger; } mDNSOpaque16;
static UInt16 mDNSVal16(mDNSOpaque16 x) { return((UInt16)(x.b[0]<<8 | x.b[1])); }
static mDNSOpaque16 mDNSOpaque16fromIntVal(UInt16 v)
{ mDNSOpaque16 x; x.b[0] = (UInt8)(v >> 8); x.b[1] = (UInt8)(v & 0xFF); return(x); }
typedef struct RegisteredService_struct RegisteredService;
struct RegisteredService_struct
{
RegisteredService *next;
DNSServiceRef sdRef;
Boolean gotresult;
DNSServiceErrorType errorCode;
char namestr[64];
char typestr[kDNSServiceMaxDomainName];
char domstr [kDNSServiceMaxDomainName];
};
static RegisteredService p1, p2, afp, http, njp;
static RegisteredService *services = NULL, **nextservice = &services;
static void RegCallback(DNSServiceRef sdRef, DNSServiceFlags flags, DNSServiceErrorType errorCode,
const char *name, const char *regtype, const char *domain, void *context)
{
RegisteredService *rs = (RegisteredService *)context;
(void)sdRef; // Unused
(void)flags; // Unused
rs->gotresult = true;
rs->errorCode = errorCode;
strcpy(rs->namestr, name);
strcpy(rs->typestr, regtype);
strcpy(rs->domstr, domain);
}
static DNSServiceErrorType RegisterService(RegisteredService *rs, mDNSOpaque16 OpaquePort,
const char name[], const char type[], const char domain[], const char txtinfo[])
{
DNSServiceErrorType err;
unsigned char txtbuffer[257];
strncpy((char*)txtbuffer+1, txtinfo, 255);
txtbuffer[256] = 0;
txtbuffer[0] = (unsigned char)strlen((char*)txtbuffer);
rs->gotresult = 0;
rs->errorCode = kDNSServiceErr_NoError;
err = DNSServiceRegister(&rs->sdRef, /* kDNSServiceFlagsAutoRename*/ 0, 0,
name, type, domain, NULL, OpaquePort.NotAnInteger, (unsigned short)(1+txtbuffer[0]), txtbuffer, RegCallback, rs);
if (err)
printf("RegisterService(%s %s %s) failed %d\n", name, type, domain, err);
else
{ *nextservice = rs; nextservice = &rs->next; }
return(err);
}
// RegisterFakeServiceForTesting() simulates the effect of services being registered on
// dynamically-allocated port numbers. No real service exists on that port -- this is just for testing.
static DNSServiceErrorType RegisterFakeServiceForTesting(RegisteredService *rs,
const char name[], const char type[], const char domain[], const char txtinfo[])
{
static UInt16 NextPort = 0xF000;
return RegisterService(rs, mDNSOpaque16fromIntVal(NextPort++), name, type, domain, txtinfo);
}
// CreateProxyRegistrationForRealService() checks to see if the given port is currently
// in use, and if so, advertises the specified service as present on that port.
// This is useful for advertising existing real services (Personal Web Sharing, Personal
// File Sharing, etc.) that currently don't register with mDNS Service Discovery themselves.
static DNSServiceErrorType CreateProxyRegistrationForRealService(RegisteredService *rs,
const char *servicetype, UInt16 PortAsNumber, const char txtinfo[])
{
mDNSOpaque16 OpaquePort = mDNSOpaque16fromIntVal(PortAsNumber);
InetAddress ia;
TBind bindReq;
OSStatus err;
TEndpointInfo endpointinfo;
EndpointRef ep = OTOpenEndpoint(OTCreateConfiguration(kTCPName), 0, &endpointinfo, &err);
if (!ep || err) { printf("OTOpenEndpoint (CreateProxyRegistrationForRealService) failed %d", err); return(err); }
ia.fAddressType = AF_INET;
ia.fPort = OpaquePort.NotAnInteger;
ia.fHost = 0;
bindReq.addr.maxlen = sizeof(ia);
bindReq.addr.len = sizeof(ia);
bindReq.addr.buf = (UInt8*)&ia;
bindReq.qlen = 0;
err = OTBind(ep, &bindReq, NULL);
if (err == kOTBadAddressErr)
err = RegisterService(rs, OpaquePort, "", servicetype, "local.", txtinfo);
else if (err)
printf("OTBind failed %d", err);
OTCloseProvider(ep);
return(err);
}
// YieldSomeTime() just cooperatively yields some time to other processes running on classic Mac OS
static Boolean YieldSomeTime(UInt32 milliseconds)
{
extern Boolean SIOUXQuitting;
EventRecord e;
WaitNextEvent(everyEvent, &e, milliseconds / 17, NULL);
SIOUXHandleOneEvent(&e);
return(SIOUXQuitting);
}
int main()
{
OSStatus err;
RegisteredService *s;
SIOUXSettings.asktosaveonclose = false;
SIOUXSettings.userwindowtitle = "\pMulticast DNS Responder";
printf("Multicast DNS Responder\n\n");
printf("This software reports errors using MacsBug breaks,\n");
printf("so if you don't have MacsBug installed your Mac may crash.\n\n");
printf("******************************************************************************\n\n");
err = InitOpenTransport();
if (err) { printf("InitOpenTransport failed %d", err); return(err); }
printf("Advertising Services...\n");
#define SRSET 0
#if SRSET==0
RegisterFakeServiceForTesting(&p1, "Web Server One", "_http._tcp.", "local.", "path=/index.html");
RegisterFakeServiceForTesting(&p2, "Web Server Two", "_http._tcp.", "local.", "path=/path.html");
#elif SRSET==1
RegisterFakeServiceForTesting(&p1, "Epson Stylus 900N", "_printer._tcp.", "local.", "rn=lpq1");
RegisterFakeServiceForTesting(&p2, "HP LaserJet", "_printer._tcp.", "local.", "rn=lpq2");
#else
RegisterFakeServiceForTesting(&p1, "My Printer", "_printer._tcp.", "local.", "rn=lpq3");
RegisterFakeServiceForTesting(&p2, "My Other Printer", "_printer._tcp.", "local.", "lrn=pq4");
#endif
// If AFP Server is running, register a record for it
CreateProxyRegistrationForRealService(&afp, "_afpovertcp._tcp.", 548, "");
// If Web Server is running, register a record for it
CreateProxyRegistrationForRealService(&http, "_http._tcp.", 80, "path=/index.html");
while (!YieldSomeTime(35))
for (s = services; s; s = s->next)
if (s->gotresult)
{
printf("%s %s %s registered\n", s->namestr, s->typestr, s->domstr);
s->gotresult = false;
}
for (s = services; s; s = s->next)
if (s->sdRef) DNSServiceRefDeallocate(s->sdRef);
CloseOpenTransport();
return(0);
}
