vbatts/grub
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

fix memory issues when used inside loops

Browse source
This commit is contained in:
BVK Chaitanya 2010-06-10 20:49:57 +05:30
parent ead4142900
commit a992a71ed8
6 changed files with 42 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

21
include/grub/script_sh.h
View file

@ -39,6 +39,7 @@ struct grub_script_cmd
struct grub_script
{
unsigned refcnt;
struct grub_script_mem *mem;
struct grub_script_cmd *cmd;
};
@ -61,7 +62,7 @@ struct grub_script_arg
char *str;
/* Parsed block argument. */
struct grub_script block;
struct grub_script *block;
/* Next argument part. */
struct grub_script_arg *next;
@ -227,6 +228,8 @@ struct grub_parser_param
struct grub_lexer_param *lexerstate;
};
void grub_script_mem_free (struct grub_script_mem *mem);
void grub_script_argv_free (struct grub_script_argv *argv);
int grub_script_argv_next (struct grub_script_argv *argv);
int grub_script_argv_append (struct grub_script_argv *argv, const char *s);
@ -354,4 +357,20 @@ grub_err_t grub_script_function_call (grub_script_function_t func,
char **
grub_script_execute_arglist_to_argv (struct grub_script_arglist *arglist, int *count);
static inline struct grub_script *
grub_script_get (struct grub_script *script)
{
script->refcnt++;
return script;
}
static inline void
grub_script_put (struct grub_script *script)
{
if (script->refcnt == 0)
grub_script_free (script);
else
script->refcnt--;
}
#endif /* ! GRUB_NORMAL_PARSER_HEADER */