R/inet.intel.R

In r-net-tools/inet.intel:

#' Get threats data.frame
#'
#' @export
UpdateThreats <- function(dstpath = tempdir(), source.db = "all"){
  valid <- c("all", "cc", "mot", "mwd", "ptd", "rsw")
  source.db <- valid[valid %in% source.db]

  if (any(c("all","cc") %in% source.db)) {
    df.cc <- GetCCData()
  } else {
    df.cc <- NewIOC()
  }

  if (any(c("all","mot") %in% source.db)) {
    df.mot <- GetMOTData()
  } else {
    df.mot <- NewIOC()
  }

  if (any(c("all","mwd") %in% source.db)) {
    df.mwd <- GetMWDData()
  } else {
    df.mwd <- NewIOC()
  }

  if (any(c("all","ptd") %in% source.db)) {
    df.ptd <- GetPTDData()
  } else {
    df.ptd <- NewIOC()
  }

  if (any(c("all","rsw") %in% source.db)) {
    df.rsw <- GetRSWData()
  } else {
    df.rsw <- NewIOC()
  }
  #df.tor <- inet.intel::GetTorData()

  # Join data
  print("Join data...")
  df.threats <- rbind(df.cc, df.mot, df.mwd, df.ptd, df.rsw)

  # Save data.frame
  print("Save data...")
  dstfile <- paste(dstpath, "df.threats.rda",
                   sep = ifelse(.Platform$OS.type == "windows", "\\", "/"))
  save(df.threats, file = dstfile)
  print(dstfile)
  dstfile <- paste(dstpath, "threats.csv",
                   sep = ifelse(.Platform$OS.type == "windows", "\\", "/"))
  write.csv(x = df.threats, file = dstfile, quote = FALSE, row.names = FALSE)
  print(dstfile)
}

#' Title
#'
#' @return data.frame
#' @export
NewIOC <- function(){
  df <- data.frame(ioc = character(),
                   type = factor(),
                   source = factor(),
                   timestamp1 = character(),
                   timestamp2 = character(),
                   source.info = character(),
                   stringsAsFactors = FALSE)
  return(df)
}

#' Title
#'
#' @param ioc
#'
#' @return json
#' @export
CheckIOC <- function(ioc = "github.com") {
  selected <- grepl(pattern = ioc, ignore.case = T, x = df.threats$ioc)
  if (any(selected)) {
    selected <- df.threats[selected,]
  } else {
    selected <- NewIOC()
  }
  selected <- jsonlite::toJSON(selected)
  return(selected)
}