コード例 #1
0
        public bool CheckExistingTracking(int cveID)
        {
            // Get existingTrackedCve
            try
            {
                TrackedCve existingTrackedCve = _context.TrackedCves
                                                .Where(tc => tc.AspNetUserID == _userManager.GetUserId(User) &&
                                                       tc.CveID == cveID)
                                                .FirstOrDefault();

                // If it exists return true otherwise false
                if (existingTrackedCve != null)
                {
                    return(true);
                }
                else
                {
                    return(false);
                }
            }
            catch (Exception)
            {
                throw;
            }
        }
コード例 #2
0
        public IActionResult Details(int cveID)
        {
            string userID = _userManager.GetUserId(User);

            if (userID == null || cveID == 0)
            {
                return(NotFound());
            }

            try
            {
                TrackedCve existingTrackedCve = _context.TrackedCves
                                                .Where(etc => etc.AspNetUserID == userID &&
                                                       etc.CveID == cveID)
                                                .Include(etc => etc.Cve)
                                                .ThenInclude(c => c.CveConfigurations)
                                                .ThenInclude(etc => etc.Product)
                                                .Include(etc => etc.Cve)
                                                .ThenInclude(c => c.References)
                                                .FirstOrDefault();

                return(View(existingTrackedCve));
            }
            catch (Exception)
            {
                return(NotFound());
            }
        }
コード例 #3
0
        public StatusCodeResult UpdateTrackedVulnerability(string notes, int cveId)
        {
            // Check for nulls and 0's
            if (notes == null || cveId == 0)
            {
                return(NotFound());
            }

            // get the existingTrackedCve
            try
            {
                TrackedCve existingTrackedCve = _context.TrackedCves
                                                .Where(etc => etc.AspNetUserID == _userManager.GetUserId(User) &&
                                                       etc.CveID == cveId)
                                                .FirstOrDefault();

                // return if null
                if (existingTrackedCve == null)
                {
                    return(NotFound());
                }

                // If notes have changed apply the changes
                if (existingTrackedCve.Notes != notes)
                {
                    existingTrackedCve.Notes = notes;
                    _context.TrackedCves.Update(existingTrackedCve);
                    _context.SaveChanges();
                    return(Ok());
                }
                else
                {
                    return(Ok());
                }
            }
            catch (Exception)
            {
                return(NotFound());
            }
        }
コード例 #4
0
        public IActionResult ChangeVulnerabilityTracking(int cveID)
        {
            string aspNetUserID = _userManager.GetUserId(User);
            // Get existingTrackedCve if it exists
            var existingTrackedCve = _context.TrackedCves
                                     .Where(tc => tc.AspNetUserID == aspNetUserID &&
                                            tc.CveID == cveID)
                                     .FirstOrDefault();

            // It it does exist, remove it
            if (existingTrackedCve != null)
            {
                try
                {
                    _context.TrackedCves.Remove(existingTrackedCve);
                    _context.SaveChanges();
                }
                catch (DbUpdateConcurrencyException e)
                {
                    TempData["Param"] = "Changing Tracking";
                    TempData["Error"] = e.Message;
                    throw e;
                }
                catch (DbUpdateException e)
                {
                    TempData["Param"] = "Changing Tracking";
                    TempData["Error"] = e.Message;
                    throw e;
                }

                // true if added
                return(Json(new { returnValue = "Removed" }));
            }
            else   // If it doesn't exist, add it
            {
                var newTrackedCve = new TrackedCve()
                {
                    AspNetUserID = aspNetUserID,
                    CveID        = cveID
                };
                try
                {
                    _context.Add(newTrackedCve);
                    _context.SaveChanges();
                }
                catch (DbUpdateConcurrencyException e)
                {
                    TempData["Param"] = "Changing Tracking";
                    TempData["Error"] = e.Message;
                    throw e;
                }
                catch (DbUpdateException e)
                {
                    TempData["Param"] = "Changing Tracking";
                    TempData["Error"] = e.Message;
                    throw e;
                }

                // false if removed
                return(Json(new { returnValue = "Added" }));
            }
        }