Смотреть на суффиксы/окончания — не надёжно, сравните:
скал (род. п.) — генерал — знал
Эти слова относятся, к глаголам или к существительным?
Думаю, варианта два: или подключить словарь со словоформами (напр.
грамматический словарь Зализняка) и искать по нему (но омонимы и омофоны будут проблемой: стекло (сущ.) — стекло (гл., пр.в.) и т.д.), или же попытаться реализовать простейший синтаксический анализатор, но тут проблемой будет свободный порядок слов в русском.
В любом случае, если нужна точность, просто это реализовать не получится.