#include <reg51.h>

#define uchar unsigned char

extern uchar key_press;

uchar get_key_press( uchar key_mask )
{
  EA = 0;
  key_mask &= key_press;                        // read key(s)
  key_press ^= key_mask;                        // clear key(s)
  EA = 1;
  return key_mask;
}