vcnet/dns_sd.c

   1 /*
   2  * Dynamic wrapper for Bonjour SDK for Windows.
   3  *
   4  * Copyright 2018 by Apple Inc.
   5  *
   6  * Licensed under Apache License v2.0.  See the file "LICENSE" for more
   7  * information.
   8  */
   9
  10 //#include <cups/http-private.h>
  11 #include <cups/thread-private.h>
  12 #include "dns_sd.h"
  13
  14
  15 /*
  16  * Pointers for functions...
  17  */
  18
  19 static int              dnssd_initialized = 0;
  20 static _cups_mutex_t    dnssd_mutex = _CUPS_MUTEX_INITIALIZER;
  21 static DNSServiceErrorType (*dnssd_add_record)(DNSServiceRef sdRef, DNSRecordRef *RecordRef, DNSServiceFlags flags, uint16_t rrtype, uint16_t rdlen, const void *rdata, uint32_t ttl);
  22 static DNSServiceErrorType (*dnssd_browse)(DNSServiceRef *sdRef, DNSServiceFlags flags, uint32_t interfaceIndex, const char *regtype, const char *domain, DNSServiceBrowseReply callBack, void *context);
  23 static DNSServiceErrorType (*dnssd_construct_full_name)(char * const fullName, const char * const service, const char * const regtype, const char * const domain);
  24 static DNSServiceErrorType (*dnssd_create_connection)(DNSServiceRef *sdRef);
  25 static DNSServiceErrorType (*dnssd_process_result)(DNSServiceRef sdRef);
  26 static DNSServiceErrorType (*dnssd_query_record)(DNSServiceRef *sdRef, DNSServiceFlags flags, uint32_t interfaceIndex, const char *fullname, uint16_t rrtype, uint16_t rrclass, DNSServiceQueryRecordReply callBack, void *context);
  27 static void (*dnssd_deallocate)(DNSServiceRef sdRef);
  28 static int (*dnssd_sock_fd)(DNSServiceRef sdRef);
  29 static DNSServiceErrorType (*dnssd_register)(DNSServiceRef *sdRef, DNSServiceFlags flags, uint32_t interfaceIndex, const char *name, const char *regtype, const char *domain, const char *host, uint16_t port, uint16_t txtLen, const void *txtRecord, DNSServiceRegisterReply callBack, void *context);
  30 static DNSServiceErrorType (*dnssd_remove_record)(DNSServiceRef sdRef, DNSRecordRef RecordRef, DNSServiceFlags flags);
  31 static DNSServiceErrorType (*dnssd_resolve)(DNSServiceRef *sdRef, DNSServiceFlags flags, uint32_t interfaceIndex, const char *name, const char *regtype, const char *domain, DNSServiceResolveReply callBack, void *context);
  32 static DNSServiceErrorType (*dnssd_update_record)(DNSServiceRef sdRef, DNSRecordRef RecordRef, DNSServiceFlags flags, uint16_t rdlen, const void *rdata, uint32_t ttl);
  33
  34 static void (*dnssd_txt_create)(TXTRecordRef *txtRecord, uint16_t bufferLen, void *buffer);
  35 static void (*dnssd_txt_deallocate)(TXTRecordRef *txtRecord);
  36 static const void *(*dnssd_txt_get_bytes_ptr)(const TXTRecordRef *txtRecord);
  37 static uint16_t (*dnssd_txt_get_count)(uint16_t txtLen, const void *txtRecord);
  38 static uint16_t (*dnssd_txt_get_length)(const TXTRecordRef *txtRecord);
  39 static DNSServiceErrorType (*dnssd_txt_get_item_at_index)(uint16_t txtLen, const void *txtRecord, uint16_t itemIndex, uint16_t keyBufLen, char *key, uint8_t *valueLen, const void **value);
  40 static const void *(*dnssd_txt_get_value_ptr)(uint16_t txtLen, const void *txtRecord, const char *key, uint8_t *valueLen);
  41 static DNSServiceErrorType (*dnssd_txt_set_value)(TXTRecordRef *txtRecord, const char *key, uint8_t valueSize, const void *value);
  42
  43
  44 /*
  45  * Function to initialize pointers...
  46  */
  47
  48 static void
  49 dnssd_init(void)
  50 {
  51   _cupsMutexLock(&dnssd_mutex);
  52   if (!dnssd_initialized)
  53   {
  54     HINSTANCE   dll_handle = LoadLibraryA("dnssd.dll");
  55
  56     if (dll_handle)
  57     {
  58       dnssd_add_record          = (DNSServiceErrorType (*)(DNSServiceRef, DNSRecordRef *, DNSServiceFlags, uint16_t, uint16_t, const void *, uint32_t))GetProcAddress(dll_handle, "DNSServiceAddRecord");
  59       dnssd_browse              = (DNSServiceErrorType(*)(DNSServiceRef *, DNSServiceFlags, uint32_t, const char *, const char *, DNSServiceBrowseReply, void *))GetProcAddress(dll_handle, "DNSServiceBrowse");
  60       dnssd_construct_full_name = (DNSServiceErrorType(*)(char * const, const char * const, const char * const, const char * const))GetProcAddress(dll_handle, "DNSServiceConstructFullName");
  61       dnssd_create_connection   = (DNSServiceErrorType(*)(DNSServiceRef *))GetProcAddress(dll_handle, "DNSServiceCreateConnection");
  62       dnssd_deallocate          = (DNSServiceErrorType(*)(DNSServiceRef))GetProcAddress(dll_handle, "DNSServiceRefDeallocate");
  63       dnssd_process_result      = (DNSServiceErrorType(*)(DNSServiceRef))GetProcAddress(dll_handle, "DNSServiceProcessResult");
  64       dnssd_query_record        = (DNSServiceErrorType(*)(DNSServiceRef *, DNSServiceFlags, uint32_t, const char *, uint16_t, uint16_t, DNSServiceQueryRecordReply, void *))GetProcAddress(dll_handle, "DNSServiceQueryRecord");
  65       dnssd_register            = (DNSServiceErrorType(*)(DNSServiceRef *, DNSServiceFlags, uint32_t, const char *, const char *, const char *, const char *, uint16_t, uint16_t, const void *, DNSServiceRegisterReply, void *))GetProcAddress(dll_handle, "DNSServiceRegister");
  66       dnssd_remove_record       = (DNSServiceErrorType(*)(DNSServiceRef, DNSRecordRef, DNSServiceFlags))GetProcAddress(dll_handle, "DNSServiceRemoveRecord");
  67       dnssd_resolve             = (DNSServiceErrorType(*)(DNSServiceRef *, DNSServiceFlags, uint32_t, const char *, const char *, const char *, DNSServiceResolveReply, void *))GetProcAddress(dll_handle, "DNSServiceResolve");
  68       dnssd_sock_fd             = (int(*)(DNSServiceRef))GetProcAddress(dll_handle, "DNSServiceRefSockFD");
  69       dnssd_update_record       = (DNSServiceErrorType(*)(DNSServiceRef, DNSRecordRef, DNSServiceFlags, uint16_t, const void *, uint32_t))GetProcAddress(dll_handle, "DNSServiceUpdateRecord");
  70
  71       dnssd_txt_create          = (void (*)(TXTRecordRef *, uint16_t, void *))GetProcAddress(dll_handle, "TXTRecordCreate");
  72       dnssd_txt_deallocate      = (void (*)(TXTRecordRef *))GetProcAddress(dll_handle, "TXTRecordDeallocate");
  73       dnssd_txt_get_bytes_ptr   = (const void *(*)(const TXTRecordRef *))GetProcAddress(dll_handle, "TXTRecordGetBytesPtr");
  74       dnssd_txt_get_count       = (uint16_t (*)(uint16_t, const void *))GetProcAddress(dll_handle, "TXTRecordGetCount");
  75       dnssd_txt_get_item_at_index = (DNSServiceErrorType (*)(uint16_t, const void *, uint16_t, uint16_t, char *, uint8_t *, const void **))GetProcAddress(dll_handle, "TXTRecordGetItemAtIndex");
  76       dnssd_txt_get_length      = (uint16_t (*)(const TXTRecordRef *))GetProcAddress(dll_handle, "TXTRecordGetLength");
  77       dnssd_txt_get_value_ptr   = (const void *(*)(uint16_t, const void *, const char *, uint8_t *))GetProcAddress(dll_handle, "TXTRecordGetValuePtr");
  78       dnssd_txt_set_value       = (DNSServiceErrorType (*)(TXTRecordRef *, const char *, uint8_t, const void *))GetProcAddress(dll_handle, "TXTRecordSetValue");
  79     }
  80
  81     dnssd_initialized = 1;
  82   }
  83   _cupsMutexUnlock(&dnssd_mutex);
  84 }
  85
  86
  87 // DNSServiceAddRecord
  88 DNSServiceErrorType DNSSD_API DNSServiceAddRecord
  89     (
  90     DNSServiceRef                       sdRef,
  91     DNSRecordRef                        *RecordRef,
  92     DNSServiceFlags                     flags,
  93     uint16_t                            rrtype,
  94     uint16_t                            rdlen,
  95     const void                          *rdata,
  96     uint32_t                            ttl
  97     )
  98 {
  99   if (!dnssd_initialized)
 100     dnssd_init();
 101
 102   if (dnssd_add_record)
 103     return (*dnssd_add_record)(sdRef, RecordRef, flags, rrtype, rdlen, rdata, ttl);
 104   else
 105     return (-1);
 106 }
 107
 108
 109 // DNSServiceBrowse
 110 DNSServiceErrorType DNSSD_API DNSServiceBrowse
 111     (
 112     DNSServiceRef                       *sdRef,
 113     DNSServiceFlags                     flags,
 114     uint32_t                            interfaceIndex,
 115     const char                          *regtype,
 116     const char                          *domain,    /* may be NULL */
 117     DNSServiceBrowseReply               callBack,
 118     void                                *context    /* may be NULL */
 119     )
 120 {
 121   if (!dnssd_initialized)
 122     dnssd_init();
 123
 124   if (dnssd_browse)
 125     return (*dnssd_browse)(sdRef, flags, interfaceIndex, regtype, domain, callBack, context);
 126   else
 127     return (-1);
 128 }
 129
 130
 131 // DNSServiceConstructFullName
 132 DNSServiceErrorType DNSSD_API DNSServiceConstructFullName
 133     (
 134     char                            * const fullName,
 135     const char                      * const service,      /* may be NULL */
 136     const char                      * const regtype,
 137     const char                      * const domain
 138     )
 139 {
 140   if (!dnssd_initialized)
 141     dnssd_init();
 142
 143   if (dnssd_construct_full_name)
 144     return (*dnssd_construct_full_name)(fullName, service, regtype, domain);
 145   else
 146     return (-1);
 147 }
 148
 149
 150 // DNSServiceCreateConnection
 151 DNSServiceErrorType DNSSD_API DNSServiceCreateConnection(DNSServiceRef *sdRef)
 152 {
 153   if (!dnssd_initialized)
 154     dnssd_init();
 155
 156   if (dnssd_create_connection)
 157     return (*dnssd_create_connection)(sdRef);
 158   else
 159     return (-1);
 160 }
 161
 162
 163 // DNSServiceProcessResult
 164 DNSServiceErrorType DNSSD_API DNSServiceProcessResult(DNSServiceRef sdRef)
 165 {
 166   if (!dnssd_initialized)
 167     dnssd_init();
 168
 169   if (dnssd_process_result)
 170     return (*dnssd_process_result)(sdRef);
 171   else
 172     return (-1);
 173 }
 174
 175
 176 // DNSServiceQueryRecord
 177 DNSServiceErrorType DNSSD_API DNSServiceQueryRecord
 178     (
 179     DNSServiceRef                       *sdRef,
 180     DNSServiceFlags                     flags,
 181     uint32_t                            interfaceIndex,
 182     const char                          *fullname,
 183     uint16_t                            rrtype,
 184     uint16_t                            rrclass,
 185     DNSServiceQueryRecordReply          callBack,
 186     void                                *context  /* may be NULL */
 187     )
 188 {
 189   if (!dnssd_initialized)
 190     dnssd_init();
 191
 192   if (dnssd_query_record)
 193     return (*dnssd_query_record)(sdRef, flags, interfaceIndex, fullname, rrtype, rrclass, callBack, context);
 194   else
 195     return (-1);
 196 }
 197
 198
 199 // DNSServiceRefDeallocate
 200 void DNSSD_API DNSServiceRefDeallocate(DNSServiceRef sdRef)
 201 {
 202   if (!dnssd_initialized)
 203     dnssd_init();
 204
 205   if (dnssd_deallocate)
 206     (*dnssd_deallocate)(sdRef);
 207 }
 208
 209
 210 // DNSServiceRefSockFD
 211 int DNSSD_API DNSServiceRefSockFD(DNSServiceRef sdRef)
 212 {
 213   if (!dnssd_initialized)
 214     dnssd_init();
 215
 216   if (dnssd_sock_fd)
 217     return (*dnssd_sock_fd)(sdRef);
 218   else
 219     return (-1);
 220 }
 221
 222
 223 // DNSServiceRegister
 224 DNSServiceErrorType DNSSD_API DNSServiceRegister
 225     (
 226     DNSServiceRef                       *sdRef,
 227     DNSServiceFlags                     flags,
 228     uint32_t                            interfaceIndex,
 229     const char                          *name,         /* may be NULL */
 230     const char                          *regtype,
 231     const char                          *domain,       /* may be NULL */
 232     const char                          *host,         /* may be NULL */
 233     uint16_t                            port,          /* In network byte order */
 234     uint16_t                            txtLen,
 235     const void                          *txtRecord,    /* may be NULL */
 236     DNSServiceRegisterReply             callBack,      /* may be NULL */
 237     void                                *context       /* may be NULL */
 238     )
 239 {
 240   if (!dnssd_initialized)
 241     dnssd_init();
 242
 243   if (dnssd_register)
 244     return (*dnssd_register)(sdRef, flags, interfaceIndex, name, regtype, domain, host, port, txtLen, txtRecord, callBack, context);
 245   else
 246     return (-1);
 247 }
 248
 249
 250 // DNSServiceRemoveRecord
 251 DNSServiceErrorType DNSSD_API DNSServiceRemoveRecord
 252     (
 253     DNSServiceRef                 sdRef,
 254     DNSRecordRef                  RecordRef,
 255     DNSServiceFlags               flags
 256     )
 257 {
 258   if (!dnssd_initialized)
 259     dnssd_init();
 260
 261   if (dnssd_remove_record)
 262     return (*dnssd_remove_record)(sdRef, RecordRef, flags);
 263   else
 264     return (-1);
 265 }
 266
 267
 268 // DNSServiceResolve
 269 DNSServiceErrorType DNSSD_API DNSServiceResolve
 270     (
 271     DNSServiceRef                       *sdRef,
 272     DNSServiceFlags                     flags,
 273     uint32_t                            interfaceIndex,
 274     const char                          *name,
 275     const char                          *regtype,
 276     const char                          *domain,
 277     DNSServiceResolveReply              callBack,
 278     void                                *context  /* may be NULL */
 279     )
 280 {
 281   if (!dnssd_initialized)
 282     dnssd_init();
 283
 284   if (dnssd_resolve)
 285     return (*dnssd_resolve)(sdRef, flags, interfaceIndex, name, regtype, domain, callBack, context);
 286   else
 287     return (-1);
 288 }
 289
 290
 291 // DNSServiceUpdateRecord
 292 DNSServiceErrorType DNSSD_API DNSServiceUpdateRecord
 293     (
 294     DNSServiceRef                       sdRef,
 295     DNSRecordRef                        RecordRef,     /* may be NULL */
 296     DNSServiceFlags                     flags,
 297     uint16_t                            rdlen,
 298     const void                          *rdata,
 299     uint32_t                            ttl
 300     )
 301 {
 302   if (!dnssd_initialized)
 303     dnssd_init();
 304
 305   if (dnssd_update_record)
 306     return (*dnssd_update_record)(sdRef, RecordRef, flags, rdlen, rdata, ttl);
 307   else
 308     return (-1);
 309 }
 310
 311
 312 // TXTRecordCreate
 313 void DNSSD_API
 314 TXTRecordCreate(
 315     TXTRecordRef     *txtRecord,
 316     uint16_t         bufferLen,
 317     void             *buffer)
 318 {
 319   if (!dnssd_initialized)
 320     dnssd_init();
 321
 322   if (dnssd_txt_create)
 323     (*dnssd_txt_create)(txtRecord, bufferLen, buffer);
 324 }
 325
 326
 327 // TXTRecordDeallocate
 328 void DNSSD_API TXTRecordDeallocate
 329     (
 330     TXTRecordRef     *txtRecord
 331     )
 332 {
 333   if (!dnssd_initialized)
 334     dnssd_init();
 335
 336   if (dnssd_txt_deallocate)
 337     (*dnssd_txt_deallocate)(txtRecord);
 338 }
 339
 340
 341 // TXTRecordGetBytesPtr
 342 const void * DNSSD_API TXTRecordGetBytesPtr
 343     (
 344     const TXTRecordRef *txtRecord
 345     )
 346 {
 347   if (!dnssd_initialized)
 348     dnssd_init();
 349
 350   if (dnssd_txt_get_bytes_ptr)
 351     return (*dnssd_txt_get_bytes_ptr)(txtRecord);
 352   else
 353     return (NULL);
 354 }
 355
 356
 357 // TXTRecordGetLength
 358 uint16_t DNSSD_API TXTRecordGetLength
 359     (
 360     const TXTRecordRef *txtRecord
 361     )
 362 {
 363   if (!dnssd_initialized)
 364     dnssd_init();
 365
 366   if (dnssd_txt_get_length)
 367     return (*dnssd_txt_get_length)(txtRecord);
 368   else
 369     return (0);
 370 }
 371
 372
 373 // TXTRecordSetValue
 374 DNSServiceErrorType DNSSD_API TXTRecordSetValue
 375     (
 376     TXTRecordRef     *txtRecord,
 377     const char       *key,
 378     uint8_t          valueSize,        /* may be zero */
 379     const void       *value            /* may be NULL */
 380     )
 381 {
 382   if (!dnssd_initialized)
 383     dnssd_init();
 384
 385   if (dnssd_txt_set_value)
 386     return (*dnssd_txt_set_value)(txtRecord, key, valueSize, value);
 387   else
 388     return (-1);
 389 }
 390
 391
 392 // TXTRecordGetCount
 393 uint16_t DNSSD_API
 394 TXTRecordGetCount(
 395     uint16_t         txtLen,
 396     const void       *txtRecord)
 397 {
 398   if (!dnssd_initialized)
 399     dnssd_init();
 400
 401   if (dnssd_txt_get_count)
 402     return (*dnssd_txt_get_count)(txtLen, txtRecord);
 403   else
 404     return (0);
 405 }
 406
 407
 408 // TXTRecordGetItemAtIndex
 409 DNSServiceErrorType DNSSD_API
 410 TXTRecordGetItemAtIndex(
 411     uint16_t         txtLen,
 412     const void       *txtRecord,
 413     uint16_t         itemIndex,
 414     uint16_t         keyBufLen,
 415     char             *key,
 416     uint8_t          *valueLen,
 417     const void       **value)
 418 {
 419   if (!dnssd_initialized)
 420     dnssd_init();
 421
 422   if (dnssd_txt_get_item_at_index)
 423     return (*dnssd_txt_get_item_at_index)(txtLen, txtRecord, itemIndex, keyBufLen, key, valueLen, value);
 424   else
 425     return (-1);
 426 }
 427
 428
 429 // TXTRecordGetValuePtr
 430 const void * DNSSD_API
 431 TXTRecordGetValuePtr(
 432     uint16_t         txtLen,
 433     const void       *txtRecord,
 434     const char       *key,
 435     uint8_t          *valueLen)
 436 {
 437   if (!dnssd_initialized)
 438     dnssd_init();
 439
 440   if (dnssd_txt_get_value_ptr)
 441     return (*dnssd_txt_get_value_ptr)(txtLen, txtRecord, key, valueLen);
 442   else
 443     return (NULL);
 444 }