kdprompt.c

Go to the documentation of this file.

/*
 * PROJECT:     ReactOS KDBG Kernel Debugger Terminal Driver
 * LICENSE:     GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
 * PURPOSE:     Terminal line-editing (Prompt) interface
 * COPYRIGHT:   Copyright 2001-2004 David Welch <welch@cwcom.net>
 *              Copyright 2004-2005 Gregor Anich <blight@blight.eu.org>
 *              Copyright 2022-2023 Hermès Bélusca-Maïto <hermes.belusca-maito@reactos.org>
 */
 
/* INCLUDES ******************************************************************/
 
#include <ntoskrnl.h>
#include "kd.h"
#include "kdterminal.h"
 
/* In kdb.h only when KDBG defined */
#ifdef KDBG
extern PCSTR
KdbGetHistoryEntry(
    _Inout_ PLONG NextIndex,
    _In_ BOOLEAN Next);
#else
PCSTR
KdbGetHistoryEntry(
    _Inout_ PLONG NextIndex,
    _In_ BOOLEAN Next)
{
    /* Dummy function */
    return NULL;
}
#endif
 
 
/* FUNCTIONS *****************************************************************/
 
SIZE_T
KdIoReadLine(
    _Out_ PCHAR Buffer,
    _In_ SIZE_T Size)
{
    PCHAR Orig = Buffer;
    ULONG ScanCode = 0;
    CHAR Key;
    BOOLEAN EchoOn = !(KdbDebugState & KD_DEBUG_KDNOECHO);
    LONG CmdHistIndex = -1; // Start at end of history.
 
    /* Flush the input buffer */
    KdpFlushTerminalInput();
 
    /* Bail out if the buffer is zero-sized */
    if (Size == 0)
        return 0;
 
    for (;;)
    {
        Key = KdpReadTermKey(&ScanCode);
 
        /* Check for return or newline */
        if ((Key == '\r') || (Key == '\n'))
        {
            *Buffer = ANSI_NULL;
            KdIoPuts("\n");
            return (SIZE_T)(Buffer - Orig);
        }
        else if (Key == KEY_BS || Key == KEY_DEL)
        {
            /* Erase the last character */
            if (Buffer > Orig)
            {
                Buffer--;
                *Buffer = ANSI_NULL;
 
                if (EchoOn)
                    KdIoPrintf("%c %c", KEY_BS, KEY_BS);
                else
                    KdIoPrintf(" %c", KEY_BS);
            }
        }
        else if (ScanCode == KEY_SCAN_UP || ScanCode == KEY_SCAN_DOWN)
        {
            PCSTR CmdHistory = KdbGetHistoryEntry(&CmdHistIndex,
                                                  (ScanCode == KEY_SCAN_DOWN));
            if (CmdHistory)
            {
                SIZE_T i;
 
                /* Erase the whole line */
                while (Buffer > Orig)
                {
                    Buffer--;
                    *Buffer = ANSI_NULL;
 
                    if (EchoOn)
                        KdIoPrintf("%c %c", KEY_BS, KEY_BS);
                    else
                        KdIoPrintf(" %c", KEY_BS);
                }
 
                /* Copy and display the history entry */
                i = min(strlen(CmdHistory), Size - 1);
                memcpy(Orig, CmdHistory, i);
                Orig[i] = ANSI_NULL;
                Buffer = Orig + i;
                KdIoPuts(Orig);
            }
        }
        else
        {
            /* Do not accept characters anymore if the buffer is full */
            if ((SIZE_T)(Buffer - Orig) >= (Size - 1))
                continue;
 
            if (EchoOn)
                KdIoPrintf("%c", Key);
 
            *Buffer = Key;
            Buffer++;
        }
    }
}
 
/* EOF */