Лингвистическая задача

Рассмотрим существительные мужского рода, оканчивающиеся на -ок: кружок, масленок, брелок и т.д. При склонении таких букв о может выступать как беглая гласная: кружка, масленком и т.д. При склонении некоторых таких слов гласная о, однако, сохраняется.
Это, во-первых, слова из трёх букв: ток, сок и т.д.; замем слова скок, блок, волок, восток, шток и слова, основа которых оканчивается на такое сочестание букв (т.е. перескок, пищеблок, юго-восток и т.д.)ж наконец, имеется и ещё ряд слов, среди которых укажем следующие: брелок, щелок, войлок, челнок, зарок, срок, урок, знаток, поток, сток, артишок.
Дан текст, среди символов которого имеется пробел.
Группа символов, предшествующая первому пробелу, представлет собой русское слово - существительно мужского рода, оканчивающееся на -ок после первого пробела идёт одна из букв и, р, д, в, т, п, указывающая падеж (именительный, родительный, дательный, винительный, творительный, предложный). Получить данное слово в указанном падеже.

Как её решать осознаю, тут в основном рутинная проверка через позицию вхождения слова в список исключений, но есть несколько аспектов:

белок -> у белка, в белке, дать белку
блок -> у блока, в блоке, дать блоку

Конец один, а склонения разные... Вот как? Чувствую, надо смотреть на последние две буквы перед -ок, если они согласные, то тогда -ока, если одна гласная, то -ка... Но это как то туманно...