Skip to content

Commit

Permalink
Feature 0.9.8 (#1680)
Browse files Browse the repository at this point in the history
* test

* Add Django mgt cmd importcomponents to batch import components

Updates to generation of components to improve adherence to OSCAL
specification by removing certain keys when value for keys is None.

Added new parameter `existing_import_record` to importing and creating
components to group multiple imports under the same import record.

* Make Elements.description TextField and required

Change Elements.description to TextField and make required.
Modify component edit modal to use a textarea for description field.
Fixes to show error on problem saving edit to library component

* Use self.element.tags.exists()

* Send proper error message when editing component

* Temporarily remove controls/migrations/0051_auto

* Add back better controls/migrations/0051_auto

* fixed control and sectioning for components in system

* check for security impact level statement when updating. Readding retrieval of security impact levels. todo on separate form for levels.

* adding component_state and component_type to system component and component library display. including component_metadata template to keep styling consistent

* pulling in some of the information from statement about system. Rest would come from questionnaires. system_information_types is not used at all.

* a todo for fisma impact level renaming

* added project_security_objs_edit to edit security objectives separately from project editing.

* update changelog and some wording in the modal

* Move action-button styles from inline to style section

* two views/urls for editing component state and type

* adding component type and state to ElementForm

* adding just the display of the state and type to component library components not the ability to change.

* changelog

* FISMA IMPACT LEVEL is now SECURITY SENSITIVITY LEVEL

* Work inprogress

* Work inprogress

* Da/quick insert (#1601)

* make sure component_type not element_type is exported

* ssp versions should be floats not integers. Information types needs a uuid

* adding empty placeholders for the required keys.

* using updated for component version

* party-uuids is still a todo

* categorizations is still a todo

* parties is still a todo

* Fix system ctl detail page err; Improve creating smt from prototypes (#1602)

Refactor creating system control statements from component library
prototype statements when adding a component from the library to a system
and reduce by an order a magnitude the time it takes to add a component to system.

Rename smt.create_instance_from_prototype to smt.create_system_control_smt_from_component_prototype_smt

Fix bug breaking rendering of system's control detail page by
removing an errant login_required decorator on a function.
Add test for system control page. Will add test(s) for system control detail page.

Co-authored-by: Greg Elin <[email protected]>

* Automatically clear, refresh output document content downloading docs

Performnce of document generation now sufficiently fast
to not require cache and manual "Refresh documents" button.

* remove comments. changelog

* Fixed an issue where statement didn't exist while exporting to oscal (#1605)

* Fixed an issue where statement didn't exist while exporting to oscal

* Update CHANGELOG

Co-authored-by: Alexander Ward <[email protected]>
Co-authored-by: Greg Elin <[email protected]>

* Align Delete section on project settings (#1604)

Co-authored-by: Greg Elin <[email protected]>

* Update CHANGELOG

* Ge/file upload extensions (#1607)

* Accepts file uploads with capitalized extensions, e.g. ".JPG".

Adjust file upload validator to recognize capitalized extensions
and also recognizes ".jpeg" in addition to ".jpg".

* Add tests for validating uppercase extensions on file uploads

* Add test fixture data

Co-authored-by: Greg Elin <[email protected]>

* Batch update cntl impl smts when component_statement changes

Implemented a faster way to update status of system controls.
When user sets a system component state to "operational" all statements
associated with that component for the system get their status set to
"Implemented". Similarly, setting component’s state to "planned" batch
sets all component statements for that system to "Planned", and
"under-development" sets component statements to "Partially Implemented".

Display system component component_state and component_type when
component is listed for a system.

* More okta changes

* export a projects ssp control implementations with export form (#1611)

* export a projects ssp control implementations with export form

* remove comments

* Correct slugify import

* Security update Python 3.2.4 due to https://snyk.io/vuln/SNYK-PYTHON-DJANGO-1298665

* Polish SSP control CSV export form

Co-authored-by: Greg Elin <[email protected]>

* Add 'Create a template' button to template library (#1610)

Co-authored-by: Greg Elin <[email protected]>

* Content-Security-Policy header permit images (*), videos youtube, vimeo

* quick fix for auth

* quick fix for auth

* quick fix for auth

* quick fix for auth

* quick fix for auth

* Force controls csv to download to browser

* quick fix for auth

* quick fix for auth

* test

* test

* test

* test

* test

* last fix and vuln update for django

* last fix and vuln update for django

* 'Back' link to question to take user to previous question (#1612)

* 'Back' link to question to take user to previous question

* Update guidedmodules/views.py

Refactor pulling back_url into project_form

Co-authored-by: davidpofo <[email protected]>

* Improve back-button styling

Co-authored-by: Greg Elin <Greg Elin>
Co-authored-by: davidpofo <[email protected]>
Co-authored-by: Greg Elin <[email protected]>

* WIP: Side-by-side comparison of components (#1620)

* created checkbox and form for submitting components for comparison. created rough start for displaying differences between prime component and rest

* for now just implementing two comparison

* click to read full text after 50 chars

* styling and added Control part

* displaying comparisons for x number of component statements against the prime component. Styling and abstracted out the comparison block into an included template

* check for pid

* removing detail/summary not really necessary

* Condense comparison listings into rows of a single table

Co-authored-by: Greg Elin <Greg Elin>

* Update CHANGELOG.md

* Rename 'compare' column to 'select' in component library (#1626)

Co-authored-by: Greg Elin <Greg Elin>

* Remove portfolio selection modal from Start a Project process

Start projects in user's default portfolio to reduce the clicks
starting a project.

Use the User.create_default_portfolio_if_missing method consistently
to consistently create the user portfolio default portfolio.

Remove the PortfolioSignupForm because registration is no longer
used in registering users.

Remove the '"project_form": AddProjectForm(request.user ...' passed
into many templates because the navbar start app option no longer
brings up the portfolio select modal.

* Update tests for default portfolio

* Bump VERSION, CHANGELOG

* Update CHANGELOG VERSION

* Add button, form to add AppSource via upload of zip file

Add button, form to App Store to provide front-end UI for
admininstrators to add an AppSource by uploading a zip file.
This simplifies setting up an AppSource for first time users.

Implementation only validates that uploaded directory is a zip
file, does not check if uploaded zip file is valid AppSource
directory structure.

Implementation assumes apps are in the 'apps' directory.

* Link to library version of component from a system's selected control component

* Display systems using a component (#1618)

* Display systems using a component

Add method controls.element.consuming_systems to produce list of systems
consuming (e.g., containing) the element.

Add tab to component library component detail page to display list of systems
containing the component.

Also, always display OSCAL tab in component library for component detail
(rather than conditional on 'enable_experimental_opencontrol' parameter).

* Show component system count in tab, better projects.exists query

* Replace list compression with query filters

Co-authored-by: Greg Elin <[email protected]>
Co-authored-by: davidpofo <[email protected]>

* Ge/fulltext search (#1631)

* Add fulltext smt search to component library search

* Note fulltext search in CHANGELOG

Co-authored-by: Greg Elin <Greg Elin>

* check if we are in a portfolio when starting a project. If so then use that portfolio and not the default for the user.

* fixed a bug where Elements of type system were shown in the selected components for a project

* Addressing github issue 1630 in group id matching. fixed a bug where Elements of type system were shown in the selected components for a project.

* Add YAML intermediary file for CMMC

* try/except to still do the component search for non-Postgres users. (#1633)

* Add a 'blank' project with no questions useful for batch project creation (#1634)

Co-authored-by: Greg Elin <Greg Elin>

* td not th

* Polish security objective ui

* Avoid errors when project has no root_task set

* Better project name when no root task set

* Align project name when listing project with no root task

* Support CMMC ver 1 OSCAL catalog

* Fix typo

* Add 'blank' compliance app to first_run

* Append '-dev' to version number

* Legacy Statements added as statements for import

* Updating regex

* Del size limit on speedyssp img upload

* updated column for imp statements

* Fix test shipped catalogs count

* td not th

* Revert "td not th"

This reverts commit e7e8b9c

* these values are safe

* removing extra differences obj.

* safe and efficiency

* adding select/deselect all. checkbox container wrap.

* control structure for compare button toggle

* Maintain sort order of compare_list otherwise Django will order ascending

* adding change component button to change what the prime component of comparison is. Still has work todo

* changed to allow user passed in for parsing

* remove commented out code from template

* Add UI for legacy statement display. Also fix StatementTypeEnum. (#1644)

* [WIP] UI to display legacy control impl smts

Create a conditional display of legacy control implementation
statements in control editor page.

Also widen width of display of editor control statements
to 1250px.

* Improve display of legacy statement

* StatementTypeEnum fixes. Closes #1643

Set all `StatementTypeEnum.<LABEL>.value` to `StatementTypeEnum.<LABEL>.name`
in order for relevant label/term to show up in Django database admin interface.

Set component library detail page Systems tab to not be inactive thereby
removing the content from the System tab showing up on the Control Implementation
Statements tab.

Update controls.tests.

Co-authored-by: Greg Elin <[email protected]>

* Update CHANGELOG

* fixing styling of portfolio table

* using django guardian ObjectPermissionChecker to prefetch permissions. Directly check permissions, to avoid N+1 query of perms with get_obj_perms

* hide_registration revert

* formatting for sid

* Use StatementEnum.*.name value

* removing change component comparison button for now.

* implemented persistent storage of checks by changing value in hidden input with jquery. Clears storage after clicking compare button or deselecting all

* Use one import record for entire file

* More OSCALize id fixes. Proper Create/Update/Del of smts

* Display other_statement count on confirm import delete

* test test_portfolio_projects

* Add project, system.root_element to import_record

Add project and system.root_element to import_record
in order to auto delete the project and system (and root element)
when the import process for importing legacy control impl smts
also creates the project.

* name not value for statement enums

* Sometimes there are not parameters and that is okay it is caught by the try/except block.

* Captialize mission for test

* name not value for enums

* captial impact... Impact

* testing parse for version

* missed one get

* is_prerelease not dev release

* using is_prerelease works for checking dev

* need to force login as authenticated user and then reset login

* url

* snyk update to avoid SQL injection vuln found in Django 3.2.4

* check if previously checked and if so then don't hide compare button.

* fix conflicting migrations detected

* systems-security-sensitivity-level

* Fix controls/0052 StatementTypeEnum migration (#1648)

Co-authored-by: Greg Elin <Greg Elin>

* Update CHANGELOG.md (#1647)

* Configure users on install

New govready_users parameter in local/environment.json
to create sample users on install.

* Add Wazuh collection form to Assessments page (#1651)

* Add Wazuh info via end-user form

* Create SecurityService class to represent Security Service

Create `sec_srvc.SecurityService` class to represent a security service
from which data could be collected.

Add form to Assessments page to collect info from Wazuh SecurityService.

* Fix sec_srvc.py

* Abstracted and made a few improvements

* Fix uuid error

* Fix testing for fields

Co-authored-by: Greg Elin <[email protected]>
Co-authored-by: Alexander Ward <[email protected]>

* checking for dev user creation pw. Create reg users not admin.

* Add CMMC baselines, assign baselines (#1649)

* Improve CMMC links, add OSCAL methods for link content

Improve CMMC catalog links to link to NIST 800-53 in GovReady.

Add methods to OSCAL catalog, get control_part, guidance links

Add get_control_part_by_name, get_control_guidance_links,
get_guidance_related_links_by_value_in_href, and
get_guidance_related_links_text_by_value_in_href to make getting
link content easier.

* Display related controls as links in control guidance

* Properly assign CMMC baselines

* Remove debugging print statements

* Fix typo

* Properly use StatementTypeEnum when saving smts

Co-authored-by: Greg Elin <[email protected]>
Co-authored-by: Greg Elin <Greg Elin>

* first_run finishing touch

* Fix assessment summary link to wazuh (#1653)

* Add Wazuh info via end-user form

* Create SecurityService class to represent Security Service

Create `sec_srvc.SecurityService` class to represent a security service
from which data could be collected.

Add form to Assessments page to collect info from Wazuh SecurityService.

* Fix sec_srvc.py

* Abstracted and made a few improvements

* Fix uuid error

* Fix testing for fields

* Fix assessment summary link to wazuh

Co-authored-by: Greg Elin <[email protected]>
Co-authored-by: Alexander Ward <[email protected]>

* Da/dropnfill (#1654)

* adding drag and fill for component import

* client-side filetype checking

* client-side file size checking 5MB max. missing div.

* missing listed catalog for CMMC. Delete extra migration. adjust test for drag-n-fill. import_project_submit for import project view/test

* json_content not id_file

* spelling

* del

* hot fix for external catalogs

* Remove baseline controls based on control's catalog_key. Fixes failure to remove controls from another catalog when resetting baselines. (#1655)

Co-authored-by: Greg Elin <[email protected]>

* Update SpeedSSP ssp template for multiple catalogs (#1656)

Co-authored-by: Greg Elin <[email protected]>

* Update admin.py

Readded AppInput

* new line fix

* fixed bug where id doesn't exist yet in db

* pinning jinja and installing compliance-trestle with unpinning of markupsafe in order to validate oscal 1.0.0.

* revert some changes to ssp-def for now and added source to statement model.

* ensure source and uuid are added correctly to import/export of component-definition

* adding oscal-version to element model. clean up output title

* replaced component-defintion for test data with official oscal 1.0.0 example component

* lowering the amount of output for circleci

* fixing exceptions

* update example link

* WIP on updating system-security-plan definition.

* missing uuid and 2 removed

* SSP export now includes the uuid and version from the root_element object

* fixing up some module logic to include real uuid and oscal version

* uuid not id

* Import component control statement even if catalog not found

Remove test to see of control exists in control catalog when importing
a component because we may import a component that has controls from
a catalog that is not yet in GovReady.

The test on the control id being in the catalog was originally done
to avoid importing a bad control (or bad control id). But it is worse
to not be able to import the component because we don't yet have the
a control catalog mentioned by the component.

* Display component, smts even if catalog missing

Do not crash on displaying a component and component
control implementation statement when a catalog
associated with statements is missing from GovReady-Q
install.

* Ge/data grid question (#1667)

* Improve appearance of data grid question

Display datagrid question wider and with smaller fonts.
Support datgrid specifying select type cell.

* Display legacy smt in system selected controls; fix component counts

Fix count on project system's components associated with a control (avoid double counting).

Display existance of legacy statement in project system's selected controls.

* Fix typo

Co-authored-by: Greg Elin <[email protected]>

* Adjust test to new rule of importing controls with bad catalog

* natsorting implementation statements before grouping by sid. Providing statement's uuid. Ensuring
all requirements are added to final control implementation dictionary. dealing with empty or incorrect catalogs. Fixed up smt parse.

* If no statements are created then delete empty component

* OSCAL SSP almost implemented just need to finish implemented reqs.

* OSCALSystemSecurityPlanSerializer is getting there still have some fields to fill.

* added new function OSCAL_ssp_export in order to export a system's security plan in OSCAL, this replaces the usual JSON export. Added a several fields of data for OSCAL SSP.

* Coverage 6.0b1 starts to use a modern hash algorithm (sha256) when fingerprinting for high-security environments. Updating the requirements and Changelog.

* added a proxy for parties and responsible parties for component oscal export

* a couple tweaks for comp

* todo for ssp validation with trestle

* read validate ssp  with trestle

* revert discussion test change.

* extra file

* test_path?

* delete invalid test files

* fixing up implementation and testing of validation of extension.

* remove extra addition for .doc

* var sleep?

* bad test case

* adding some comments. Better logging/messaging for schema. Fixed splitting of control id.

* Added test of OSCAL ssp export.

* explicitly login as user

* test baseline json file should be test_baselines.json not baseline

* create system

* check part correctly. test fix

* get system from self

* try/except

* last try

* done

* Remove duplicate loads of select2 in base.html

* Da/cleanup export sspcsv (#1674)

* created a de-oscalizing function and applied to selected controls on import.

* some super route way of making sure the catalogs produce 3.1.

* add test of de_oscalize_control

* Da/discussion updates (#1675)

* get task for the attached object. added some more notes

* fixed signature for getting a task for the attached object.

* Added get_discussion_autocompletes changes to get the organization from the discussion(should be 1 to 1). If another models wants to implement it they can look at the get_discussion_autocompletes in TaskAnswer

* Added get_discussion_autocompletes changes to get the organization from the discussion(should be 1 to 1). If another models wants to implement it they can look at the get_discussion_autocompletes in TaskAnswer

* addressing shell script issues for dockerfile_exec_gunicorn.sh from GH issues 1659 (#1670)

Co-authored-by: root <[email protected]>

* Some controls have characters that we currently don't expect. However it shouldn't just return an empty string as this errors out. Also the controls are already in oscal format. (#1676)

* Auto-start a particular project (#1640)

* [WIP] Auto-start a particular project

Enable the auto-starting of a particular project, e.g.,
start a project without having to visit the compliance app
store.

Eventually, start a project and launch the first question.

* Rename auto start app variables

* Finish project auto start, auto start question, redirect actions

Complete work on automatically starting a project defined in
by a System Setting.

Allow the auto start system setting to define the project template
(e.g. compliance app) to use and even the module to automatically
start.

Also add new question actions to redirect to project page
or project component after answering a question.

TO DO:
- Add error handling (e.g., missing/incorrect project)
- Add tests

* Bump VERSION

Co-authored-by: Greg Elin <Greg Elin>
Co-authored-by: Greg Elin <[email protected]>

* Move Catalog data into database, faster control select autocomplete (#1673)

* Move Catalog data into database, faster control select autocomplete

Move control catalog data into database so catalogs shipped
with GovReady and user added catalogs all accessed in database.
Also faster reading in containers.
Adjust first_run to load catalogs.
Speed up auocomplete by only selecting matching controls.

* Fix tests: add catalogs to db in setup

* Read baselines from database

* Sync with 0.9.7

* Fix tests

* Fix tests 2

* Use better DB query instead of closure

Co-authored-by: Greg Elin <[email protected]>

* Ge/mvp july 2021 (#1679)

* [WIP] Auto-start a particular project

Enable the auto-starting of a particular project, e.g.,
start a project without having to visit the compliance app
store.

Eventually, start a project and launch the first question.

* Rename auto start app variables

* Provide selected ocmponent data to project page

* Finish project auto start, auto start question, redirect actions

Complete work on automatically starting a project defined in
by a System Setting.

Allow the auto start system setting to define the project template
(e.g. compliance app) to use and even the module to automatically
start.

Also add new question actions to redirect to project page
or project component after answering a question.

TO DO:
- Add error handling (e.g., missing/incorrect project)
- Add tests

* Add route for single system-component-control

* issue in bootstrap

* better styling for the drag-n-fill area

* missing carrot. reverting the removal of project import form done in 29ff7c2

* Fix skip buttons and question action crash

* Update CHANGELOG.md

* Update CHANGELOG.md

* Produce dictionary of producer_element control impl smts

* Templatetags for math

* Calculate dict of element ctl impl smts status

Co-authored-by: Greg Elin <Greg Elin>
Co-authored-by: Greg Elin <[email protected]>
Co-authored-by: davidpofo <[email protected]>

* Bump version to 0.9.8

* Remove commented-out, unneeded tests

Co-authored-by: davidpofo <[email protected]>
Co-authored-by: Greg Elin <[email protected]>
Co-authored-by: alexanderward <[email protected]>
Co-authored-by: Greg Elin <Greg Elin>
Co-authored-by: mike <[email protected]>
Co-authored-by: Mike Guelfi <[email protected]>
Co-authored-by: Ward <[email protected]>
Co-authored-by: root <[email protected]>
  • Loading branch information
8 people authored Aug 9, 2021
1 parent 61667d7 commit 9114234
Show file tree
Hide file tree
Showing 26 changed files with 475 additions and 229 deletions.
25 changes: 23 additions & 2 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,24 @@
GovReady-Q Release Notes
========================

v0.9.8 (August 09, 2021)
------------------------

**Developer changes**

* Add SystemSettings `auto_start_project` to permit the automatic start of a particular project and automatic start of a question.
* Add questions actions to redirect to project home page or project components.
* Support auto start of project via global System Setting.
* Create new route for displaying a single system component control.
* New controls.models.System property producer_elements_control_impl_smts_dict to get dictionary of control implementation statements associated with a system element.
* New controls.models.System property producer_elements_control_impl_smts_status_dict to get dictionary of status of control implementation statements associated with a system element.

**Data changes**

* Add JSONfield `value` to SystemSettings model to support specific detail values.
* Retrieve Catalog data from database instead of file system with new controls.models.CatalogData model.


v0.9.7 (August 06, 2021)
------------------------

Expand All @@ -12,7 +30,7 @@ v0.9.7 (August 06, 2021)
**Developer changes**

* Support datagrid specifying select type cell.
* Added new function OSCAL_ssp_export in order to export a system's security plan in OSCAL, this replaces the usual JSON export. Added a several fields of data for OSCAL SSP.
* Added new function OSCAL_ssp_export in order to export a system's security plan in OSCAL, this replaces the usual JSON export. Added a several fields of data for OSCAL SSP.
* If a component to be imported has a catalog key that is not found in the internal or external catalog list then it will be skipped and logged.
* If no statements are created the resulting element/component is deleted.
* Component and System Security Plan exports pass OSCAL 1.0.0 schema validation.
Expand All @@ -35,6 +53,7 @@ v0.9.6 (July 15, 2021)
* Added a Select/Deselect button for component comparison choice.
* Add accordion to assessment page to provide information on getting data from Wazuh.
* Add form to Assessments page to collect Wazuh information.
* Support auto start of project via global System Setting.

**Bug fixes**

Expand All @@ -45,13 +64,15 @@ v0.9.6 (July 15, 2021)
* Add custom Django command to batch import legacy control implementation statements from legacy SSPs Excel spreadsheet exports. Currently supports CSAM.
* Added missing unit test for portfolio project endpoint.
* Add `sec_srvc.SecurityService` class to represent a security service from which data could be collected.
* Add SystemSettings `auto_start_project` to permit the automatic start of a particular project and automatic start of a question.
* Add questions actions to redirect to project home page or project components.

**Data changes**

* Set all `StatementTypeEnum.<LABEL>.value` to `StatementTypeEnum.<LABEL>.name` in order for relevant label/term to show up in Django database admin interface.
* Create baselines for CMMC catalog.
* Fisma impact level is now represented as Security Sensitivity level following OSCAL's schema.

* Add JSONfield `value` to SystemSettings model to support specific detail values.

v0.9.5 (June 23, 2021)
----------------------
Expand Down
2 changes: 1 addition & 1 deletion VERSION
Original file line number Diff line number Diff line change
@@ -1 +1 @@
v0.9.7
v0.9.8
6 changes: 5 additions & 1 deletion controls/admin.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
from django.contrib import admin
from django.http import HttpResponse
from .models import ImportRecord, Statement, Element, ElementControl, ElementRole, System, CommonControlProvider, CommonControl, ElementCommonControl, Poam, Deployment, SystemAssessmentResult
from .oscal import CatalogData
from guardian.admin import GuardedModelAdmin
from simple_history.admin import SimpleHistoryAdmin

Expand Down Expand Up @@ -89,6 +90,9 @@ class SystemAssessmentResultAdmin(admin.ModelAdmin):
list_display = ('id', 'name', 'system', 'deployment', 'uuid')
search_fields = ('id', 'name', 'system', 'deployment', 'uuid')

class CatalogDataAdmin(admin.ModelAdmin):
list_display = ('catalog_key',)
search_fields = ('catalog_key',)

admin.site.register(ImportRecord, ImportRecordAdmin)
admin.site.register(Statement, StatementAdmin)
Expand All @@ -102,4 +106,4 @@ class SystemAssessmentResultAdmin(admin.ModelAdmin):
admin.site.register(Poam, PoamAdmin)
admin.site.register(Deployment, DeploymentAdmin)
admin.site.register(SystemAssessmentResult, SystemAssessmentResultAdmin)

admin.site.register(CatalogData, CatalogDataAdmin)
33 changes: 33 additions & 0 deletions controls/migrations/0057_catalogdata.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
# Generated by Django 3.2.4 on 2021-08-05 02:12

from django.db import migrations, models
import django.db.models.manager
import jsonfield.fields


class Migration(migrations.Migration):

dependencies = [
('controls', '0056_element_oscal_version'),
]

operations = [
migrations.CreateModel(
name='CatalogData',
fields=[
('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
('catalog_key', models.CharField(help_text='Unique key for catalog', max_length=100, unique=True)),
('catalog_json', jsonfield.fields.JSONField(blank=True, help_text='JSON object representing the OSCAL-formatted control catalog.', null=True)),
('created', models.DateTimeField(auto_now_add=True, db_index=True)),
('updated', models.DateTimeField(auto_now=True, db_index=True)),
],
options={
'abstract': False,
'base_manager_name': 'prefetch_manager',
},
managers=[
('objects', django.db.models.manager.Manager()),
('prefetch_manager', django.db.models.manager.Manager()),
],
),
]
19 changes: 19 additions & 0 deletions controls/migrations/0058_catalogdata_baselines_json.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
# Generated by Django 3.2.4 on 2021-08-05 15:19

from django.db import migrations
import jsonfield.fields


class Migration(migrations.Migration):

dependencies = [
('controls', '0057_catalogdata'),
]

operations = [
migrations.AddField(
model_name='catalogdata',
name='baselines_json',
field=jsonfield.fields.JSONField(blank=True, help_text='JSON object representing the baselines for the catalog.', null=True),
),
]
95 changes: 40 additions & 55 deletions controls/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,14 +15,13 @@
from controls.enums.components import ComponentTypeEnum, ComponentStateEnum
from siteapp.model_mixins.tags import TagModelMixin
from controls.enums.statements import StatementTypeEnum
from controls.oscal import Catalogs, Catalog, check_and_extend
from controls.oscal import Catalogs, Catalog, CatalogData
import uuid
import tools.diff_match_patch.python3 as dmp_module
from copy import deepcopy
from django.db import transaction

BASELINE_PATH = os.path.join(os.path.dirname(__file__),'data','baselines')
EXTERNAL_BASELINE_PATH = os.path.join(f"{os.getcwd()}",'local', 'controls', 'data', 'baselines')
ORGPARAM_PATH = os.path.join(os.path.dirname(__file__),'data','org_defined_parameters')

class ImportRecord(models.Model):
Expand Down Expand Up @@ -582,7 +581,6 @@ def get_security_impact_level(self):
except Statement.DoesNotExist:
return {}


@property
def smts_common_controls_as_dict(self):
common_controls = self.root_element.common_controls.all()
Expand Down Expand Up @@ -737,6 +735,35 @@ def get_producer_elements(self):

producer_elements = cached_property(get_producer_elements)

def get_producer_elements_control_impl_smts_dict(self):
smts = self.root_element.statements_consumed.filter(statement_type=StatementTypeEnum.CONTROL_IMPLEMENTATION.name)
components_smts = {}
for smt in smts:
if smt.producer_element:
# components.add(smt.producer_element)
if smt.producer_element in components_smts:
components_smts[smt.producer_element].append(smt)
else:
components_smts[smt.producer_element] = [smt]
return components_smts

producer_elements_control_impl_smts_dict = cached_property(get_producer_elements_control_impl_smts_dict)

def get_producer_elements_control_impl_smts_status_dict(self):
components_smts = self.producer_elements_control_impl_smts_dict
components_smts_status = {}
for component in components_smts.keys():
cmpt_smts_status = {}
for smt in components_smts[component]:
if smt.status in cmpt_smts_status:
cmpt_smts_status[smt.status] += 1
else:
cmpt_smts_status[smt.status] = 1
components_smts_status[component] = cmpt_smts_status
return components_smts_status

producer_elements_control_impl_smts_status_dict = cached_property(get_producer_elements_control_impl_smts_status_dict)

def set_component_control_status(self, element, status):
"""Batch update status of system control implementation statements for a specific element."""

Expand Down Expand Up @@ -783,9 +810,7 @@ class Baselines (object):
def __init__(self):

self.file_path = BASELINE_PATH
self.external_file_path = EXTERNAL_BASELINE_PATH
self.baselines_keys = self._list_keys()
# self.index = self._build_index()

# Usage
# from controls.models import Baselines
Expand All @@ -796,44 +821,18 @@ def __init__(self):
# # Returns ['ac-1', 'ac-2', 'ac-2.1', 'ac-2.2', ...]
# bs.get_baseline_controls('NIST_SP-800-53_rev4', 'moderate')

def _list_files(self):
return self.extend_external_baselines([
'NIST_SP-800-53_rev4_baselines.json',
# 'NIST_SP-800-53_rev5_baselines.json',
'NIST_SP-800-171_rev1_baselines.json',
'CMMC_ver1_baselines.json'
], "files")


def _list_keys(self):
return self.extend_external_baselines([
'NIST_SP-800-53_rev4',
# 'NIST_SP-800-53_rev5',
'NIST_SP-800-171_rev1',
'CMMC_ver1'
], "keys")

# TODO: only return keys for records that have baselines?
return list(CatalogData.objects.order_by('catalog_key').values_list('catalog_key', flat=True).distinct())

def _load_json(self, baselines_key):
"""Read baseline file - JSON"""
# TODO Escape baselines_key
self.data_file = baselines_key + "_baselines.json"
data_file = os.path.join(self.file_path, self.data_file)
# Does file exist?
if not os.path.isfile(data_file):
# Check if there any external oscal baseline files
try:
data_file = os.path.join(self.external_file_path, self.data_file)
except:
print("ERROR: {} does not exist".format(data_file))
return False
# Load file as json
try:
with open(data_file, 'r') as json_file:
data = json.load(json_file)
return data
except:
print("ERROR: {} could not be read or could not be read as json".format(data_file))

catalog_record = CatalogData.objects.get(catalog_key=baselines_key)
baselines = catalog_record.baselines_json
if baselines:
return baselines
else:
return False

def get_baseline_controls(self, baselines_key, baseline_name):
Expand All @@ -849,22 +848,6 @@ def get_baseline_controls(self, baselines_key, baseline_name):
print("Requested baseline name not found in baselines_key data file")
return False

@property
def body(self):
return self.legacy_imp_smt


def extend_external_baselines(self, baseline_info, extendtype):
"""
Add external baselines to list of baselines
"""
os.makedirs(EXTERNAL_BASELINE_PATH, exist_ok=True)
external_baselines = [file for file in os.listdir(EXTERNAL_BASELINE_PATH) if
file.endswith('.json')]

baseline_info = check_and_extend(baseline_info, external_baselines, extendtype, "_baselines")
return baseline_info

class OrgParams(object):
"""
Represent list of organizational defined parameters. Temporary
Expand Down Expand Up @@ -1016,3 +999,5 @@ def __repr__(self):
# # For debugging.
# return "<AssesmentResult %s id=%d>" % (self.statement, self.id)



Loading

0 comments on commit 9114234

Please sign in to comment.