Add missing headers for last change.

[melted] / src / mvcp-console / mvcp-console.c

diff --git a/src/mvcp-console/mvcp-console.c b/src/mvcp-console/mvcp-console.c
index bda4918..7daebaa 100644 (file)
--- a/src/mvcp-console/mvcp-console.c
+++ b/src/mvcp-console/mvcp-console.c
@@ -30,6 +30,8 @@
 
 /* Application header files */
 #include <melted/melted_local.h>
+#include <melted/melted_commands.h>
+#include <melted/melted_unit.h>
 #include <mvcp/mvcp_remote.h>
 #include <mvcp/mvcp_util.h>
 
@@ -47,6 +49,13 @@ void report( mvcp_response response )
                        printf( "%4d: %s\n", index, mvcp_response_get_line( response, index ) );
 }
 
+static void cleanup()
+{
+       int i;
+       for ( i = 0; i < MAX_UNITS; i++ )
+               melted_unit_close( melted_get_unit(i) );
+}
+
 int main( int argc, char **argv  )
 {
        mvcp_parser parser = NULL;
@@ -54,6 +63,7 @@ int main( int argc, char **argv  )
        char temp[ 1024 ];
        int index = 1;
 
+       atexit( cleanup );
        if ( argc > 2 && !strcmp( argv[ 1 ], "-s" ) )
        {
                printf( "Melted Client Instance\n" );