#include <stdlib.h>
#include "stack.h"

// Pushes data as pointer onto the stack.
StackNode *push(StackNode *stack, void *data)
{
    StackNode *newNode = (StackNode *)malloc(sizeof(StackNode));
    if (newNode == NULL)
        return stack; // allocation failed: return unchanged stack

    newNode->data = data;
    newNode->next = stack;
    return newNode;
}

// Deletes the top element of the stack (latest added element) and releases its memory. (Pointer to data has to be
// freed by caller.)
StackNode *pop(StackNode *stack)
{
    if (stack == NULL)
        return NULL;

    StackNode *next = stack->next;
    free(stack);
    return next;
}

// Returns the data of the top element.
void *top(StackNode *stack)
{
    if (stack == NULL)
        return NULL;
    return stack->data;
}

// Clears stack and releases all memory.
void clearStack(StackNode *stack)
{
    while (stack != NULL)
    {
        StackNode *next = stack->next;
        free(stack);
        stack = next;
    }
}