/* IBM_PROLOG_BEGIN_TAG                                                   */
/* This is an automatically generated prolog.                             */
/*                                                                        */
/* $Source: src/usr/testcore/kernel/segmenttest.H $                       */
/*                                                                        */
/* IBM CONFIDENTIAL                                                       */
/*                                                                        */
/* COPYRIGHT International Business Machines Corp. 2011,2012              */
/*                                                                        */
/* p1                                                                     */
/*                                                                        */
/* Object Code Only (OCO) source materials                                */
/* Licensed Internal Code Source Materials                                */
/* IBM HostBoot Licensed Internal Code                                    */
/*                                                                        */
/* The source code for this program is not published or otherwise         */
/* divested of its trade secrets, irrespective of what has been           */
/* deposited with the U.S. Copyright Office.                              */
/*                                                                        */
/* Origin: 30                                                             */
/*                                                                        */
/* IBM_PROLOG_END_TAG                                                     */
#ifndef __SEGMENTTEST_H
#define __SEGMENTTEST_H
/**
 *  @file segmenttest.H
 *
 *  @brief Test cases for the virtual memory segments
*/
#include <cxxtest/TestSuite.H>
#include <arch/ppc.H>
#include <sys/time.h>
#include <sys/task.h>
#include <sys/mmio.h>
#include <usr/vmmconst.h>

#include <list>

class segmenttest: public CxxTest::TestSuite
{
    public:

        void testDevSeg()
        {
            int rc = 0;
            uint64_t ra = 2*1024*1024;

            printkd("Map Device @ ra = 0x%lX using dev_map\n",ra);
            uint64_t* virtAddrDEV = static_cast<uint64_t*>
                     (mmio_dev_map(reinterpret_cast<void*>(ra), THIRTYTWO_GB));
            if (virtAddrDEV == NULL)
            {
                TS_FAIL("Failed to map using mmio_dev_map\n");
            }

            printkd("Unmap Device @ va = %p using dev_unmap\n",virtAddrDEV);
            rc = mmio_dev_unmap(reinterpret_cast<void*>(virtAddrDEV));
            if (rc != 0)
            {
                TS_FAIL("Failed to unmap using mmio_dev_unmap\n");
            }
        }

        void testSegBlock()
        {
            int rc = -1;
            msg_q_t mq = msg_q_create(); //Create empty message queue
            uint64_t va = 0xC800000000; //800GB
            uint64_t size = 0x100000; //1MB
            printkd("Allocate 1MB block with empty msgq @ vaddr = 800GB within base segment\n");
            rc = mm_alloc_block(mq,reinterpret_cast<void*>(va),size);
            if (rc != 0)
            {
                TS_FAIL("Failed to create BaseSegment block\n");
            }
            msg_q_t mq2 = msg_q_create(); //Create empty message queue
            uint64_t va2 = 0xE100000000; //900GB
            uint64_t size2 = 0x100000; //1MB
            printkd("Allocate 1MB block with empty msgq @ vaddr = 900GB within base segment\n");
            rc = mm_alloc_block(mq2,reinterpret_cast<void*>(va2),size2);
            if (rc != 0)
            {
                TS_FAIL("Failed to create BaseSegment block\n");
            }
        }

        // Verify we can allocate a block from the second device segment.
        void testManyDeviceBlocks()
        {
            std::list<void*> blocks;

            void* block = NULL;
            while (reinterpret_cast<uint64_t>(block)
                    < (VMM_VADDR_DEVICE_SEGMENT_FIRST + VMM_SEGMENT_SIZE))
            {
                block = mmio_dev_map(reinterpret_cast<void*>(10 * TERABYTE),
                                     THIRTYTWO_GB);

                if (block == NULL)
                {
                    TS_FAIL("Unable to allocate device block.");
                    break;
                }
                blocks.push_back(block);
            }

            while(!blocks.empty())
            {
                mmio_dev_unmap(blocks.front());
                blocks.pop_front();
            }
        }

       // Create 2 blocks at different specified address's and sizes and direct
       // map this.
        void testSegLinearBlock()
        {
            int rc = -1;
            uint64_t phys = 0;
            uint64_t addr = 0x8000000 - 0x5000; //128M- 16K
            uint64_t size = 0x1000; //4K

            rc = mm_linear_map(reinterpret_cast<void*>(addr),size);
            if (rc != 0)
            {
                TS_FAIL("Failed to create BaseSegment block\n");
            }

            phys = mm_virt_to_phys( reinterpret_cast<void*>(addr) );

            if( phys != addr )
            {
                TS_TRACE( "Block> virt=%lX, phys=%lX", &addr, phys );
                TS_FAIL("Unexpected Physical Address for block.");
            }


            uint64_t addr2 = 0x8000000 - 0x3000; //128M- 12k
            uint64_t size2 = 0x3000; //12K

            rc = mm_linear_map(reinterpret_cast<void*>(addr2),size2);

            if (rc != 0)
            {
                TS_FAIL("Failed to create BaseSegment block\n");
            }

            phys = mm_virt_to_phys( reinterpret_cast<void*>(addr2) );

            if( phys != addr2 )
            {
                TS_TRACE( "Block> virt=%lX, phys=%lX", &addr2, phys );
                TS_FAIL("Unexpected Physical Address for block.");
                
            }

        }

};

#endif