17. Extract the Nibble from an 8-bit Register

#include <stdio.h>

unsigned char extractNibble(unsigned char reg, int pos) {
    // Write your code here

    unsigned char shift_bit = 0;
    unsigned char result = 0;

    for(int i = 0; i < 4; i++){

        if(pos == 1){
            shift_bit = reg >> 4;
            result |= (shift_bit & (0x0F));
        }

        else if(pos == 0){
            shift_bit = reg >> 0;
            result |= (shift_bit & (0x0F));
        }

    }

    return result;
}

int main() {
    unsigned char reg;
    int pos;
    scanf("%hhu %d", &reg, &pos);
    printf("%d", extractNibble(reg, pos));
    return 0;
}