@@ -214,50 +214,21 @@ impl_region_model_context::terminate_path ()
}
bool
-impl_region_model_context::get_malloc_map (sm_state_map **out_smap,
- const state_machine **out_sm,
- unsigned *out_sm_idx)
-{
- unsigned malloc_sm_idx;
- if (!m_ext_state.get_sm_idx_by_name ("malloc", &malloc_sm_idx))
- return false;
-
- *out_smap = m_new_state->m_checker_states[malloc_sm_idx];
- *out_sm = &m_ext_state.get_sm (malloc_sm_idx);
- *out_sm_idx = malloc_sm_idx;
- return true;
-}
-
-bool
-impl_region_model_context::get_fd_map (sm_state_map **out_smap,
- const state_machine **out_sm,
- unsigned *out_sm_idx)
-{
- unsigned fd_sm_idx;
- if (!m_ext_state.get_sm_idx_by_name ("file-descriptor", &fd_sm_idx))
- return false;
-
- *out_smap = m_new_state->m_checker_states[fd_sm_idx];
- *out_sm = &m_ext_state.get_sm (fd_sm_idx);
- *out_sm_idx = fd_sm_idx;
- return true;
-}
-
-bool
-impl_region_model_context::get_taint_map (sm_state_map **out_smap,
- const state_machine **out_sm,
- unsigned *out_sm_idx)
+impl_region_model_context::get_state_map_by_name (const char *name,
+ sm_state_map **out_smap,
+ const state_machine **out_sm,
+ unsigned *out_sm_idx)
{
if (!m_new_state)
return false;
- unsigned taint_sm_idx;
- if (!m_ext_state.get_sm_idx_by_name ("taint", &taint_sm_idx))
+ unsigned sm_idx;
+ if (!m_ext_state.get_sm_idx_by_name (name, &sm_idx))
return false;
- *out_smap = m_new_state->m_checker_states[taint_sm_idx];
- *out_sm = &m_ext_state.get_sm (taint_sm_idx);
- *out_sm_idx = taint_sm_idx;
+ *out_smap = m_new_state->m_checker_states[sm_idx];
+ *out_sm = &m_ext_state.get_sm (sm_idx);
+ *out_sm_idx = sm_idx;
return true;
}
@@ -96,15 +96,10 @@ class impl_region_model_context : public region_model_context
{
return &m_ext_state;
}
- bool get_fd_map (sm_state_map **out_smap,
- const state_machine **out_sm,
- unsigned *out_sm_idx) final override;
- bool get_malloc_map (sm_state_map **out_smap,
- const state_machine **out_sm,
- unsigned *out_sm_idx) final override;
- bool get_taint_map (sm_state_map **out_smap,
- const state_machine **out_sm,
- unsigned *out_sm_idx) final override;
+ bool get_state_map_by_name (const char *name,
+ sm_state_map **out_smap,
+ const state_machine **out_sm,
+ unsigned *out_sm_idx) override;
const gimple *get_stmt () const override { return m_stmt; }
@@ -737,19 +737,33 @@ class region_model_context
virtual const extrinsic_state *get_ext_state () const = 0;
- /* Hook for clients to access the "fd" state machine in
+ /* Hook for clients to access the a specific state machine in
any underlying program_state. */
- virtual bool get_fd_map (sm_state_map **out_smap,
- const state_machine **out_sm,
- unsigned *out_sm_idx) = 0;
- /* Likewise for the "malloc" state machine. */
- virtual bool get_malloc_map (sm_state_map **out_smap,
- const state_machine **out_sm,
- unsigned *out_sm_idx) = 0;
- /* Likewise for the "taint" state machine. */
- virtual bool get_taint_map (sm_state_map **out_smap,
- const state_machine **out_sm,
- unsigned *out_sm_idx) = 0;
+ virtual bool get_state_map_by_name (const char *name,
+ sm_state_map **out_smap,
+ const state_machine **out_sm,
+ unsigned *out_sm_idx) = 0;
+
+ /* Precanned ways for clients to access specific state machines. */
+ bool get_fd_map (sm_state_map **out_smap,
+ const state_machine **out_sm,
+ unsigned *out_sm_idx)
+ {
+ return get_state_map_by_name ("file-descriptor", out_smap, out_sm,
+ out_sm_idx);
+ }
+ bool get_malloc_map (sm_state_map **out_smap,
+ const state_machine **out_sm,
+ unsigned *out_sm_idx)
+ {
+ return get_state_map_by_name ("malloc", out_smap, out_sm, out_sm_idx);
+ }
+ bool get_taint_map (sm_state_map **out_smap,
+ const state_machine **out_sm,
+ unsigned *out_sm_idx)
+ {
+ return get_state_map_by_name ("taint", out_smap, out_sm, out_sm_idx);
+ }
/* Get the current statement, if any. */
virtual const gimple *get_stmt () const = 0;
@@ -796,21 +810,10 @@ public:
const extrinsic_state *get_ext_state () const override { return NULL; }
- bool get_fd_map (sm_state_map **,
- const state_machine **,
- unsigned *) override
- {
- return false;
- }
- bool get_malloc_map (sm_state_map **,
- const state_machine **,
- unsigned *) override
- {
- return false;
- }
- bool get_taint_map (sm_state_map **,
- const state_machine **,
- unsigned *) override
+ bool get_state_map_by_name (const char *,
+ sm_state_map **,
+ const state_machine **,
+ unsigned *) override
{
return false;
}
@@ -929,25 +932,12 @@ class region_model_context_decorator : public region_model_context
return m_inner->get_ext_state ();
}
- bool get_fd_map (sm_state_map **out_smap,
- const state_machine **out_sm,
- unsigned *out_sm_idx) override
- {
- return m_inner->get_fd_map (out_smap, out_sm, out_sm_idx);
- }
-
- bool get_malloc_map (sm_state_map **out_smap,
- const state_machine **out_sm,
- unsigned *out_sm_idx) override
- {
- return m_inner->get_malloc_map (out_smap, out_sm, out_sm_idx);
- }
-
- bool get_taint_map (sm_state_map **out_smap,
- const state_machine **out_sm,
- unsigned *out_sm_idx) override
+ bool get_state_map_by_name (const char *name,
+ sm_state_map **out_smap,
+ const state_machine **out_sm,
+ unsigned *out_sm_idx) override
{
- return m_inner->get_taint_map (out_smap, out_sm, out_sm_idx);
+ return m_inner->get_state_map_by_name (name, out_smap, out_sm, out_sm_idx);
}
const gimple *get_stmt () const override