log/compiler/message.cc

Go to the documentation of this file.

// Copyright (C) 2011-2016 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
#include <config.h>
 
#include <cctype>
#include <cstddef>
#include <fstream>
#include <iostream>
#include <string>
#include <vector>
 
#include <errno.h>
#include <getopt.h>
#include <string.h>
#include <time.h>
#include <unistd.h>
 
#include <exceptions/exceptions.h>
 
#include <util/filename.h>
#include <util/strutil.h>
 
#include <log/log_messages.h>
#include <log/message_dictionary.h>
#include <log/message_exception.h>
#include <log/message_reader.h>
 
#include <log/logger.h>
 
#include <boost/foreach.hpp>
 
using namespace std;
using namespace isc::log;
using namespace isc::util;
 
 
 
 
void
version() {
    cout << VERSION << "\n";
}
 
 
void
usage() {
    cout <<
        "Usage: kea-msg-compiler [-h] [-v] [-d dir] <message-file>\n" <<
        "\n" <<
        "-h       Print this message and exit\n" <<
        "-v       Print the program version and exit\n" <<
        "-d <dir> Place output files in given directory\n" <<
        "\n" <<
        "<message-file> is the name of the input message file.\n";
}
 
 
 
string
currentTime() {
 
    // Get a text representation of the current time.
    time_t curtime;
    time(&curtime);
    struct tm* timeinfo;
    timeinfo = localtime(&curtime);
 
    char buffer[80];
    strftime(buffer, 80, "%a %b %d %Y %H:%M", timeinfo);
 
    return (std::string(buffer));
}
 
 
 
string
sentinel(Filename& file) {
 
    string name = file.name();
    string ext = file.extension();
    string sentinel_text = name + "_" + ext.substr(1);
    isc::util::str::uppercase(sentinel_text);
    return (sentinel_text);
}
 
 
 
string
quoteString(const string& instring) {
 
    // Create the output string and reserve the space needed to hold the input
    // string. (Most input strings will not contain quotes, so this single
    // reservation should be all that is needed.)
    string outstring;
    outstring.reserve(instring.size());
 
    // Iterate through the input string, preceding quotes with a slash.
    for (size_t i = 0; i < instring.size(); ++i) {
        if (instring[i] == '"') {
            outstring += '\\';
        }
        outstring += instring[i];
    }
 
    return (outstring);
}
 
 
 
vector<string>
sortedIdentifiers(MessageDictionary& dictionary) {
    vector<string> ident;
 
    for (MessageDictionary::const_iterator i = dictionary.begin();
         i != dictionary.end(); ++i) {
        ident.push_back(i->first);
    }
    sort(ident.begin(), ident.end());
 
    return (ident);
}
 
 
 
vector<string>
splitNamespace(string ns) {
 
    // Namespaces components are separated by double colon characters -
    // convert to single colons.
    size_t dcolon;
    while ((dcolon = ns.find("::")) != string::npos) {
        ns.replace(dcolon, 2, ":");
    }
 
    // ... and return the vector of namespace components split on the single
    // colon.
    return (isc::util::str::tokens(ns, ":"));
}
 
 
void
writeOpeningNamespace(ostream& output, const vector<string>& ns) {
    if (!ns.empty()) {
 
        // Output namespaces in correct order
        for (vector<string>::size_type i = 0; i < ns.size(); ++i) {
            output << "namespace " << ns[i] << " {\n";
        }
        output << "\n";
    }
}
 
 
void
writeClosingNamespace(ostream& output, const vector<string>& ns) {
    if (!ns.empty()) {
        for (int i = ns.size() - 1; i >= 0; --i) {
            output << "} // namespace " << ns[i] << "\n";
        }
        output << "\n";
    }
}
 
 
void
writeHeaderFile(const string& file, const vector<string>& ns_components,
                MessageDictionary& dictionary, const char* output_directory)
{
    Filename message_file(file);
    Filename header_file(Filename(message_file.name()).useAsDefault(".h"));
    if (output_directory != NULL) {
        header_file.setDirectory(output_directory);
    }
 
    // Text to use as the sentinels.
    string sentinel_text = sentinel(header_file);
 
    // zero out the errno to be safe
    errno = 0;
 
    // Open the output file for writing
    ofstream hfile(header_file.fullName().c_str());
 
    if (hfile.fail()) {
        isc_throw_4(MessageException, "Failed to open output file",
                    LOG_OPEN_OUTPUT_FAIL, header_file.fullName(),
                    strerror(errno), 0);
    }
 
    // Write the header preamble.  If there is an error, we'll pick it up
    // after the last write.
 
    hfile <<
        "// File created from " << message_file.fullName() << " on " <<
            currentTime() << "\n" <<
         "\n" <<
         "#ifndef " << sentinel_text << "\n" <<
         "#define "  << sentinel_text << "\n" <<
         "\n" <<
         "#include <log/message_types.h>\n" <<
         "\n";
 
    // Write the message identifiers, bounded by a namespace declaration
    writeOpeningNamespace(hfile, ns_components);
 
    vector<string> idents = sortedIdentifiers(dictionary);
    for (vector<string>::const_iterator j = idents.begin();
        j != idents.end(); ++j) {
        hfile << "extern const isc::log::MessageID " << *j << ";\n";
    }
    hfile << "\n";
 
    writeClosingNamespace(hfile, ns_components);
 
    // ... and finally the postamble
    hfile << "#endif // " << sentinel_text << "\n";
 
    // Report errors (if any) and exit
    if (hfile.fail()) {
        isc_throw_4(MessageException, "Error writing to output file",
                    LOG_WRITE_ERROR, header_file.fullName(), strerror(errno),
                    0);
    }
 
    hfile.close();
}
 
 
 
char
replaceNonAlphaNum(char c) {
    return (isalnum(c) ? c : '_');
}
 
 
void
writeProgramFile(const string& file, const vector<string>& ns_components,
                 MessageDictionary& dictionary,
                 const char* output_directory)
{
    Filename message_file(file);
    Filename program_file(Filename(message_file.name()).useAsDefault(".cc"));
    if (output_directory) {
        program_file.setDirectory(output_directory);
    }
 
    // zero out the errno to be safe
    errno = 0;
 
    // Open the output file for writing
    ofstream ccfile(program_file.fullName().c_str());
 
    if (ccfile.fail()) {
        isc_throw_4(MessageException, "Error opening output file",
                    LOG_OPEN_OUTPUT_FAIL, program_file.fullName(),
                    strerror(errno), 0);
    }
 
    // Write the preamble.  If there is an error, we'll pick it up after
    // the last write.
 
    ccfile <<
        "// File created from " << message_file.fullName() << " on " <<
            currentTime() << "\n" <<
         "\n" <<
         "#include <cstddef>\n" <<
         "#include <log/message_types.h>\n" <<
         "#include <log/message_initializer.h>\n" <<
         "\n";
 
    // Declare the message symbols themselves.
 
    writeOpeningNamespace(ccfile, ns_components);
 
    vector<string> idents = sortedIdentifiers(dictionary);
    for (vector<string>::const_iterator j = idents.begin();
        j != idents.end(); ++j) {
        ccfile << "extern const isc::log::MessageID " << *j <<
            " = \"" << *j << "\";\n";
    }
    ccfile << "\n";
 
    writeClosingNamespace(ccfile, ns_components);
 
    // Now the code for the message initialization.
 
    ccfile <<
         "namespace {\n" <<
         "\n" <<
         "const char* values[] = {\n";
 
    // Output the identifiers and the associated text.
    idents = sortedIdentifiers(dictionary);
    for (vector<string>::const_iterator i = idents.begin();
        i != idents.end(); ++i) {
            ccfile << "    \"" << *i << "\", \"" <<
                quoteString(dictionary.getText(*i)) << "\",\n";
    }
 
 
    // ... and the postamble
    ccfile <<
        "    NULL\n" <<
        "};\n" <<
        "\n" <<
        "const isc::log::MessageInitializer initializer(values);\n" <<
        "\n" <<
        "} // Anonymous namespace\n" <<
        "\n";
 
    // Report errors (if any) and exit
    if (ccfile.fail()) {
        isc_throw_4(MessageException, "Error writing to output file",
                    LOG_WRITE_ERROR, program_file.fullName(), strerror(errno),
                    0);
    }
 
    ccfile.close();
}
 
 
 
void
errorDuplicates(MessageReader& reader) {
 
    // Get the duplicates (the overflow) and, if present, sort them into some
    // order and remove those which occur more than once (which mean that they
    // occur more than twice in the input file).
    MessageReader::MessageIDCollection duplicates = reader.getNotAdded();
    if (!duplicates.empty()) {
        cout << "Error: the following duplicate IDs were found:\n";
 
        sort(duplicates.begin(), duplicates.end());
        MessageReader::MessageIDCollection::iterator new_end =
            unique(duplicates.begin(), duplicates.end());
        for (MessageReader::MessageIDCollection::iterator i = duplicates.begin();
            i != new_end; ++i) {
            cout << "    " << *i << "\n";
        }
        exit(1);
    }
}
 
 
 
int
main(int argc, char* argv[]) {
 
    const char* soptions = "hvpd:";               // Short options
 
    optind = 1;             // Ensure we start a new scan
    int  opt;               // Value of the option
 
    const char *output_directory = NULL;
 
    while ((opt = getopt(argc, argv, soptions)) != -1) {
        switch (opt) {
            case 'd':
                output_directory = optarg;
                break;
 
            case 'h':
                usage();
                return (0);
 
            case 'v':
                version();
                return (0);
 
            default:
                // A message will have already been output about the error.
                return (1);
        }
    }
 
    // Do we have the message file?
    if (optind < (argc - 1)) {
        cout << "Error: excess arguments in command line\n";
        usage();
        return (1);
    } else if (optind >= argc) {
        cout << "Error: missing message file\n";
        usage();
        return (1);
    }
    string message_file = argv[optind];
 
    try {
        // Have identified the file, so process it.  First create a local
        // dictionary into which the data will be put.
        MessageDictionary dictionary;
 
        // Read the data into it.
        MessageReader reader(&dictionary);
        reader.readFile(message_file);
 
        // Error (and quit) if there are of any duplicates encountered.
        errorDuplicates(reader);
 
        // Get the namespace into which the message definitions will be put and
        // split it into components.
        vector<string> ns_components =
            splitNamespace(reader.getNamespace());
 
        // Write the header file.
        writeHeaderFile(message_file, ns_components, dictionary,
                        output_directory);
 
        // Write the file that defines the message symbols and text
        writeProgramFile(message_file, ns_components, dictionary,
                         output_directory);
 
    }
    catch (const MessageException& e) {
        // Create an error message from the ID and the text
        const MessageDictionaryPtr& global = MessageDictionary::globalDictionary();
        string text = e.id();
        text += ", ";
        text += global->getText(e.id());
        // Format with arguments
        vector<string> args(e.arguments());
        for (size_t i(0); i < args.size(); ++ i) {
            try {
                replacePlaceholder(&text, args[i], i + 1);
            } catch (...) {
                // Error in error handling: nothing right to do...
            }
        }
 
        cerr << text << "\n";
 
        return (1);
    }
 
    return (0);
 
}