From 51a4b2d9bc81a7367dfb36f9bc9b02974d4bc860 Mon Sep 17 00:00:00 2001
From: Sebastian Hugentobler <sebastian@vanwa.ch>
Date: Thu, 17 Mar 2022 12:46:28 +0100
Subject: [PATCH] fix k function

---
 .idea/gradle.xml                   | 1 +
 src/main/java/ch/fhnw/kry/SPN.java | 6 ++++--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/.idea/gradle.xml b/.idea/gradle.xml
index ba1ec5c..611e7c8 100644
--- a/.idea/gradle.xml
+++ b/.idea/gradle.xml
@@ -1,5 +1,6 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
+  <component name="GradleMigrationSettings" migrationVersion="1" />
   <component name="GradleSettings">
     <option name="linkedExternalProjectsSettings">
       <GradleProjectSettings>
diff --git a/src/main/java/ch/fhnw/kry/SPN.java b/src/main/java/ch/fhnw/kry/SPN.java
index a93c850..92de8d7 100644
--- a/src/main/java/ch/fhnw/kry/SPN.java
+++ b/src/main/java/ch/fhnw/kry/SPN.java
@@ -1,10 +1,12 @@
 package ch.fhnw.kry;
 
 public class SPN {
+    private final static int ROUND_KEY_LENGTH = 16;
+
     public int k(int key, int i) {
         i *= 4;
-        int mask = 0xFFFF >>> i;
+        int mask = 0xFFFF_0000 >>> i;
 
-        return key & mask;
+        return key & mask >>> ROUND_KEY_LENGTH - i;
     }
 }