Changeset 3461 for trunk/src/kash/memalloc.c

Timestamp:

Sep 15, 2020, 2:55:26 PM (5 years ago)

Author:

bird

Message:

kash: Cache one pstack_block since parsecmd seems to be called for each statement line in a script. Also, mark pstack blocks as 'done' before returning from parsecmd and prevent them from becoming current (curpstack) again upon popping.

File:

• trunk/src/kash/memalloc.c (modified) (6 diffs)

trunk/src/kash/memalloc.c

@@ -371 +371 @@
                 TRACE2((NULL, "pstackrelease: %p - %u refs (%s)\n", pst, refs, caller)); K_NOREF(caller);
                 if (refs == 0) {
-                        shinstance * const psh = shthread_get_shell();
                         struct stack_block *top;
                         while ((top = pst->top) != &pst->first)
@@ -382 +381 @@
                         pst->nextbyte = NULL;
                         pst->top = NULL;
-                        /** @todo push into an alloc cache rather than freeing it */
-                        sh_free(psh, pst);
+
+                        if (!psh->freepstack)
+                                psh->freepstack = pst;
+                        else
+                                sh_free(psh, pst);
                 }
         } else
@@ -395 +397 @@
         while (target < psh->pstacksize) {
                 unsigned idx = --psh->pstacksize;
-                pstack_block *psk = psh->pstack[idx];
+                pstack_block *pst = psh->pstack[idx];
                 psh->pstack[idx] = NULL;
-                if (psh->curpstack == psk)
-                    psh->curpstack = idx > 0 ? psh->pstack[idx - 1] : NULL;
-                pstackrelease(psh, psk, "popstackmark");
+                if (psh->curpstack == pst) {
+                        pstack_block *pstnext;
+                        if (idx <= 0 || (pstnext = psh->pstack[idx - 1])->done)
+                                psh->curpstack = NULL;
+                        else
+                                psh->curpstack = pstnext;
+                }
+                pstackrelease(psh, pst, "popstackmark");
         }
 
@@ -454 +461 @@
          * Allocate and initialize it.
          */
-        pst = (pstack_block *)ckmalloc(psh, blocksize);
+        pst = psh->freepstack;
+        if (pst)
+                psh->freepstack = NULL;
+        else
+                pst = (pstack_block *)ckmalloc(psh, blocksize);
         pst->nextbyte          = &pst->first.space[0];
         pst->avail             = blocksize - offsetof(pstack_block, first.space);
         pst->topsize           = blocksize - offsetof(pstack_block, first.space);
+        pst->top               = &pst->first;
         pst->strleft           = 0;
-        pst->top               = &pst->first;
         pst->allocations       = 0;
         pst->bytesalloced      = 0;
@@ -468 +479 @@
         pst->fragmentation     = 0;
         pst->refs              = 1;
-        pst->padding           = 42;
+        pst->done              = K_FALSE;
         pst->first.prev        = NULL;
 
@@ -480 +491 @@
         TRACE2((psh, "pstackallocpush: %p - entry %u\n", pst, psh->pstacksize - 1));
         return pst;
+}
+
+/**
+ * Marks the block as done, preventing it from being marked current again.
+ */
+void pstackmarkdone(pstack_block *pst)
+{
+        pst->done = K_TRUE;
 }