cpptest.cpp

一个实时的驱动支持库

/*
    Test driver for a C++ run-time library that supports C++ exception
    handling and run-time type information.
    Copyright (C) 2003, 2004 Bo Brant閚.
*/

extern "C" {
#include <ntddk.h>
}
#include <libcpp.h>
#include <typeinfo.h>
#include <winexcpt.h>

class ClassA{virtual void f(){}};
class ClassAA : public ClassA{};
class ClassB{};

class Object {
public:
    Object(char* s) : m_name(s)                 { DbgPrint("Object %s was constructed\n", m_name); }
    Object(const Object& o) : m_name(o.m_name)  { DbgPrint("Object %s was copied\n", m_name); }
    ~Object()                                   { DbgPrint("Object %s was deleted\n", m_name); }
    char* GetName()                             { return m_name; }
private:
    char* m_name;
};

Object g("global");

extern "C"
NTSTATUS
DriverEntry(
    IN PDRIVER_OBJECT   DriverObject,
    IN PUNICODE_STRING  RegistryPath
    )
{
    libcpp_init();

    DbgPrint("Entering the driver\n");

    DbgPrint("\n");

    //
    // Test run-time type information.
    //

    const type_info& gt = typeid(g);

    DbgPrint("The name of the global object g is: %s\n", gt.name());

    ClassA a1, a2;
    ClassAA aa;
    ClassB b;

    void* pv = dynamic_cast<void*>(&a1);

    ClassA* ap = &aa;
    ClassAA* aap = dynamic_cast<ClassAA*>(ap);

    DbgPrint("The type name of the object ap points to is: %s\n", typeid(*ap).name());

    if (typeid(a1) == typeid(a2))
    {
        DbgPrint("%s and %s is of the same type\n", typeid(a1).name(), typeid(a2).name());
    }

    if (typeid(a1) != typeid(b))
    {
        DbgPrint("%s and %s is not of the same type\n", typeid(a1).name(), typeid(b).name());
    }

    DbgPrint("\n");

    //
    // Test exception handling.
    //

    try {
        Object l1("local 1");
        DbgPrint("Throwing an exception\n");
        throw Object("an exception");
    }
    catch (Object& e) {
        DbgPrint("Catching %s\n", e.GetName());
    }

    DbgPrint("\n");
#if 0
    //
    // Test translation between SEH and C++ exception handling.
    //

    WinException::install();

    try {
        Object l2("local 2");
        DbgPrint("Dereference a NULL pointer inside a try block\n");
        volatile int *x = 0, y = *x; // dereference NULL pointer
    }
    catch (WinException& e) {
        DbgPrint("Caught WinException %#x\n", e.ExceptionCode());
    }
    catch (...) {
        DbgPrint("Caught exception in catch-all clause\n");
    }

    DbgPrint("\n");
#endif
    //
    // Test re-throwing an exception.
    //

    try {
        Object l3("local 3");
        try {
            Object l4("local 4");
            DbgPrint("Throwing an exception to re-throw\n");
            throw Object("an exception to re-throw");
        }
        catch (Object& e) {
            DbgPrint("Catching %s the first time\n", e.GetName());
            throw;
        }
    }
    catch (Object& e) {
        DbgPrint("Catching %s the second time\n", e.GetName());
    }

    DbgPrint("\n");
#if 0
    //
    // Test exception handling at DISPATCH_LEVEL.
    //

    KIRQL SavedIrql;

    DbgPrint("Raising IRQL to DISPATCH_LEVEL\n");

    KeRaiseIrql(DISPATCH_LEVEL, &SavedIrql);

    try {
        Object l5("local 5");
        DbgPrint("Throwing an exception at DISPATCH_LEVEL\n");
        throw Object("an exception at DISPATCH_LEVEL");
    }
    catch (Object& e) {
        DbgPrint("Catching %s\n", e.GetName());
    }

    DbgPrint("Lower IRQL from DISPATCH_LEVEL\n");

    KeLowerIrql(SavedIrql);

    DbgPrint("\n");
#endif
    //
    // Done testing so we don't load the driver.
    //

    DbgPrint("Exiting the driver\n");

    libcpp_exit();

    return STATUS_UNSUCCESSFUL;
}