validate_tld.cpp

Go to the documentation of this file.

/* TLD library -- TLD validation command line tools
 * Copyright (c) 2011-2025  Made to Order Software Corp.  All Rights Reserved
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 *
 * The above copyright notice and this permission notice shall be included
 * in all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */
 
#include "libtld/tld.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
 
int err_count = 0;
 
int verbose = 0;
 
const char *schemes = "afp,adiumxtra,aw,beshare,bolo,cap,coap,crid,dns,feed,file,"
                      "finger,fish,ftp,ftps,git,gopher,http,https,icap,imap,"
                      "ipp,irc,irc6,ircs,mumble,mupdate,mysql,nfs,nntp,"
                      "opaquelocktoken,pop,psql,psyc,rmi,rsync,rtmp,rtsp,rtspu,"
                      "sftp,shttp,sieve,smb,snmp,soap.beep,soap.beeps,soldat,"
                      "ssh,teamspeak,telnet,tftp,tip,udp,unreal,ut2004,vemmi,"
                      "ventrilo,wais,webcal,wyciwyg,z39.50r,z39.50s";
 
char const * user_schemes = nullptr;
 
void check_uri(char const * uri)
{
    tld_result result;
    if(strncasecmp(uri, "mailto:", 7) == 0)
    {
        tld_email_list mail;
        result = mail.parse(uri + 7, 0);
    }
    else
    {
        struct tld_info info;
        char const * s(user_schemes == nullptr ? schemes : user_schemes);
        result = tld_check_uri(uri, &info, s, 0);
 
        if(verbose)
        {
            std::cout << "URI:      " << uri << std::endl;                                      // LCOV_EXCL_LINE
            std::cout << "Category: " << static_cast<int>(info.f_category) << std::endl;        // LCOV_EXCL_LINE
            std::cout << "Status:   " << static_cast<int>(info.f_status) << std::endl;          // LCOV_EXCL_LINE
            if(info.f_country[0] != '\0')                                                       // LCOV_EXCL_LINE
            {
                std::cout << "Country:  " << info.f_country << std::endl;                       // LCOV_EXCL_LINE
            }
            if(info.f_tld != nullptr)
            {
                // port or path may follow this TLD
                //
                char const * e(strchr(info.f_tld, ':'));
                if(e == nullptr)
                {
                    e = strchr(info.f_tld, '/');
                }
                if(e == nullptr)
                {
                    std::cout << "TLD:      " << info.f_tld << std::endl;                       // LCOV_EXCL_LINE
                }
                else
                {
                    std::cout << "TLD:      " << std::string(info.f_tld, e - info.f_tld) << std::endl; // LCOV_EXCL_LINE
                }
                std::cout << "Offset:   " << info.f_offset << std::endl;                        // LCOV_EXCL_LINE
            }
        }
    }
    if(result != TLD_RESULT_SUCCESS)
    {
        std::cerr << "error: URI \"" << uri << "\" is not considered valid." << std::endl;
        ++err_count;
    }
}
 
void list()
{
    for(const char *s(schemes); *s != '\0'; ++s)
    {
        if(*s == ',')
        {
            printf("\n");
        }
        else
        {
            printf("%c", *s);
        }
    }
    printf("\n");
    exit(1);
}
 
void usage()
{
    printf("Usage: validate_tld [-<opts>] <uri> | <email>\n");
    printf("Where <uri> or <email> are URIs starting with a valid scheme.\n");
    printf("The <email> scheme is mailto:.\n");
    printf("Where -<opts> are:\n");
    printf("  -h | --help               print out this help screen\n");
    printf("  -l | --list               print the default list of schemes\n");
    printf("  -s | --schemes <list>     set the list of schemes with user's defined schemes\n");
    printf("                            the list is a comma separate set of scheme names\n");
    printf("  -v | --verbose            request some verbosity of the tool's work\n");
    exit(1);
}
 
int main(int argc, char *argv[])
{
    try
    {
        bool uri(false);
 
        for(int i(1); i < argc; ++i)
        {
            if(argv[i][0] == '-')
            {
                if(strcmp(argv[i], "-h") == 0 || strcmp(argv[i], "--help") == 0)
                {
                    usage();
                    /*NOTREACHED*/
                }
                else if(strcmp(argv[i], "-l") == 0 || strcmp(argv[i], "--list") == 0)
                {
                    list();
                    /*NOTREACHED*/
                }
                else if(strcmp(argv[i], "--version") == 0)
                {
                    printf("%s\n", LIBTLD_VERSION);
                    if(verbose)
                    {
                        printf("libtld v%s\n", tld_version());
                    }
                    exit(1);
                }
                else if(strcmp(argv[i], "-s") == 0 || strcmp(argv[i], "--schemes") == 0)
                {
                    ++i;
                    if(i >= argc)
                    {
                        fprintf(stderr, "error: the --schemes option requires a list of comma separated schemes.\n");
                    }
                    user_schemes = argv[i];
                }
                else if(strcmp(argv[i], "-v") == 0 || strcmp(argv[i], "--verbose") == 0)
                {
                    verbose = 1;
                }
            }
            else
            {
                uri = true;
                check_uri(argv[i]);
            }
        }
 
        if(!uri)
        {
            fprintf(stderr, "error: no URI was specified on the command line.\n");
            ++err_count;
        }
 
        return err_count > 0 ? 1 : 0;
    }
    catch(std::exception const& e) // LCOV_EXCL_LINE
    {
        // an exception occurred, print out the message and exit with an error
        //
        // note that this tool is not expecting any exception
        // because we only access the C interface which does
        // not throw
        //
        std::cerr << "exception: " << e.what() << std::endl; // LCOV_EXCL_LINE
        exit(1); // LCOV_EXCL_LINE
    }
}
 
// vim: ts=4 sw=4 et