CVE-2021-41206: Tensorflow Incomplete Validation Of Shapes Vulnerability

Q: What is CVE-2021-41206?

CVE-2021-41206 is a security vulnerability in Tensorflow involving Incomplete validation of shapes in multiple TF ops under CWE-354: Improper Validation of Integrity Check Value.

Q: Which Tensorflow versions are affected?

Affected versions include releases prior to 2.6.1, 2.5.2, 2.4.4 across affected packages: tensorflow, tensorflow-cpu, tensorflow-gpu.

Q: What version fixes CVE-2021-41206?

Upgrading to 2.6.1, 2.5.2, 2.4.4 (or the latest patched release for your respective branch) resolves the vulnerability.

Q: Is tensorflow affected?

Yes, tensorflow is affected in releases prior to 2.6.1, 2.5.2, 2.4.4. Upgrade your dependency to 2.6.1, 2.5.2, 2.4.4 or later.

Q: Is tensorflow-cpu affected?

Yes, tensorflow-cpu is affected in releases prior to 2.6.1, 2.5.2, 2.4.4. Upgrade your dependency to 2.6.1, 2.5.2, 2.4.4 or later.

Q: What is the CVSS score of CVE-2021-41206?

CVE-2021-41206 has a CVSS v3.1 base score of 7.0 (HIGH) with vector string CVSS:3.1/AV:L/AC:H/PR:L/UI:N/S:U/C:H/I:H/A:H.

Q: What CWE is associated with CVE-2021-41206?

CVE-2021-41206 is classified under CWE-354: Improper Validation of Integrity Check Value (The product does not validate or incorrectly validates the integrity check values or "checksums" of a message. This may prevent it from detecting if the data has been modified or corrupted in transmission.).

Q: How can I check PyPI dependencies for CVE-2021-41206?

Run the dependency tree command: 'pip show tensorflow'. This command lists direct and transitive dependencies in your build configuration so you can verify if a vulnerable version is resolved.

HIGH

Advisory Identifier: GHSA-pgcq-h79j-2f69

CVSS 7.0

CWE-354

Published 2021-11-10

Advisory Summary

CVE-2021-41206 is a HIGH (CVSS 7.0) incomplete validation of shapes vulnerability (CWE-354: Improper Validation of Integrity Check Value) affecting Tensorflow libraries in the PyPI ecosystem. The issue involves incomplete validation of shapes in multiple tf ops. Remediation is available by upgrading affected packages to version 2.6.1, 2.5.2, 2.4.4.

CVE-2021-41206 at a Glance

CVE Identifier CVE-2021-41206

GHSA Identifier GHSA-pgcq-h79j-2f69

Severity HIGH

CVSS v3.1 Base Score 7.0

CWE Classification CWE-354: Improper Validation of Integrity Check Value

Affected Product Tensorflow

Package Ecosystem PyPI

Published Date 2021-11-10

Last Updated 2024-11-13

Security Impact

CVSS: 7.0

The security impact metrics below are parsed directly from the CVSS v3.1 vector string (CVSS:3.1/AV:L/AC:H/PR:L/UI:N/S:U/C:H/I:H/A:H).

Attack Vector Local (AV:L)

Attack Complexity High (AC:H)

Privileges Required Low (PR:L)

User Interaction None (UI:N)

Scope Unchanged (S:U)

Confidentiality High (C:H)

Integrity High (I:H)

Availability High (A:H)

How to Fix CVE-2021-41206

To remediate this vulnerability, upgrade all affected Tensorflow dependencies to a patched release. Both direct and transitive dependencies should be checked, as frameworks or security libraries may pull in vulnerable versions transitively.

tensorflow fixed: 2.6.1

pip install tensorflow>=2.6.1

tensorflow fixed: 2.5.2

pip install tensorflow>=2.5.2

tensorflow fixed: 2.4.4

pip install tensorflow>=2.4.4

tensorflow-cpu fixed: 2.6.1

pip install tensorflow-cpu>=2.6.1

tensorflow-cpu fixed: 2.5.2

pip install tensorflow-cpu>=2.5.2

tensorflow-cpu fixed: 2.4.4

pip install tensorflow-cpu>=2.4.4

tensorflow-gpu fixed: 2.6.1

pip install tensorflow-gpu>=2.6.1

tensorflow-gpu fixed: 2.5.2

pip install tensorflow-gpu>=2.5.2

tensorflow-gpu fixed: 2.4.4

pip install tensorflow-gpu>=2.4.4

Verify Direct and Transitive Dependencies

Run the following command in your project directory to verify that all resolved dependencies use the patched release:

pip show tensorflow

How to Check if You Are Affected

To determine whether your Python environment is affected by CVE-2021-41206, inspect your installed packages using pip:

pip show tensorflow

Analyze the terminal output to locate any version of Tensorflow in your dependency tree. If any resolved version matches the vulnerable ranges listed in the table below (for example, versions earlier than 2.6.1), your application is affected.

Note: The pip show tensorflow command checks currently installed packages in your active Python environment.

Technical Details

Vulnerability Mechanism & Technical Analysis

This vulnerability is classified under CWE-354: Improper Validation of Integrity Check Value, where the product does not validate or incorrectly validates the integrity check values or "checksums" of a message. this may prevent it from detecting if the data has been modified or corrupted in transmission. In Tensorflow, software state validation or input boundaries are circumvented during processing. Review the official source advisory statement below for specific affected components, classes, and reproduction details.

Source Advisory Statement

### Impact Several TensorFlow operations are missing validation for the shapes of the tensor arguments involved in the call. Depending on the API, this can result in undefined behavior and segfault or `CHECK`-fail related crashes but in some scenarios writes and reads from heap populated arrays are also possible. We have discovered these issues internally via tooling while working on improving/testing GPU op determinism. As such, we don't have reproducers and there will be multiple fixes for these issues. ### Patches We have patched the issue in GitHub commits [68422b215e618df5ad375bcdc6d2052e9fd3080a](https://github.com/tensorflow/tensorflow/commit/68422b215e618df5ad375bcdc6d2052e9fd3080a), [4d74d8a00b07441cba090a02e0dd9ed385145bf4](https://github.com/tensorflow/tensorflow/commit/4d74d8a00b07441cba090a02e0dd9ed385145bf4), [579261dcd446385831fe4f7457d802a59685121d](https://github.com/tensorflow/tensorflow/commit/579261dcd446385831fe4f7457d802a59685121d), [da4aad5946be30e5f049920fa076e1f7ef021261](https://github.com/tensorflow/tensorflow/commit/da4aad5946be30e5f049920fa076e1f7ef021261), [4dddb2fd0b01cdd196101afbba6518658a2c9e07](https://github.com/tensorflow/tensorflow/commit/4dddb2fd0b01cdd196101afbba6518658a2c9e07), and [e7f497570abb6b4ae5af4970620cd880e4c0c904](https://github.com/tensorflow/tensorflow/commit/e7f497570abb6b4ae5af4970620cd880e4c0c904). These fixes will be included in TensorFlow 2.7.0. We will also cherrypick these commits on TensorFlow 2.6.1, TensorFlow 2.5.2, and TensorFlow 2.4.4, as these are also affected and still in supported range. ### For more information Please consult [our security guide](https://github.com/tensorflow/tensorflow/blob/master/SECURITY.md) for more information regarding the security model and how to contact us with issues and questions.

CVE-2021-41206 Affected Tensorflow Versions

9 component(s)

Fixed Versions:

2.6.12.5.22.4.4

Product / Package	Ecosystem	Affected Versions	Fixed Version
tensorflow pypi/tensorflow	PyPI	≥ 2.6.0 < 2.6.1	2.6.1
tensorflow pypi/tensorflow	PyPI	≥ 2.5.0 < 2.5.2	2.5.2
tensorflow pypi/tensorflow	PyPI	< 2.4.4	2.4.4
tensorflow-cpu pypi/tensorflow-cpu	PyPI	≥ 2.6.0 < 2.6.1	2.6.1
tensorflow-cpu pypi/tensorflow-cpu	PyPI	≥ 2.5.0 < 2.5.2	2.5.2
tensorflow-cpu pypi/tensorflow-cpu	PyPI	< 2.4.4	2.4.4
tensorflow-gpu pypi/tensorflow-gpu	PyPI	≥ 2.6.0 < 2.6.1	2.6.1
tensorflow-gpu pypi/tensorflow-gpu	PyPI	≥ 2.5.0 < 2.5.2	2.5.2
tensorflow-gpu pypi/tensorflow-gpu	PyPI	< 2.4.4	2.4.4

References

GitHub Advisory / Wiki ADVISORY

https://github.com/tensorflow/tensorflow/security/advisories/GHSA-pgcq-h79j-2f69

NVD — CVE-2021-41206 NVD

https://nvd.nist.gov/vuln/detail/CVE-2021-41206

Security Fix Commit FIX COMMIT

https://github.com/tensorflow/tensorflow/commit/4d74d8a00b07441cba090a02e0dd9ed385145bf4

Security Fix Commit FIX COMMIT

https://github.com/tensorflow/tensorflow/commit/4dddb2fd0b01cdd196101afbba6518658a2c9e07

Security Fix Commit FIX COMMIT

https://github.com/tensorflow/tensorflow/commit/579261dcd446385831fe4f7457d802a59685121d

Security Fix Commit FIX COMMIT

https://github.com/tensorflow/tensorflow/commit/68422b215e618df5ad375bcdc6d2052e9fd3080a

Security Fix Commit FIX COMMIT

https://github.com/tensorflow/tensorflow/commit/da4aad5946be30e5f049920fa076e1f7ef021261

Security Fix Commit FIX COMMIT

https://github.com/tensorflow/tensorflow/commit/e7f497570abb6b4ae5af4970620cd880e4c0c904

Tensorflow Source Repository REPOSITORY

https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow-cpu/PYSEC-2021-845.yaml

Tensorflow Source Repository REPOSITORY

https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow-gpu/PYSEC-2021-847.yaml

Tensorflow Source Repository REPOSITORY

https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow/PYSEC-2021-843.yaml

Tensorflow Source Repository REPOSITORY

https://github.com/tensorflow/tensorflow

Frequently Asked Questions

What is CVE-2021-41206?

CVE-2021-41206 is a security vulnerability in Tensorflow involving Incomplete validation of shapes in multiple TF ops under CWE-354: Improper Validation of Integrity Check Value.

Which Tensorflow versions are affected?

Affected versions include releases prior to 2.6.1, 2.5.2, 2.4.4 across affected packages: tensorflow, tensorflow-cpu, tensorflow-gpu.

What version fixes CVE-2021-41206?

Upgrading to 2.6.1, 2.5.2, 2.4.4 (or the latest patched release for your respective branch) resolves the vulnerability.

Is tensorflow affected?

Yes, tensorflow is affected in releases prior to 2.6.1, 2.5.2, 2.4.4. Upgrade your dependency to 2.6.1, 2.5.2, 2.4.4 or later.

Is tensorflow-cpu affected?

Yes, tensorflow-cpu is affected in releases prior to 2.6.1, 2.5.2, 2.4.4. Upgrade your dependency to 2.6.1, 2.5.2, 2.4.4 or later.

What is the CVSS score of CVE-2021-41206?

CVE-2021-41206 has a CVSS v3.1 base score of 7.0 (HIGH) with vector string CVSS:3.1/AV:L/AC:H/PR:L/UI:N/S:U/C:H/I:H/A:H.

What CWE is associated with CVE-2021-41206?

CVE-2021-41206 is classified under CWE-354: Improper Validation of Integrity Check Value (The product does not validate or incorrectly validates the integrity check values or "checksums" of a message. This may prevent it from detecting if the data has been modified or corrupted in transmission.).

How can I check PyPI dependencies for CVE-2021-41206?

Run the dependency tree command: 'pip show tensorflow'. This command lists direct and transitive dependencies in your build configuration so you can verify if a vulnerable version is resolved.

Machine-readable OSV JSON OSV v1.5 Schema

Normalized schema output

View Raw JSON

{
  "schema_version" : "1.4.0",
  "id" : "GHSA-pgcq-h79j-2f69",
  "modified" : "2024-11-13T21:54:40Z",
  "published" : "2021-11-10T19:03:38Z",
  "aliases" : [ "CVE-2021-41206" ],
  "summary" : "Incomplete validation of shapes in multiple TF ops",
  "details" : "### Impact\nSeveral TensorFlow operations are missing validation for the shapes of the tensor arguments involved in the call. Depending on the API, this can result in undefined behavior and segfault or `CHECK`-fail related crashes but in some scenarios writes and reads from heap populated arrays are also possible.\n\nWe have discovered these issues internally via tooling while working on improving/testing GPU op determinism. As such, we don't have reproducers and there will be multiple fixes for these issues.\n\n### Patches\nWe have patched the issue in GitHub commits [68422b215e618df5ad375bcdc6d2052e9fd3080a](https://github.com/tensorflow/tensorflow/commit/68422b215e618df5ad375bcdc6d2052e9fd3080a), [4d74d8a00b07441cba090a02e0dd9ed385145bf4](https://github.com/tensorflow/tensorflow/commit/4d74d8a00b07441cba090a02e0dd9ed385145bf4), [579261dcd446385831fe4f7457d802a59685121d](https://github.com/tensorflow/tensorflow/commit/579261dcd446385831fe4f7457d802a59685121d), [da4aad5946be30e5f049920fa076e1f7ef021261](https://github.com/tensorflow/tensorflow/commit/da4aad5946be30e5f049920fa076e1f7ef021261), [4dddb2fd0b01cdd196101afbba6518658a2c9e07](https://github.com/tensorflow/tensorflow/commit/4dddb2fd0b01cdd196101afbba6518658a2c9e07), and [e7f497570abb6b4ae5af4970620cd880e4c0c904](https://github.com/tensorflow/tensorflow/commit/e7f497570abb6b4ae5af4970620cd880e4c0c904).\n\nThese fixes will be included in TensorFlow 2.7.0. We will also cherrypick these commits on TensorFlow 2.6.1, TensorFlow 2.5.2, and TensorFlow 2.4.4, as these are also affected and still in supported range.\n\n### For more information\nPlease consult [our security guide](https://github.com/tensorflow/tensorflow/blob/master/SECURITY.md) for more information regarding the security model and how to contact us with issues and questions.",
  "severity" : [ {
    "type" : "CVSS_V3",
    "score" : "CVSS:3.1/AV:L/AC:H/PR:L/UI:N/S:U/C:H/I:H/A:H"
  }, {
    "type" : "CVSS_V4",
    "score" : "CVSS:4.0/AV:L/AC:L/AT:P/PR:L/UI:N/VC:H/VI:H/VA:H/SC:N/SI:N/SA:N"
  } ],
  "affected" : [ {
    "package" : {
      "ecosystem" : "PyPI",
      "name" : "tensorflow"
    },
    "ranges" : [ {
      "type" : "ECOSYSTEM",
      "events" : [ {
        "introduced" : "2.6.0"
      }, {
        "fixed" : "2.6.1"
      } ]
    } ]
  }, {
    "package" : {
      "ecosystem" : "PyPI",
      "name" : "tensorflow"
    },
    "ranges" : [ {
      "type" : "ECOSYSTEM",
      "events" : [ {
        "introduced" : "2.5.0"
      }, {
        "fixed" : "2.5.2"
      } ]
    } ]
  }, {
    "package" : {
      "ecosystem" : "PyPI",
      "name" : "tensorflow"
    },
    "ranges" : [ {
      "type" : "ECOSYSTEM",
      "events" : [ {
        "introduced" : "0"
      }, {
        "fixed" : "2.4.4"
      } ]
    } ]
  }, {
    "package" : {
      "ecosystem" : "PyPI",
      "name" : "tensorflow-cpu"
    },
    "ranges" : [ {
      "type" : "ECOSYSTEM",
      "events" : [ {
        "introduced" : "2.6.0"
      }, {
        "fixed" : "2.6.1"
      } ]
    } ]
  }, {
    "package" : {
      "ecosystem" : "PyPI",
      "name" : "tensorflow-cpu"
    },
    "ranges" : [ {
      "type" : "ECOSYSTEM",
      "events" : [ {
        "introduced" : "2.5.0"
      }, {
        "fixed" : "2.5.2"
      } ]
    } ]
  }, {
    "package" : {
      "ecosystem" : "PyPI",
      "name" : "tensorflow-cpu"
    },
    "ranges" : [ {
      "type" : "ECOSYSTEM",
      "events" : [ {
        "introduced" : "0"
      }, {
        "fixed" : "2.4.4"
      } ]
    } ]
  }, {
    "package" : {
      "ecosystem" : "PyPI",
      "name" : "tensorflow-gpu"
    },
    "ranges" : [ {
      "type" : "ECOSYSTEM",
      "events" : [ {
        "introduced" : "2.6.0"
      }, {
        "fixed" : "2.6.1"
      } ]
    } ]
  }, {
    "package" : {
      "ecosystem" : "PyPI",
      "name" : "tensorflow-gpu"
    },
    "ranges" : [ {
      "type" : "ECOSYSTEM",
      "events" : [ {
        "introduced" : "2.5.0"
      }, {
        "fixed" : "2.5.2"
      } ]
    } ]
  }, {
    "package" : {
      "ecosystem" : "PyPI",
      "name" : "tensorflow-gpu"
    },
    "ranges" : [ {
      "type" : "ECOSYSTEM",
      "events" : [ {
        "introduced" : "0"
      }, {
        "fixed" : "2.4.4"
      } ]
    } ]
  } ],
  "references" : [ {
    "type" : "WEB",
    "url" : "https://github.com/tensorflow/tensorflow/security/advisories/GHSA-pgcq-h79j-2f69"
  }, {
    "type" : "ADVISORY",
    "url" : "https://nvd.nist.gov/vuln/detail/CVE-2021-41206"
  }, {
    "type" : "WEB",
    "url" : "https://github.com/tensorflow/tensorflow/commit/4d74d8a00b07441cba090a02e0dd9ed385145bf4"
  }, {
    "type" : "WEB",
    "url" : "https://github.com/tensorflow/tensorflow/commit/4dddb2fd0b01cdd196101afbba6518658a2c9e07"
  }, {
    "type" : "WEB",
    "url" : "https://github.com/tensorflow/tensorflow/commit/579261dcd446385831fe4f7457d802a59685121d"
  }, {
    "type" : "WEB",
    "url" : "https://github.com/tensorflow/tensorflow/commit/68422b215e618df5ad375bcdc6d2052e9fd3080a"
  }, {
    "type" : "WEB",
    "url" : "https://github.com/tensorflow/tensorflow/commit/da4aad5946be30e5f049920fa076e1f7ef021261"
  }, {
    "type" : "WEB",
    "url" : "https://github.com/tensorflow/tensorflow/commit/e7f497570abb6b4ae5af4970620cd880e4c0c904"
  }, {
    "type" : "WEB",
    "url" : "https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow-cpu/PYSEC-2021-845.yaml"
  }, {
    "type" : "WEB",
    "url" : "https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow-gpu/PYSEC-2021-847.yaml"
  }, {
    "type" : "WEB",
    "url" : "https://github.com/pypa/advisory-database/tree/main/vulns/tensorflow/PYSEC-2021-843.yaml"
  }, {
    "type" : "PACKAGE",
    "url" : "https://github.com/tensorflow/tensorflow"
  } ],
  "database_specific" : {
    "cwe_ids" : [ "CWE-354" ],
    "severity" : "HIGH",
    "github_reviewed" : true,
    "github_reviewed_at" : "2021-11-08T22:41:31Z",
    "nvd_published_at" : "2021-11-05T22:15:00Z"
  }
}

CVSS V3

7.0 HIGH

Attack Vector Local (AV:L)

Attack Complexity High (AC:H)

Privileges Required Low (PR:L)

User Interaction None (UI:N)

Scope Unchanged (S:U)

Confidentiality High (C:H)

Integrity High (I:H)

Availability High (A:H)

Vector String

CVSS:3.1/AV:L/AC:H/PR:L/UI:N/S:U/C:H/I:H/A:H

Advisory Metadata

Source ID: GHSA-pgcq-h79j-2f69
Schema Version: 1.4.0
Published: 2021-11-10 19:03
Last Modified: 2024-11-13 21:54
Database Record File: advisories/github-reviewed/2021/11/GHSA-pgcq-h79j-2f69/GHSA-pgcq-h79j-2f69.json