Changed format of partials

Fix linux build
yifanlu · Feb 20, 2017 · 6c1c4dc · 6c1c4dc
1 parent 08ec0b6
commit 6c1c4dc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/psvimg-keyfind.c b/psvimg-keyfind.c
@@ -3,6 +3,7 @@
  * This software may be modified and distributed under the terms
  * of the MIT license.  See the LICENSE file for details.
  */
+#define _GNU_SOURCE
 #include <fcntl.h>
 #include <poll.h>
 #include <signal.h>
@@ -63,7 +64,7 @@ int find_key(int fd, uint32_t guess[KEY_LEN/sizeof(uint32_t)], int idx, uint8_t
   uint8_t tmp[AES_BLOCK_SIZE];
   status_t st;
 
-  memset(&guess[idx], 0, KEY_LEN - idx * sizeof(uint32_t));
+  memset(guess, 0, KEY_LEN);
   memset(zeros, 0, sizeof(zeros));
   st.found = 0;
   gcry_cipher_open(&ctx, GCRY_CIPHER_AES256, GCRY_CIPHER_MODE_ECB, 0);

diff --git a/utils.c b/utils.c
@@ -4,6 +4,7 @@
  * of the MIT license.  See the LICENSE file for details.
  */
 #include <stdlib.h>
+#include <stdint.h>
 #include <string.h>
 #include <unistd.h>