/*
 * c++check - c/c++ syntax checking
 * Copyright (C) 2007 Daniel Marjamäki
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/
 */

//---------------------------------------------------------------------------
#include "CheckClass.h"

#include <locale>

#include <string>
#include <sstream>
#include <cstring>
#include <algorithm>


#ifdef __BORLANDC__
#include <ctype.h>
#include <mem.h>
#endif
//---------------------------------------------------------------------------

CheckClass::CheckClass( const Tokenizer *tokenizer, const Settings &settings, ErrorLogger *errorLogger )
{
    _tokenizer = tokenizer;
    _settings = settings;
    _errorLogger = errorLogger;
}

CheckClass::~CheckClass()
{

}

//---------------------------------------------------------------------------

struct CheckClass::VAR *CheckClass::ClassChecking_GetVarList(const TOKEN *tok1)
{
    // Get variable list..
    struct VAR *varlist = NULL;
    unsigned int indentlevel = 0;
    for (const TOKEN *tok = tok1; tok; tok = tok->next)
    {
        if (!tok->next)
            break;

        if (tok->aaaa0() == '{')
            indentlevel++;
        if (tok->aaaa0() == '}')
        {
            if (indentlevel <= 1)
                break;
            indentlevel--;
        }


        if (indentlevel==1 && (strchr(";{}", tok->aaaa0()) || (tok->aaaa0()!=':' && strchr(tok->aaaa(), ':'))))
        {
            const TOKEN *next = tok->next;

            const char *varname = 0;

            // Is it a variable declaration?
            if ( TOKEN::Match(next,"%type% %var% ;") )
            {
                const char *types[] = {"bool", "char", "int", "short", "long", "float", "double", 0};
                for ( int type = 0; types[type]; type++ )
                {
                    if ( strcmp(next->aaaa(), types[type]) == 0)
                    {
                        varname = next->next->aaaa();
                        break;
                    }
                }
            }

            // Pointer?
            else if ( TOKEN::Match(next, "%type% * %var% ;") )
            {
                varname = next->strAt(2);
            }

            if (varname)
            {
                struct VAR *var = new VAR;
                memset(var, 0, sizeof(struct VAR));
                var->name = varname;
                var->init = false;
                var->next = varlist;
                varlist   = var;
            }
        }
    }

    return varlist;
}
//---------------------------------------------------------------------------

const TOKEN * CheckClass::FindClassFunction( const TOKEN *tok, const char classname[], const char funcname[], int &indentlevel )
{
    if ( indentlevel < 0 || tok == NULL )
        return NULL;

    std::ostringstream classPattern;
    classPattern << "class " << classname << " :|{";

    std::ostringstream internalPattern;
    internalPattern << funcname << " (";

    std::ostringstream externalPattern;
    externalPattern << classname << " :: " << funcname << " (";

    for ( ;tok; tok = tok->next )
    {
        if ( indentlevel == 0 && TOKEN::Match(tok, classPattern.str().c_str()) )
        {
            while ( tok && tok->str() != "{" )
                tok = tok->next;
            if ( tok )
                tok = tok->next;
            if ( ! tok )
                break;
            indentlevel = 1;
        }

        if ( tok->str() == "{" )
        {
            // If indentlevel==0 don't go to indentlevel 1. Skip the block.
            if ( indentlevel > 0 )
                ++indentlevel;

            else
            {
                for ( ; tok; tok = tok->next )
                {
                    if ( tok->str() == "{" )
                        ++indentlevel;
                    else if ( tok->str() == "}" )
                    {
                        --indentlevel;
                        if ( indentlevel <= 0 )
                            break;
                    }
                }
                if ( tok == NULL )
                    return NULL;

                continue;
            }
        }

        if ( tok->str() == "}" )
        {
            indentlevel--;
            if ( indentlevel < 0 )
                return NULL;
        }

        if ( indentlevel == 1 )
        {
            // Member function implemented in the class declaration?
            if (!TOKEN::Match(tok,"~") && TOKEN::Match(tok->next, internalPattern.str().c_str()))
            {
                const TOKEN *tok2 = tok;
                while ( tok2 && tok2->str() != "{" && tok2->str() != ";" )
                    tok2 = tok2->next;
                if ( tok2 && tok2->str() == "{" )
                    return tok->next;
            }
        }

        else if ( indentlevel == 0 && TOKEN::Match(tok, externalPattern.str().c_str()) )
        {
            return tok;
        }
    }

    // Not found
    return NULL;
}
//---------------------------------------------------------------------------

void CheckClass::InitVar(struct VAR *varlist, const char varname[])
{
    for (struct VAR *var = varlist; var; var = var->next)
    {
        if ( strcmp(var->name, varname) == 0 )
        {
            var->init = true;
            break;
        }
    }
}
//---------------------------------------------------------------------------

void CheckClass::ClassChecking_VarList_Initialize(const TOKEN *tok1, const TOKEN *ftok, struct VAR *varlist, const char classname[], std::list<std::string> &callstack)
{
    bool Assign = false;
    unsigned int indentlevel = 0;

    for (; ftok; ftok = ftok->next)
    {
        if (!ftok->next)
            break;

        // Class constructor.. initializing variables like this
        // clKalle::clKalle() : var(value) { }
        if (indentlevel==0)
        {
            if (Assign && TOKEN::Match(ftok, "%var% ("))
            {
                InitVar( varlist, ftok->aaaa() );
            }

            Assign |= (ftok->aaaa0() == ':');
        }


        if (ftok->aaaa0() == '{')
        {
            indentlevel++;
            Assign = false;
        }

        if (ftok->aaaa0() == '}')
        {
            if (indentlevel <= 1)
                break;
            indentlevel--;
        }

        if ( indentlevel < 1 )
            continue;

        // Before a new statement there is "[{};)=]" or "else"
        if ( ! TOKEN::Match(ftok, "[{};)=]") && ! TOKEN::Match(ftok, "else") )
            continue;

        // Using the operator= function to initialize all variables..
        if ( TOKEN::Match(ftok->next, "* this = ") )
        {
            for (struct VAR *var = varlist; var; var = var->next)
                var->init = true;
            break;
        }

        if (!TOKEN::Match(ftok->next, "%var%") && !TOKEN::Match(ftok->next, "this . %var%"))
            continue;

        // Goto the first token in this statement..
        ftok = ftok->next;

        // Skip "this->"
        if ( TOKEN::Match(ftok, "this .") )
            ftok = ftok->tokAt(2);

        // Clearing all variables..
        if (TOKEN::Match(ftok,"memset ( this ,"))
        {
            for (struct VAR *var = varlist; var; var = var->next)
                var->init = true;
            break;
        }

        // Calling member function?
        else if (TOKEN::Match(ftok, "%var% ("))
        {
            // No recursive calls!
            if ( std::find(callstack.begin(),callstack.end(),ftok->aaaa()) == callstack.end() )
            {
                callstack.push_back( ftok->aaaa() );
                int i = 0;
                const TOKEN *ftok2 = FindClassFunction( tok1, classname, ftok->aaaa(), i );
                ClassChecking_VarList_Initialize(tok1, ftok2, varlist, classname, callstack);
            }
        }

        // Assignment of member variable?
        else if (TOKEN::Match(ftok, "%var% ="))
        {
            InitVar( varlist, ftok->aaaa() );
        }

        // The functions 'clear' and 'Clear' are supposed to initialize variable.
        if (TOKEN::Match(ftok,"%var% . clear (") || TOKEN::Match(ftok,"%var% . Clear ("))
        {
            InitVar( varlist, ftok->aaaa() );
        }
    }
}






//---------------------------------------------------------------------------
// ClassCheck: Check that all class constructors are ok.
//---------------------------------------------------------------------------

void CheckClass::CheckConstructors()
{
    const char pattern_class[] = "class %var% {";

    // Locate class
    const TOKEN *tok1 = TOKEN::findmatch( _tokenizer->tokens(), pattern_class );
    while (tok1)
    {
        const char *className[2];
        className[0] = tok1->strAt( 1 );
        className[1] = 0;

        // TODO: handling of private constructors should be improved.
        bool hasPrivateConstructor = false;
        {
            int indentlevel = 0;
            bool isPrivate = true;
            for ( const TOKEN *tok = tok1; tok; tok = tok->next )
            {
                // Indentation
                if ( tok->str() == "{" )
                    ++indentlevel;

                else if ( tok->str() == "}" )
                {
                    --indentlevel;
                    if (indentlevel <= 0)
                        break;
                }

                // Parse class contents (indentlevel == 1)..
                if ( indentlevel == 1 )
                {
                    // What section are we in.. private/non-private
                    if ( tok->str() == "private:" )
                        isPrivate = true;
                    else if ( tok->str() == "protected:" || tok->str() == "public:" )
                        isPrivate = false;

                    // Is there a private constructor?
                    else if ( isPrivate && TOKEN::Match(tok, "%var1% (", className) )
                    {
                        hasPrivateConstructor = true;
                        break;
                    }
                }
            }
        }

        if ( hasPrivateConstructor )
        {
            // TODO: Handle private constructors.
            // Right now to avoid false positives I just bail out
            tok1 = TOKEN::findmatch( tok1->next, pattern_class );
            continue;
        }

        // Are there a class constructor?
        const TOKEN *constructor_token = TOKEN::findmatch( tok1, "%any% %var1% (", className );
        while ( TOKEN::Match( constructor_token, "~" ) )
            constructor_token = TOKEN::findmatch( constructor_token->next, "%any% %var1% (", className );

        // There are no constructor.
        if ( ! constructor_token )
        {
            // If "--style" has been given, give a warning
            if ( _settings._checkCodingStyle )
            {
                // If the class has member variables there should be an constructor
                struct VAR *varlist = ClassChecking_GetVarList(tok1);
                if ( varlist )
                {
                    std::ostringstream ostr;
                    ostr << _tokenizer->fileLine(tok1);
                    ostr << " The class '" << className[0] << "' has no constructor";
                    _errorLogger->reportErr(ostr.str());
                }
                // Delete the varlist..
                while (varlist)
                {
                    struct VAR *nextvar = varlist->next;
                    delete varlist;
                    varlist = nextvar;
                }
            }

            tok1 = TOKEN::findmatch( tok1->next, pattern_class );
            continue;
        }

        // Check that all member variables are initialized..
        struct VAR *varlist = ClassChecking_GetVarList(tok1);

        // Check constructors
        CheckConstructors( tok1, varlist, className[0] );

        // Check assignment operators
        CheckConstructors( tok1, varlist, "operator =" );

        // Delete the varlist..
        while (varlist)
        {
            struct VAR *nextvar = varlist->next;
            delete varlist;
            varlist = nextvar;
        }

        tok1 = TOKEN::findmatch( tok1->next, pattern_class );
    }
}

void CheckClass::CheckConstructors(const TOKEN *tok1, struct VAR *varlist, const char funcname[])
{
    const char * const className = tok1->strAt(1);

    int indentlevel = 0;
    const TOKEN *constructor_token = FindClassFunction( tok1, className, funcname, indentlevel );
    std::list<std::string> callstack;
    ClassChecking_VarList_Initialize(tok1, constructor_token, varlist, funcname, callstack);
    while ( constructor_token )
    {
        // Check if any variables are uninitialized
        for (struct VAR *var = varlist; var; var = var->next)
        {
            if ( var->init )
                continue;

            // Is it a static member variable?
            std::ostringstream pattern;
            pattern << className << "::" << var->name << "=";
            if (TOKEN::findmatch(_tokenizer->tokens(), pattern.str().c_str()))
                continue;

            // It's non-static and it's not initialized => error
            std::ostringstream ostr;
            ostr << _tokenizer->fileLine(constructor_token);
            ostr << " Uninitialized member variable '" << className << "::" << var->name << "'";
            _errorLogger->reportErr(ostr.str());
        }

        for ( struct VAR *var = varlist; var; var = var->next )
            var->init = false;

        constructor_token = FindClassFunction( constructor_token->next, className, funcname, indentlevel );
        callstack.clear();
        ClassChecking_VarList_Initialize(tok1, constructor_token, varlist, className, callstack);
    }
}


//---------------------------------------------------------------------------
// ClassCheck: Unused private functions
//---------------------------------------------------------------------------

void CheckClass::CheckUnusedPrivateFunctions()
{
    // Locate some class
    const char *pattern_class[] = {"class","","{",NULL};
    for (const TOKEN *tok1 = TOKEN::findtoken(_tokenizer->tokens(), pattern_class); tok1; tok1 = TOKEN::findtoken(tok1->next, pattern_class))
    {
        const char *classname = tok1->next->aaaa();

        // The class implementation must be available..
        const char *pattern_classconstructor[] = {"","::","",NULL};
        pattern_classconstructor[0] = classname;
        pattern_classconstructor[2] = classname;
        if (!TOKEN::findtoken(_tokenizer->tokens(),pattern_classconstructor))
            continue;

        // Get private functions..
        std::list<std::string> FuncList;
        FuncList.clear();
        bool priv = false;
        unsigned int indent_level = 0;
        for (const TOKEN *tok = tok1; tok; tok = tok->next)
        {
            if (TOKEN::Match(tok,"friend %var%"))
            {
                // Todo: Handle friend classes
                FuncList.clear();
                break;
            }

            if (tok->aaaa0() == '{')
                indent_level++;
            if (tok->aaaa0() == '}')
            {
                if (indent_level <= 1)
                    break;
                indent_level--;
            }
            if (strcmp(tok->aaaa(),"};") == 0)
                break;
            if (strcmp(tok->aaaa(),"private:") == 0)
                priv = true;
            else if (strcmp(tok->aaaa(),"public:") == 0)
                priv = false;
            else if (strcmp(tok->aaaa(),"protected:") == 0)
                priv = false;
            else if (priv && indent_level == 1)
            {
                if ( TOKEN::Match(tok, "typedef %type% (") )
                    tok = tok->tokAt(2);

                if (TOKEN::Match(tok, "%var% (") &&
                    !TOKEN::Match(tok,classname))
                {
                    FuncList.push_back(tok->aaaa());
                }
            }
        }

        // Check that all private functions are used..
        const char *pattern_function[] = {"","::",NULL};
        pattern_function[0] = classname;
        bool HasFuncImpl = false;
        const TOKEN *ftok = _tokenizer->tokens();
        while (ftok)
        {
            ftok = TOKEN::findtoken(ftok,pattern_function);
            int numpar = 0;
            while (ftok && ftok->aaaa0()!=';' && ftok->aaaa0()!='{')
            {
                if (ftok->aaaa0() == '(')
                    numpar++;
                else if (ftok->aaaa0() == ')')
                    numpar--;
                ftok = ftok->next;
            }

            if (!ftok)
                break;

            if (ftok->aaaa0() != ';' && numpar == 0)
            {
                HasFuncImpl = true;

                indent_level = 0;
                while (ftok)
                {
                    if (ftok->aaaa0() == '{')
                        indent_level++;
                    if (ftok->aaaa0() == '}')
                    {
                        if (indent_level<=1)
                            break;
                        indent_level--;
                    }
                    if (ftok->next && ftok->next->aaaa0() == '(')
                        FuncList.remove(ftok->aaaa());
                    ftok = ftok->next;
                }
            }

            if (ftok)
                ftok = ftok->next;
        }

        while (HasFuncImpl && !FuncList.empty())
        {
            bool fp = false;

            // Final check; check if the function pointer is used somewhere..
            const char *_pattern[] = {"=","",NULL};
            _pattern[1] = FuncList.front().c_str();
            fp |= (TOKEN::findtoken(_tokenizer->tokens(), _pattern) != NULL);
            _pattern[0] = "return";
            fp |= (TOKEN::findtoken(_tokenizer->tokens(), _pattern) != NULL);
            _pattern[0] = "(";
            fp |= (TOKEN::findtoken(_tokenizer->tokens(), _pattern) != NULL);
            _pattern[0] = ")";
            fp |= (TOKEN::findtoken(_tokenizer->tokens(), _pattern) != NULL);
            _pattern[0] = ",";
            fp |= (TOKEN::findtoken(_tokenizer->tokens(), _pattern) != NULL);

            if (!fp)
            {
                std::ostringstream ostr;
                ostr << "Class '" << classname << "', unused private function: '" << FuncList.front() << "'";
                _errorLogger->reportErr(ostr.str());
            }
            FuncList.pop_front();
        }
    }
}

//---------------------------------------------------------------------------
// ClassCheck: Check that memset is not used on classes
//---------------------------------------------------------------------------

void CheckClass::CheckMemset()
{
    // Locate all 'memset' tokens..
    for (const TOKEN *tok = _tokenizer->tokens(); tok; tok = tok->next)
    {
        if (!TOKEN::Match(tok,"memset") && !TOKEN::Match(tok,"memcpy") && !TOKEN::Match(tok,"memmove"))
            continue;

        // Todo: Handle memcpy and memmove
        const char *type = NULL;
        if (TOKEN::Match(tok, "memset ( %var% , %num% , sizeof ( %type% ) )"))
            type = tok->strAt(8);
        else if (TOKEN::Match(tok, "memset ( & %var% , %num% , sizeof ( %type% ) )"))
            type = tok->strAt(9);
        else if (TOKEN::Match(tok, "memset ( %var% , %num% , sizeof ( struct %type% ) )"))
            type = tok->strAt(9);
        else if (TOKEN::Match(tok, "memset ( & %var% , %num% , sizeof ( struct %type% ) )"))
            type = tok->strAt(10);
        else if (TOKEN::Match(tok, "%type% ( %var% , %var% , sizeof ( %type% ) )"))
            type = tok->strAt(8);

        // No type defined => The tokens didn't match
        if (!(type && type[0]))
            continue;

        // Warn if type is a class..
        const char *pattern1[] = {"class","",NULL};
        pattern1[1] = type;
        if (TOKEN::findtoken(_tokenizer->tokens(),pattern1))
        {
            std::ostringstream ostr;
            ostr << _tokenizer->fileLine(tok) << ": Using '" << tok->aaaa() << "' on class.";
            _errorLogger->reportErr(ostr.str());
            continue;
        }

        // Warn if type is a struct that contains any std::*
        const char *pattern2[] = {"struct","","{",NULL};
        pattern2[1] = type;
        for (const TOKEN *tstruct = TOKEN::findtoken(_tokenizer->tokens(), pattern2); tstruct; tstruct = tstruct->next)
        {
            if (tstruct->aaaa0() == '}')
                break;

            if (TOKEN::Match(tstruct, "std :: %type% %var% ;"))
            {
                std::ostringstream ostr;
                ostr << _tokenizer->fileLine(tok) << ": Using '" << tok->aaaa() << "' on struct that contains a 'std::" << tstruct->strAt(2) << "'";
                _errorLogger->reportErr(ostr.str());
                break;
            }
        }
    }
}
//---------------------------------------------------------------------------




//---------------------------------------------------------------------------
// ClassCheck: "void operator=("
//---------------------------------------------------------------------------

void CheckClass::CheckOperatorEq1()
{
    if (const TOKEN *tok = TOKEN::findmatch(_tokenizer->tokens(), "void operator = ("))
    {
        std::ostringstream ostr;
        ostr << _tokenizer->fileLine(tok) << ": 'operator=' should return something";
        _errorLogger->reportErr(ostr.str());
    }
}
//---------------------------------------------------------------------------




//---------------------------------------------------------------------------
// A destructor in a base class should be virtual
//---------------------------------------------------------------------------

void CheckClass::virtualDestructor()
{
	const char pattern_classdecl[] = "class %var% : %var%";

    const TOKEN *derived = TOKEN::findmatch(_tokenizer->tokens(), pattern_classdecl);
    while (derived)
    {
        // Iterate through each base class...
		derived = derived->tokAt(3);
        while ( TOKEN::Match(derived, "%var%") )
        {
            bool isPublic = TOKEN::Match(derived, "public");

            // What kind of inheritance is it.. public|protected|private
            if ( TOKEN::Match( derived, "public|protected|private" ) )
                derived = derived->next;

            // Name of base class..
            const char *baseName[2];
            baseName[0] = derived->strAt(0);
            baseName[1] = 0;

            // Update derived so it's ready for the next loop.
            derived = derived->next;
            if ( TOKEN::Match(derived, ",") )
                derived = derived->next;

            // If not public inheritance, skip checking of this base class..
            if ( ! isPublic )
                continue;

            // Find the destructor declaration for the base class.
            const TOKEN *base = TOKEN::findmatch(_tokenizer->tokens(), "%any% ~ %var1% (", baseName);
            while (TOKEN::Match(base, "::"))
                base = TOKEN::findmatch(base->next, "%any% ~ %var1% (", baseName);

            // Check that there is a destructor..
            if ( ! base )
            {
                // Is the class declaration available?
                base = TOKEN::findmatch(_tokenizer->tokens(), "class %var1% :|{", baseName);
                if ( base )
                {
                    std::ostringstream errmsg;
                    errmsg << _tokenizer->fileLine(base) << ": Base class " << baseName[0] << " doesn't have a virtual destructor";
                    _errorLogger->reportErr(errmsg.str());
                }
            }

            // There is a destructor. Check that it's virtual..
            else if ( ! TOKEN::Match(base, "virtual") )
            {
                std::ostringstream errmsg;
                errmsg << _tokenizer->fileLine(base) << ": The destructor for the base class " << baseName[0] << " is not virtual";
                _errorLogger->reportErr(errmsg.str());
            }
        }

		// Goto next class
        derived = TOKEN::findmatch(derived, pattern_classdecl);
    }
}
//---------------------------------------------------------------------------