release/src/router/dnscrypt/src/plugins/example-ldns-opendns-deviceid/example-ldns-opendns-deviceid.c

   1
   2 #include <assert.h>
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <string.h>
   6
   7 #ifdef _WIN32
   8 # include <ws2tcpip.h>
   9 #endif
  10
  11 #include <dnscrypt/plugin.h>
  12 #include <ldns/ldns.h>
  13
  14 DCPLUGIN_MAIN(__FILE__);
  15
  16 #define EDNS_HEADER "0004000f4f70656e444e53"
  17 #define EDNS_DEV_ID "cafebabedeadbeef"
  18
  19 const char *
  20 dcplugin_description(DCPlugin * const dcplugin)
  21 {
  22     return "Add an OpenDNS device identifier to outgoing queries";
  23 }
  24
  25 const char *
  26 dcplugin_long_description(DCPlugin * const dcplugin)
  27 {
  28     return
  29         "This plugin tags outgoing packets with a 64-bit shared secret key,\n"
  30         "that the OpenDNS Umbrella service uses to identify users.\n"
  31         "\n"
  32         "If you happen to have an OpenDNS Umbrella account,\n"
  33         "your secret key ('device') can be retrieved with:\n"
  34         "\n"
  35         "$ dig txt debug.opendns.com\n"
  36         "\n"
  37         "# env OPENDNS_DEVICE_ID='<device id>' dnscrypt-proxy --plugin \n"
  38         "  libdcplugin_example_ldns_opendns_deviceid.la";
  39 }
  40
  41 int
  42 dcplugin_init(DCPlugin * const dcplugin, int argc, char *argv[])
  43 {
  44     char   *device_id;
  45     char   *device_id_env;
  46     char   *edns_hex;
  47     size_t  edns_hex_size = sizeof EDNS_HEADER EDNS_DEV_ID;
  48
  49     edns_hex = malloc(sizeof EDNS_HEADER EDNS_DEV_ID);
  50     dcplugin_set_user_data(dcplugin, edns_hex);
  51     if (edns_hex == NULL) {
  52         return -1;
  53     }
  54     memcpy(edns_hex, EDNS_HEADER EDNS_DEV_ID, edns_hex_size);
  55     assert(sizeof EDNS_DEV_ID - 1U == (size_t) 16U);
  56     device_id = device_id_env = getenv("OPENDNS_DEVICE_ID");
  57     if (device_id != NULL) {
  58         memcpy(edns_hex + sizeof EDNS_HEADER - (size_t) 1U,
  59                device_id, sizeof EDNS_DEV_ID);
  60     }
  61     if (device_id_env != NULL) {
  62         memset(device_id_env, 0, strlen(device_id_env));
  63     }
  64     return 0;
  65 }
  66
  67 int
  68 dcplugin_destroy(DCPlugin *dcplugin)
  69 {
  70     free(dcplugin_get_user_data(dcplugin));
  71
  72     return 0;
  73 }
  74
  75 DCPluginSyncFilterResult
  76 dcplugin_sync_pre_filter(DCPlugin *dcplugin, DCPluginDNSPacket *dcp_packet)
  77 {
  78     uint8_t  *new_packet;
  79     ldns_rdf *edns_data;
  80     ldns_pkt *packet;
  81     size_t    new_packet_size;
  82
  83     ldns_wire2pkt(&packet, dcplugin_get_wire_data(dcp_packet),
  84                   dcplugin_get_wire_data_len(dcp_packet));
  85
  86     edns_data = ldns_rdf_new_frm_str(LDNS_RDF_TYPE_HEX,
  87                                      dcplugin_get_user_data(dcplugin));
  88     ldns_pkt_set_edns_data(packet, edns_data);
  89
  90     ldns_pkt2wire(&new_packet, packet, &new_packet_size);
  91     if (dcplugin_get_wire_data_max_len(dcp_packet) >= new_packet_size) {
  92         dcplugin_set_wire_data(dcp_packet, new_packet, new_packet_size);
  93     }
  94
  95     free(new_packet);
  96     ldns_pkt_free(packet);
  97
  98     return DCP_SYNC_FILTER_RESULT_OK;
  99 }