Merge pull request #243 from pwpiwi/cleanup

[proxmark3-svn] / client / cmdlfviking.c

diff --git a/client/cmdlfviking.c b/client/cmdlfviking.c
index e43a9748902369cd28ba1d5b0d1e3681c4f74411..601c29f22e712de36a07d9111c5764cced73b70c 100644 (file)
--- a/client/cmdlfviking.c
+++ b/client/cmdlfviking.c
@@ -10,6 +10,7 @@
 #include <string.h>
 #include <inttypes.h>
 #include "proxmark3.h"
 #include <string.h>
 #include <inttypes.h>
 #include "proxmark3.h"

+#include "cmdlfviking.h"

 #include "ui.h"
 #include "util.h"
 #include "graph.h"
 #include "ui.h"
 #include "util.h"
 #include "graph.h"


@@ -17,7 +18,6 @@
 #include "cmddata.h"
 #include "cmdmain.h"
 #include "cmdlf.h"
 #include "cmddata.h"
 #include "cmdmain.h"
 #include "cmdlf.h"

-#include "cmdlfviking.h"

 #include "lfdemod.h"
 static int CmdHelp(const char *Cmd);
 
 #include "lfdemod.h"
 static int CmdHelp(const char *Cmd);
 


@@ -47,16 +47,43 @@ int usage_lf_viking_sim(void) {
 
 uint64_t getVikingBits(uint32_t id) {
        //calc checksum
 
 uint64_t getVikingBits(uint32_t id) {
        //calc checksum

-       uint8_t checksum = (id>>24) ^ ((id>>16) & 0xFF) ^ ((id>>8) & 0xFF) ^ (id & 0xFF) ^ 0xF2 ^ 0xA8;
-       return ((uint64_t)0xF2 << 56) | (id << 8) | checksum;
+       uint8_t checksum = ((id>>24) & 0xFF) ^ ((id>>16) & 0xFF) ^ ((id>>8) & 0xFF) ^ (id & 0xFF) ^ 0xF2 ^ 0xA8;
+       return ((uint64_t)0xF2 << 56) | ((uint64_t)id << 8) | checksum;
+}
+
+//could be moved to a viking file
+//by marshmellow
+//see ASKDemod for what args are accepted
+int CmdVikingDemod(const char *Cmd) {
+       if (!ASKDemod(Cmd, false, false, 1)) {
+               if (g_debugMode) PrintAndLog("ASKDemod failed");
+               return 0;
+       }
+       size_t size = DemodBufferLen;
+       //call lfdemod.c demod for Viking
+       int ans = VikingDemod_AM(DemodBuffer, &size);
+       if (ans < 0) {
+               if (g_debugMode) PrintAndLog("Error Viking_Demod %d", ans);
+               return 0;
+       }
+       //got a good demod
+       uint32_t raw1 = bytebits_to_byte(DemodBuffer+ans, 32);
+       uint32_t raw2 = bytebits_to_byte(DemodBuffer+ans+32, 32);
+       uint32_t cardid = bytebits_to_byte(DemodBuffer+ans+24, 32);
+       uint8_t  checksum = bytebits_to_byte(DemodBuffer+ans+32+24, 8);
+       PrintAndLog("Viking Tag Found: Card ID %08X, Checksum: %02X", cardid, (unsigned int) checksum);
+       PrintAndLog("Raw: %08X%08X", raw1,raw2);
+       setDemodBuf(DemodBuffer+ans, 64, 0);
+       return 1;

 }
 }

+

 //by marshmellow
 //see ASKDemod for what args are accepted
 int CmdVikingRead(const char *Cmd) {
        // read lf silently
        CmdLFRead("s");
        // get samples silently
 //by marshmellow
 //see ASKDemod for what args are accepted
 int CmdVikingRead(const char *Cmd) {
        // read lf silently
        CmdLFRead("s");
        // get samples silently

-       getSamples("30000",false);
+       getSamples("10000",false);

        // demod and output viking ID   
        return CmdVikingDemod(Cmd);
 }
        // demod and output viking ID   
        return CmdVikingDemod(Cmd);
 }


@@ -66,7 +93,7 @@ int CmdVikingClone(const char *Cmd) {
        uint64_t rawID = 0;
        bool Q5 = false;
        char cmdp = param_getchar(Cmd, 0);
        uint64_t rawID = 0;
        bool Q5 = false;
        char cmdp = param_getchar(Cmd, 0);

-       if (strlen(Cmd) < 0 || cmdp == 'h' || cmdp == 'H') return usage_lf_viking_clone();
+       if (strlen(Cmd) == 0 || cmdp == 'h' || cmdp == 'H') return usage_lf_viking_clone();

 
        id = param_get32ex(Cmd, 0, 0, 16);
        if (id == 0) return usage_lf_viking_clone();
 
        id = param_get32ex(Cmd, 0, 0, 16);
        if (id == 0) return usage_lf_viking_clone();


@@ -74,8 +101,8 @@ int CmdVikingClone(const char *Cmd) {
                Q5 = true;
 
        rawID = getVikingBits(id);
                Q5 = true;
 
        rawID = getVikingBits(id);

-
-       UsbCommand c = {CMD_VIKING_CLONE_TAG,{rawID >> 32, rawID & 0xFFFF, Q5}};
+       PrintAndLog("Cloning - ID: %08X, Raw: %08X%08X",id,(uint32_t)(rawID >> 32),(uint32_t) (rawID & 0xFFFFFFFF));
+       UsbCommand c = {CMD_VIKING_CLONE_TAG,{rawID >> 32, rawID & 0xFFFFFFFF, Q5}};

        clearCommandBuffer();
        SendCommand(&c);
        //check for ACK
        clearCommandBuffer();
        SendCommand(&c);
        //check for ACK


@@ -89,7 +116,7 @@ int CmdVikingSim(const char *Cmd) {
        uint8_t clk = 32, encoding = 1, separator = 0, invert = 0;
        char cmdp = param_getchar(Cmd, 0);
 
        uint8_t clk = 32, encoding = 1, separator = 0, invert = 0;
        char cmdp = param_getchar(Cmd, 0);
 

-       if (strlen(Cmd) < 0 || cmdp == 'h' || cmdp == 'H') return usage_lf_viking_sim();
+       if (strlen(Cmd) == 0 || cmdp == 'h' || cmdp == 'H') return usage_lf_viking_sim();

        id = param_get32ex(Cmd, 0, 0, 16);
        if (id == 0) return usage_lf_viking_sim();
 
        id = param_get32ex(Cmd, 0, 0, 16);
        if (id == 0) return usage_lf_viking_sim();
 


@@ -110,7 +137,8 @@ int CmdVikingSim(const char *Cmd) {
 
 static command_t CommandTable[] = {
        {"help",  CmdHelp,        1, "This help"},
 
 static command_t CommandTable[] = {
        {"help",  CmdHelp,        1, "This help"},

-       {"read",  CmdVikingRead,  0, "Attempt to read and Extract tag data"},
+       {"demod", CmdVikingDemod, 1, "Demodulate a Viking tag from the GraphBuffer"},
+       {"read",  CmdVikingRead,  0, "Attempt to read and Extract tag data from the antenna"},

        {"clone", CmdVikingClone, 0, "<8 digit ID number> clone viking tag"},
        {"sim",   CmdVikingSim,   0, "<8 digit ID number> simulate viking tag"},
        {NULL, NULL, 0, NULL}
        {"clone", CmdVikingClone, 0, "<8 digit ID number> clone viking tag"},
        {"sim",   CmdVikingSim,   0, "<8 digit ID number> simulate viking tag"},
        {NULL, NULL, 0, NULL}