/*----------------------------------------------------------------------------*/
/* smadmin.c                                                                  */
/*----------------------------------------------------------------------------*/

/*----------------------------------------------------------------------------*/
/* This file is part of LibShMem.                                             */
/*                                                                            */
/*    LibShMem 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.                                     */
/*                                                                            */
/*    LibShMem 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 LibShMem. If not, see <https://www.gnu.org/licenses/>.       */
/*----------------------------------------------------------------------------*/



/*----------------------------------------------------------------------------*/
/* Includes                                                                   */
/*----------------------------------------------------------------------------*/

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
#include <node.h>

#ifndef SM_MODE
#   define SM_MODE 0 /* Utilisation des API sécurisés */
#endif

#include <shmem.h>




/*----------------------------------------------------------------------------*/
/* Definitions                                                                */
/*----------------------------------------------------------------------------*/

//VER_INFO_EXPORT (smadmin, "$Revision: 2.3 $", "$Name:  $", __FILE__, "$Author: agibert $")

#define LGD_MODULE_NAME "sma"



#define USAGE "Usage : %s [ --help | --version [-v] | --create | --destroy ]\n"

#define QUIT           0
#define BASE_INIT      1
#define BASE_OPEN      2
#define BASE_INFO      3
#define BASE_CLOSE     4
#define BASE_END       5
#define MHH_INIT       6
#define MHH_OPEN       7
#define MHH_CHECK      8 
#define MHH_COMPRESS   9
#define MHH_CLOSE      10 
#define MHH_END        11
#define ALLOC_CHUNK    12
#define WRITE_CHUNK    13
#define READ_CHUNK     14
#define FREE_CHUNK     15
#define DSR_DUMP       16
#define ACR_DUMP       17
#define FCR_DUMP       18

#define NAME_LEN       100
#define NAME_SIZE      ( NAME_LEN + 1)

#define TMP_LEN        100
#define TMP_SIZE       ( TMP_LEN + 1)

#define TYPE_NUMBER    1
#define TYPE_STRING    2



char  menu[ 1000];

char  tmp[ TMP_SIZE];





#define MENU_ENTRY_ADD( string, id, label)  sprintf( tmp,  (string), (id), (label)); strcat( menu, tmp);







/*----------------------------------------------------------------------------*/
/* Prototypes                                                                 */
/*----------------------------------------------------------------------------*/

/*----------------------------------------------------------------------------*/
/* Init_Menu                                                                 */
/*----------------------------------------------------------------------------*/
void  Init_Menu( void);



/*----------------------------------------------------------------------------*/
/* Print_Menu                                                                 */
/*----------------------------------------------------------------------------*/

int  Print_Menu( void);



/*----------------------------------------------------------------------------*/
/* Entry_Promp                                                                */
/*----------------------------------------------------------------------------*/

void  Entry_Prompt( void  *, char  *, short);







/*----------------------------------------------------------------------------*/
/* Init_Menu                                                                 */
/*----------------------------------------------------------------------------*/

void  Init_Menu( void)
{
    sprintf( menu, "Menu :\n");

    MENU_ENTRY_ADD( "  - %2d) %-18s\n", QUIT,         "Quit");
    MENU_ENTRY_ADD( "  - %2d) %-18s",   BASE_INIT,    "Init library");
    MENU_ENTRY_ADD( "  - %2d) %-18s",   BASE_OPEN,    "Open library");
    MENU_ENTRY_ADD( "  - %2d) %-18s\n", BASE_INFO,    "Info library");
    MENU_ENTRY_ADD( "  - %2d) %-18s",   BASE_CLOSE,   "Close library");
    MENU_ENTRY_ADD( "  - %2d) %-18s",   BASE_END,     "End library");
    MENU_ENTRY_ADD( "  - %2d) %-18s\n", MHH_INIT,     "Init heap");
    MENU_ENTRY_ADD( "  - %2d) %-18s",   MHH_OPEN,     "Open heap");
    MENU_ENTRY_ADD( "  - %2d) %-18s",   MHH_CHECK,    "Check/recover heap");
    MENU_ENTRY_ADD( "  - %2d) %-18s\n", MHH_COMPRESS, "Compress heap");
    MENU_ENTRY_ADD( "  - %2d) %-18s",   MHH_CLOSE,    "Close heap");
    MENU_ENTRY_ADD( "  - %2d) %-18s",   MHH_END,      "End heap");
    MENU_ENTRY_ADD( "  - %2d) %-18s\n", ALLOC_CHUNK,  "Allocate chunk");
    MENU_ENTRY_ADD( "  - %2d) %-18s",   WRITE_CHUNK,  "Write chunk");
    MENU_ENTRY_ADD( "  - %2d) %-18s",   READ_CHUNK,   "Read chunk");
    MENU_ENTRY_ADD( "  - %2d) %-18s\n", FREE_CHUNK,   "Free chunk");
    MENU_ENTRY_ADD( "  - %2d) %-18s",   DSR_DUMP,     "Dump DSR");
    MENU_ENTRY_ADD( "  - %2d) %-18s",   ACR_DUMP,     "Dump ACR");
    MENU_ENTRY_ADD( "  - %2d) %-18s\n", FCR_DUMP,     "Dump FCR");
}





/*----------------------------------------------------------------------------*/
/* Print_Menu                                                                 */
/*----------------------------------------------------------------------------*/

int  Print_Menu( void)
{
    fprintf (stdout, "------------------------------------------------------\n%s", menu);

    tmp[0] = '\0';
    while( tmp[0] == '\0')
    {
        printf ("\nChoice ? ");
        fgets( tmp, TMP_SIZE, stdin);
        tmp[ strlen( tmp) - 1] = '\0';
    }

    return atoi( tmp);
}





/*----------------------------------------------------------------------------*/
/* Entry_Promp                                                                */
/*----------------------------------------------------------------------------*/

void  Entry_Prompt( void  *Out_Ptr, char  *Prompt_Ptr, short  Type)
{
    static char  tmp[ TMP_SIZE];

    
    fprintf( stdout, "\n%s", Prompt_Ptr);
    
    fgets( tmp, TMP_SIZE, stdin);
    tmp[ strlen( tmp) - 1] = '\0';
    
    switch( Type)
    {
        case TYPE_NUMBER:
        {
            if( sscanf( tmp, "0x%lx", Out_Ptr) != 1)
            {
                if( sscanf( tmp, "%ld", Out_Ptr) != 1)
                {
                    fprintf( stdout, "\nInvalid number\n");
                    break;
                }
            }

            break;
        }

        case TYPE_STRING:
        {
            strcpy( ( char *)Out_Ptr, tmp);
            
            break;
        }        
    }
}





/*----------------------------------------------------------------------------*/
/* main                                                                       */
/*----------------------------------------------------------------------------*/

int  main( int  argc, char  **argv)
{
    LGT_Status   lg_status;
    int          choice;
    char         name[ NAME_SIZE];
    int          Mode, Locked;
    void        *ptr;
    char         answer[ 10];
    SMT_Heap    *Heap;
    int          Nb_Detected, Nb_Corrected;
    size_t       size;


    if( ( lg_status = LG_Library_Open( LGD_LOG_WRITER_DEFAULT, true)) != LGS_OK)
    {
        fprintf( stderr, "Can't open LibLog library: (%d)\n", lg_status);
        return( -1);
    }

    LG_LOG_INFO( "Start SMAdmin");

    
    /* Lancement de commande d'administration */

    if( argc >= 2)
    {
        if( !strcmp( argv[ 1], "--help"))
        {
            fprintf( stderr, USAGE, argv[ 0]);
            return( -1);
        }
        
/*    else if( !strcmp( argv[1], "--version"))
      {
      if( argc >= 3 && !strcmp( argv[2], "-v"))
      return VER_Object_Print( stdout, VERD_VERBOSE);
      else
      return VER_Object_Print( stdout, VERD_MINIMAL);
      }
*/
        
        else if( !strcmp( argv[1], "--create"))
        {
            if( SM_Library_Open( 0, NULL, SMD_CREATE | SMD_DEBUG_ALL) != SMS_OK)
            {
                fprintf( stderr, "=> Impossible de créer l'instance de la librairie LIBSHMEM\n");
                return( -1);
            }

            return( 0);
        }
        else if( !strcmp( argv[1], "--destroy"))
        {
            if( SM_Library_Open( 0, NULL, SMD_OPEN | SMD_DEBUG_ALL) != SMS_OK || SM_Library_Close( SMD_DESTROY) != SMS_OK)
            {
                fprintf( stderr, "=> Impossible de détruire l'instance de la librairie LIBSHMEM\n");
                return( -1);
            }
            return( 0);
        }
        else
        {
            fprintf( stderr, USAGE, argv[ 0]);
            return( -1);
        }
    }

    
    /* Lancement du menu intercatif */

    Init_Menu();

    choice = Print_Menu();

    while( choice != QUIT)
    {
        strcpy( answer, "yes");

        switch( choice)
        {
            case BASE_INIT:
            {
                fprintf( stdout, "\nReturn code = %s\n", SM_Library_Open( 0, NULL, SMD_CREATE | SMD_DEBUG_ALL) == SMS_OK ? "OK" : "NOK" );
                
                break;
            }

            case BASE_OPEN:
            {
                fprintf( stdout, "\nReturn code = %s\n", SM_Library_Open( 0, NULL, SMD_OPEN | SMD_DEBUG_ALL) == SMS_OK ? "OK" : "NOK" );
                
                break;
            }

            case BASE_INFO:
            {
                SM_Library_Dump( stderr);
                
                break;
            }

            case BASE_CLOSE:
            {
                fprintf( stdout, "\nReturn code = %s\n", SM_Library_Close( SMD_CLOSE) == SMS_OK ? "OK" : "NOK" );
                
                break;
            }

            case BASE_END:
            {
                fprintf( stdout, "\nReturn code = %s\n", SM_Library_Close( SMD_DESTROY) == SMS_OK ? "OK" : "NOK" );
                
                break;
            }

            case MHH_INIT:
            {
                Entry_Prompt( name,  "Heap name ? ",       TYPE_STRING);
                Entry_Prompt( &size, "Allocation size ? ", TYPE_NUMBER);
                
                fprintf( stdout, "\nReturn code = %s\n", SM_Heap_Open( name, &Heap, size, SMD_CREATE, &Locked) == SMS_OK ? "OK" : "NOK" );
                
                break;
            }

            case MHH_OPEN:
            {
                Entry_Prompt( name,  "Heap name ? ",                      TYPE_STRING);
                Entry_Prompt( &size, "Opening mode( read=1  write=2) ? ", TYPE_NUMBER);

                if( size == '1')
                {
                    Mode = SMD_READ;
                }
                else
                {
                    Mode = SMD_WRITE;
                }
            
                if( !strcmp( name, HEAP_SYSTEM))
                {
                    fprintf( stdout, \
                             "\n*** Warning : you can not make this operation on the system heap ***\n");
                    break;
                }

                fprintf( stdout, "\nReturn code = %s\n", SM_Heap_Open( name, &Heap, 0, SMD_OPEN | Mode, &Locked) == SMS_OK ? "OK" : "NOK" );

                break;
            }

            case MHH_COMPRESS:
            {
                Entry_Prompt( name,  "Heap name ? ", TYPE_STRING);

                if( SM_Heap_Open( name, &Heap, 0, SMD_OPEN | SMD_WRITE, &Locked) == SMS_OK)
                {
                    size_t Compress;
                    SM_Heap_Compress( Heap, &Compress);
                    fprintf( stdout, "\nCompression size = %d byte(s)\n", (int)Compress);
                    if( Locked == TRUE) SM_Heap_Unlock( Heap);
                }
                
                break;
            }

            case MHH_CHECK:
            {
                Entry_Prompt( name,  "Heap name ? ", TYPE_STRING);

                if( SM_Heap_Open( name, &Heap, 0, SMD_OPEN | SMD_WRITE, &Locked) == SMS_OK)
                {
                    Nb_Detected = Nb_Corrected = 0;
                    SM_Heap_Check( Heap, &Nb_Detected, &Nb_Corrected, stderr);
                    if( Locked == TRUE) SM_Heap_Unlock( Heap);
                }
                
                break;
            }

            case MHH_CLOSE:
            {
                Entry_Prompt( name,  "Heap name ? ", TYPE_STRING);
                
                if( !strcmp( name, HEAP_SYSTEM))
                {
                    fprintf( stdout, \
                             "\n*** Warning : you can not make this operation on the system heap ***\n");
                    break;
                }
                
                if( SM_Heap_IsOpen( name, &Heap) == SMS_OK)
                {
                    fprintf( stdout, "\nReturn code = %s\n", SM_Heap_Close( Heap) == SMS_OK ? "OK" : "NOK" );
                }
                else
                {
                    fprintf( stdout, "\nHeap %s is not opened\n", name);
                }
                                
                break;
            }

            case MHH_END:
            {
                Entry_Prompt( name,  "Heap name ? ", TYPE_STRING);

                if( !strcmp( name, HEAP_SYSTEM))
                {
                    fprintf( stdout, \
                             "\n*** Warning : you can not make this operation on the system heap ***\n");
                    break;
                }

                fprintf( stdout, "\nReturn code = %s\n", SM_Heap_End( name) == SMS_OK ? "OK" : "NOK" );

                break;
            }

            case ALLOC_CHUNK:
            {
                Entry_Prompt( name,  "Heap name ? ",       TYPE_STRING);
                Entry_Prompt( &size, "Allocation size ? ", TYPE_NUMBER);
                                
                if( SM_Heap_Open( name, &Heap, 0, SMD_OPEN | SMD_WRITE, &Locked) == SMS_OK)
                {
                    SM_Chunk_Alloc( Heap, size, &ptr);
                    if( Locked == TRUE) SM_Heap_Unlock( Heap);
                }
                
                fprintf( stdout, "\nAddress = 0x%p( %lu)\n", ptr, ptr);
                
                break;
            }

            case WRITE_CHUNK:
            {
                Entry_Prompt( &ptr, "Address ? ",          TYPE_NUMBER);
                Entry_Prompt( ptr,  "String to put in ? ", TYPE_STRING);
                
                fprintf( stdout, "\nOK\n");
                break;
            }

            case READ_CHUNK:
            {
                Entry_Prompt( &ptr, "Address ? ", TYPE_NUMBER);
            
                fprintf( stdout, "\nValeur du pointeur = %s\n",( char *)ptr);
            
                break;
            }

            case FREE_CHUNK:
            {
                Entry_Prompt( name, "Heap name ? ", TYPE_STRING);
                Entry_Prompt( &ptr, "Address ? ",   TYPE_NUMBER);
            
                if( SM_Heap_Open( name, &Heap, 0, SMD_OPEN | SMD_WRITE, &Locked) == SMS_OK)
                {
                    fprintf( stdout, "\nReturn code = %s\n", \
                             SM_Chunk_Free( Heap, ptr) == SMS_OK ? "OK" : "NOK" );
                    if( Locked == TRUE) SM_Heap_Unlock( Heap);
                }

                break;
            }

            case DSR_DUMP:
            {
                Entry_Prompt( name,  "Heap name ? ", TYPE_STRING);
                
                if( SM_Heap_Open( name, &Heap, 0, SMD_OPEN | SMD_READ, &Locked) == SMS_OK)
                {
                    ND_DataStruct_Info_Print(  stdout, Heap->MHH->DSR, NDD_RECURSIVE_MODE_PARENT_CHILD, 0, 0);
                    ND_DataStruct_Value_Print( stdout, Heap->MHH->DSR, NDD_RECURSIVE_MODE_PARENT_CHILD, 0, 0);
                    if( Locked == TRUE) SM_Heap_Unlock( Heap);
                }
                
                break;
            }

            case ACR_DUMP:
            {
                Entry_Prompt( name,  "Heap name ? ", TYPE_STRING);

                if( SM_Heap_Open( name, &Heap, 0, SMD_OPEN | SMD_READ, &Locked) == SMS_OK)
                {
                    ND_DataStruct_Info_Print(  stdout, Heap->MHH->ACR, NDD_RECURSIVE_MODE_PARENT_CHILD, 0, 0);
                    ND_DataStruct_Value_Print( stdout, Heap->MHH->ACR, NDD_RECURSIVE_MODE_PARENT_CHILD, 0, 0);
                    if( Locked == TRUE) SM_Heap_Unlock( Heap);
                }
                
                break;
            }

            case FCR_DUMP:
            {
                Entry_Prompt( name,  "Heap name ? ", TYPE_STRING);
                
                if( SM_Heap_Open( name, &Heap, 0, SMD_OPEN | SMD_READ, &Locked) == SMS_OK)
                {
                    ND_DataStruct_Info_Print(  stdout, Heap->MHH->FCR, NDD_RECURSIVE_MODE_PARENT_CHILD, 0, 0);
                    ND_DataStruct_Value_Print( stdout, Heap->MHH->FCR, NDD_RECURSIVE_MODE_PARENT_CHILD, 0, 0);
                    if( Locked == TRUE) SM_Heap_Unlock( Heap);
                }
                
                break;
            }

            case QUIT:
            {
                fprintf( stdout, "\nExiting program ...\n");
                
                break;
            }

            default:
            {
                fprintf( stdout, "\nUndefined choice %d\n", choice);
            }
        }
        
        choice = Print_Menu();
    }
    

    LG_LOG_INFO( "End SMAdmin");

    if( ( lg_status = LG_Library_Close( true)) != LGS_OK)
    {
        fprintf( stderr, "Can't close LibLog library: (%d)\n", lg_status);
        return( -1);
    }

    return( 0);
}