Little Changes to Work properly

Signed-off-by: Abhinav Singh <[email protected]>

example/main.c

@@ -0,0 +1,50 @@
+#include "../include/node.h"
+#include "../include/seriliaze.h"
+
+
+#include<stdio.h>
+
+
+void funcA1(){
+    childSpan(__PRETTY_FUNCTION__);
+    pop();
+}
+
+void funcA2(){
+    childSpan(__PRETTY_FUNCTION__);
+    pop();
+}
+
+void funcA(){
+    childSpan(__PRETTY_FUNCTION__);
+    funcA1();
+    funcA2();
+    pop();
+}
+
+void funcB1(){
+    childSpan(__PRETTY_FUNCTION__);
+    pop();
+}
+
+void funcB2(){
+    childSpan(__PRETTY_FUNCTION__);
+    pop();
+}
+
+void funcB(){
+    childSpan(__PRETTY_FUNCTION__);
+    funcB1();
+    funcB2();
+    pop();
+}
+
+int main(){
+   newSpan(__PRETTY_FUNCTION__);
+    funcA();
+    funcB();
+    pop();
+
+    serialize(root);
+    printf("%s\n",getStream());
+}

include/node.h

@@ -3,11 +3,11 @@
 
 #include<stdlib.h>
 #include<string.h>
+#include "stack.h"
 
 #define CALL_SIZE 5
 #define NAME_SIZE 400
 
-typedef struct node node;
 
 struct node{
     char name[NAME_SIZE];
@@ -19,6 +19,7 @@ struct node{
 extern node* root;
 
 void newSpan(const char *);
-void childSpan(const char*, node*);
+void childSpan(const char*);
+void closeSpan();
 
 #endif

include/seriliaze.h

@@ -1,5 +1,13 @@
+
+#ifndef _SERIALIZE_
+#define _SERIALIZE_
+
 #include "node.h"
 #include "string.h"
+#include <stdlib.h>
+#include <stdio.h>
 
 void serialize(node*);
-char* getStream();
+char* getStream();
+
+#endif

include/stack.h

@@ -4,8 +4,7 @@
 
 #include<stdlib.h>
 
-#include "node.h"
-
+typedef struct node node;
 typedef struct stack stack;
 
 struct stack{

src/node.c

@@ -2,16 +2,29 @@
 
 node* root = NULL;
 
+void updateStack(node* n,short i){
+    if(i)
+        push(n);
+    else
+        pop();
+}
+
 void newSpan(const char* name){
     root = (node*)malloc(sizeof(node));
     strcpy(root->name, name);
+    updateStack(root,1);
 }
 
-void childSpan(const char* name, node* parentSpan){
+void childSpan(const char* name){
     node* nnode = (node*)malloc(sizeof(node));
     nnode->prev = NULL;
     nnode->cnt=0;
-    nnode->prev = parentSpan;
+    nnode->prev = top->address;
     strcpy(nnode->name, name);
-    parentSpan->called[parentSpan->cnt++] = nnode;
+    top->address->called[top->address->cnt++] = nnode;
+    updateStack(nnode,1);
+}
+
+void closeSpan(){
+    updateStack(NULL,0);
 }

src/serialize.c

@@ -16,5 +16,14 @@ void serialize(node* root){
 }
 
 char* getStream(){
+    FILE* ptr;
+    ptr = fopen("file.txt","w");
+    if(ptr == NULL){
+        printf("Error happened opening file\n");
+        return NULL;
+    }
+    fprintf(ptr,"%d\n",CALL_SIZE);
+    fprintf(ptr,"%s",stream);
+    fclose(ptr);
     return stream;
 }

src/stack.c

@@ -21,6 +21,7 @@ void pop(){
         return;
     stack* hld = top->prev;
     free(top);
-    hld->next = NULL;
+    if(hld)
+        hld->next = NULL;
     top=hld;
 }