120 lines
2.5 KiB
C
120 lines
2.5 KiB
C
#include <stdio.h> // fopen, fwrite, fclose
|
|
#include <stdlib.h> // malloc, free
|
|
#include <string.h> // memset
|
|
|
|
#include "uwatec_smart.h"
|
|
#include "utils.h"
|
|
|
|
#define WARNING(expr) \
|
|
{ \
|
|
message ("%s:%d: %s\n", __FILE__, __LINE__, expr); \
|
|
}
|
|
|
|
|
|
int test_dump_memory (const char* filename)
|
|
{
|
|
device_t *device = NULL;
|
|
|
|
const unsigned int size = 2 * 1024 * 1024;
|
|
unsigned char *data = malloc (size * sizeof (unsigned char));
|
|
if (data == NULL)
|
|
return DEVICE_STATUS_MEMORY;
|
|
memset (data, 0, size * sizeof (unsigned char));
|
|
|
|
message ("uwatec_smart_device_open\n");
|
|
int rc = uwatec_smart_device_open (&device);
|
|
if (rc != DEVICE_STATUS_SUCCESS) {
|
|
WARNING ("Cannot open device.");
|
|
free (data);
|
|
return rc;
|
|
}
|
|
|
|
message ("uwatec_smart_device_handshake\n");
|
|
rc = uwatec_smart_device_handshake (device);
|
|
if (rc != DEVICE_STATUS_SUCCESS) {
|
|
WARNING ("Handshake failed.");
|
|
device_close (device);
|
|
free (data);
|
|
return rc;
|
|
}
|
|
|
|
message ("uwatec_smart_device_version\n");
|
|
unsigned char version[UWATEC_SMART_VERSION_SIZE] = {0};
|
|
rc = uwatec_smart_device_version (device, version, sizeof (version));
|
|
if (rc != DEVICE_STATUS_SUCCESS) {
|
|
WARNING ("Cannot identify computer.");
|
|
device_close (device);
|
|
free (data);
|
|
return rc;
|
|
}
|
|
|
|
message ("device_download\n");
|
|
rc = device_download (device, data, size);
|
|
if (rc < 0) {
|
|
WARNING ("Cannot read data.");
|
|
device_close (device);
|
|
free (data);
|
|
return rc;
|
|
}
|
|
|
|
message ("Dumping data\n");
|
|
FILE* fp = fopen (filename, "wb");
|
|
if (fp != NULL) {
|
|
fwrite (data, sizeof (unsigned char), rc, fp);
|
|
fclose (fp);
|
|
}
|
|
|
|
message ("device_close\n");
|
|
rc = device_close (device);
|
|
if (rc != DEVICE_STATUS_SUCCESS) {
|
|
WARNING ("Cannot close device.");
|
|
free (data);
|
|
return rc;
|
|
}
|
|
|
|
free (data);
|
|
|
|
return DEVICE_STATUS_SUCCESS;
|
|
}
|
|
|
|
|
|
const char* errmsg (int rc)
|
|
{
|
|
switch (rc) {
|
|
case DEVICE_STATUS_SUCCESS:
|
|
return "Success";
|
|
case DEVICE_STATUS_UNSUPPORTED:
|
|
return "Unsupported operation";
|
|
case DEVICE_STATUS_TYPE_MISMATCH:
|
|
return "Device type mismatch";
|
|
case DEVICE_STATUS_ERROR:
|
|
return "Generic error";
|
|
case DEVICE_STATUS_IO:
|
|
return "Input/output error";
|
|
case DEVICE_STATUS_MEMORY:
|
|
return "Memory error";
|
|
case DEVICE_STATUS_PROTOCOL:
|
|
return "Protocol error";
|
|
case DEVICE_STATUS_TIMEOUT:
|
|
return "Timeout";
|
|
default:
|
|
return "Unknown error";
|
|
}
|
|
}
|
|
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
message_set_logfile ("SMART.LOG");
|
|
|
|
int a = test_dump_memory ("SMART.DMP");
|
|
|
|
message ("\nSUMMARY\n");
|
|
message ("-------\n");
|
|
message ("test_dump_memory: %s\n", errmsg (a));
|
|
|
|
message_set_logfile (NULL);
|
|
|
|
return 0;
|
|
}
|