mtp deinit function

fix rx/tx/tmp buffers sizes.
v0.8.1

inc/mtp.h

@@ -101,6 +101,7 @@ typedef struct mtp_ctx_
 }mtp_ctx;
 
 mtp_ctx * mtp_init_responder();
+
 int  mtp_incoming_packet(mtp_ctx * ctx);
 void mtp_set_usb_handle(mtp_ctx * ctx, void * handle, uint32_t max_packet_size);
 
@@ -110,4 +111,6 @@ uint32_t mtp_add_storage(mtp_ctx * ctx, char * path, char * description);
 char * mtp_get_storage_description(mtp_ctx * ctx, uint32_t storage_id);
 char * mtp_get_storage_root(mtp_ctx * ctx, uint32_t storage_id);
 
-#define APP_VERSION "v0.8.0"
+void mtp_deinit_responder(mtp_ctx * ctx);
+
+#define APP_VERSION "v0.8.1"

src/mtp.c

@@ -64,19 +64,19 @@ mtp_ctx * mtp_init_responder()
 	{
 		memset(ctx,0,sizeof(mtp_ctx));
 
-		ctx->wrbuffer = malloc(MAX_RX_BUFFER_SIZE);
+		ctx->wrbuffer = malloc(MAX_TX_BUFFER_SIZE);
 		if(!ctx->wrbuffer)
 			goto init_error;
 
-		ctx->rdbuffer = malloc(128*1024);
+		ctx->rdbuffer = malloc( MAX_RX_BUFFER_SIZE );
 		if(!ctx->rdbuffer)
 			goto init_error;
 
-		ctx->rdbuffer2 = malloc(128*1024);
+		ctx->rdbuffer2 = malloc( MAX_RX_BUFFER_SIZE );
 		if(!ctx->rdbuffer2)
 			goto init_error;
 
-		ctx->temp_array = malloc(128*1024);
+		ctx->temp_array = malloc( MAX_STORAGE_NB * sizeof(uint32_t) );
 		if(!ctx->temp_array)
 			goto init_error;
 
@@ -108,6 +108,26 @@ mtp_ctx * mtp_init_responder()
 	return 0;
 }
 
+void mtp_deinit_responder(mtp_ctx * ctx)
+{
+	if( ctx )
+	{
+		if(ctx->wrbuffer)
+			free(ctx->wrbuffer);
+
+		if(ctx->rdbuffer)
+			free(ctx->rdbuffer);
+
+		if(ctx->rdbuffer2)
+			free(ctx->rdbuffer2);
+
+		if(ctx->temp_array)
+			free(ctx->temp_array);
+
+		free(ctx);
+	}
+}
+
 int delete_tree(mtp_ctx * ctx,uint32_t handle)
 {
 	int ret;

src/umtprd.c

@@ -91,8 +91,13 @@ int main(int argc, char *argv[])
 	else
 	{
 		PRINT_ERROR("USB Init failed !");
+
+		mtp_deinit_responder(mtp_context);
+
 		exit(-2);
 	}
 
+	mtp_deinit_responder(mtp_context);
+
 	exit(0);
 }