# frozen_string_literal: true # # ronin-payloads - A Ruby micro-framework for writing and running exploit # payloads. # # Copyright (c) 2007-2023 Hal Brodigan (postmodern.mod3 at gmail.com) # # ronin-payloads is free software: you can redistribute it and/or modify # it under the terms of the GNU Lesser General Public License as published # by the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # ronin-payloads is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Lesser General Public License for more details. # # You should have received a copy of the GNU Lesser General Public License # along with ronin-payloads. If not, see . # require 'ronin/payloads/binary_payload' module Ronin module Payloads # # A {Payload} class that represents all Go payloads. # class GoPayload < BinaryPayload # # Returns the type or kind of payload. # # @return [Symbol] # # @note # This is used internally to map an payload class to a printable type. # # @api private # def self.payload_type :go end # # Compiles one or more source files using the `go build` command. # # @param [Array] source_files # The source file(s) to compile. # # @param [String, nil] output # The output file path. # # @raise [BuildFailed] # The `go build` command failed or is not installed. # def compile(*source_files, output: nil) args = ['go', 'build'] if output args << '-o' << output end args.concat(source_files) case system(*args) when false raise(BuildFailed,"go command failed: #{args.join(' ')}") when nil raise(BuildFailed,"go command not installed") end end end end end