Ejemplo n.º 1
0
	def write(self, contents, meta, httpmeta=None):
		"""Writes a 411 file with the given plaintext contents, and
		meta information to the local filesystem. Respects the master
		file's owner, group, and permission bits. Contents and meta 
		arguments are in the format from get()."""

		try:
			# Get standard 411 headers - safe because they were
			# encrypted.
			filename = meta["name"]
			owner = meta["owner"]
			uid, gid = map(int, owner.split("."))
			mode_oct = meta["mode"]
		except ValueError:
			raise Error411, "File has malformed 411 headers."

		try:
			# This possibly could be done with an OO file class,
			# but not worth it if we only have two choices.
			mode = int(mode_oct, 8)
			if stat.S_ISDIR(mode):
				mkdir(filename)
			elif stat.S_ISREG(mode):
				path, name = os.path.split(filename)
				mkdir(path)
				(f, tmp_filename) = tempfile.mkstemp()
				os.write(f, contents)
				os.close(f)
				os.chown(tmp_filename, uid, gid)
				os.chmod(tmp_filename, stat.S_IMODE(mode))
				# mv temp file
				shutil.move(tmp_filename, filename)
				
			else:
				raise Error411, \
					"File %s is an unknown type" % filename
			# Set file meta data
			os.chown(filename, uid, gid)
			os.chmod(filename, stat.S_IMODE(mode))

		except (IOError, OSError), msg:
			raise Error411, \
				"Could not write %s: %s" % (filename, msg)
Ejemplo n.º 2
0
    def write(self, contents, meta, httpmeta=None):
        """Writes a 411 file with the given plaintext contents, and
		meta information to the local filesystem. Respects the master
		file's owner, group, and permission bits. Contents and meta 
		arguments are in the format from get()."""

        try:
            # Get standard 411 headers - safe because they were
            # encrypted.
            filename = meta["name"]
            owner = meta["owner"]
            uid, gid = map(int, owner.split("."))
            mode_oct = meta["mode"]
        except ValueError:
            raise Error411, "File has malformed 411 headers."

        try:
            # This possibly could be done with an OO file class,
            # but not worth it if we only have two choices.
            mode = int(mode_oct, 8)
            if stat.S_ISDIR(mode):
                mkdir(filename)
            elif stat.S_ISREG(mode):
                path, name = os.path.split(filename)
                mkdir(path)
                (f, tmp_filename) = tempfile.mkstemp()
                os.write(f, contents)
                os.close(f)
                os.chown(tmp_filename, uid, gid)
                os.chmod(tmp_filename, stat.S_IMODE(mode))
                # mv temp file
                shutil.move(tmp_filename, filename)

            else:
                raise Error411, \
                 "File %s is an unknown type" % filename
            # Set file meta data
            os.chown(filename, uid, gid)
            os.chmod(filename, stat.S_IMODE(mode))

        except (IOError, OSError), msg:
            raise Error411, \
             "Could not write %s: %s" % (filename, msg)
Ejemplo n.º 3
0
    def run(self, filename):
        """Prepare a 411 file for publishing. Uses a hybrid
		RSA-Blowfish encryption method."""

        # set umask to something sane
        os.umask(0022)

        if not filename:
            raise Error411, "Please specify a file."
        if self.chroot:
            filename = os.path.join(self.chroot, filename)

        if self.doName:
            # Used by 411 makefile.
            fullpath = os.path.abspath(filename)
            if self.chroot:
                fullpath = fullpath.replace(self.chroot, '')
            print os.path.join(self.dir411, self.group, self.path411(fullpath))
            return

        if not os.access(filename, os.R_OK):
            raise Error411, "I cannot find or see '%s'" % filename
        if not os.access(self.dir411, os.W_OK):
            raise Error411, "I do not have permission to write to '%s'" % self.dir411

        # What kind of file is this?
        s = os.stat(filename)
        # File type and permissions in a decimal int.
        mode = s[stat.ST_MODE]
        # File type and permission bits in an octal string.
        mode_oct = oct(mode)
        # Owner and group id as numeric ints (dont want to rely on
        # etc/group file).
        owner = "%d.%d" % (s[stat.ST_UID], s[stat.ST_GID])

        # Get the normalized, absolute path name of the file.
        fullpath = os.path.abspath(filename)
        if self.chroot:
            fullpath = fullpath.replace(self.chroot, '')
        filename411 = self.path411(fullpath)

        #####MODIFICATIONS TO SUPPORT FILTERING######
        # Look for filters in /opt/rocks/var/plugins/411
        plugin411_path = '/opt/rocks/var/plugins/411/'
        sys.path.append(plugin411_path)
        mod_file = None
        # Iterate through all the plugins to find the one
        # that will filter the file.
        for plugin_file in os.listdir(plugin411_path):
            if not plugin_file.endswith('.py'):
                continue
            mod_name = plugin_file.split('.py')[0]
            # Import the plugin
            mod = __import__(mod_name)
            # Get the filename that the plugin will
            # process
            plugin = mod.Plugin()
            if plugin.get_filename() == fullpath:
                mod_file = plugin_file
                break
            else:
                plugin = None
        if mod_file == None:
            filter = None
        else:
            f = open(os.path.join(plugin411_path, mod_file), 'r')
            filter = f.read()
            f.close()
        #####END MODIFICATIONS TO SUPPORT FILTERING######

        # 411 file meta data encoded in file. HTTP header style.
        header = "<?xml version='1.0' standalone='yes'?>\n"
        header = "<service411>\n"
        header += "<name>%s</name>\n" % fullpath
        header += "<mode>%s</mode>\n" % mode_oct
        header += "<owner>%s</owner>\n" % owner

        plaintext = header

        if stat.S_ISREG(mode):
            file = open(filename, 'r')
            plaintext += "<content>\n<![CDATA[\n"

            content = file.read()
            # Support for pre-send function
            try:
                # if the pre_send function exists in the plugin
                # filter the content through it, before sending
                # it over
                f = getattr(plugin, 'pre_send')
                content = f(content)
            except AttributeError:
                pass
            plaintext += base64.b64encode(content)
            plaintext += "]]>\n</content>\n"
            file.close()

        elif stat.S_ISDIR(mode):
            plaintext += "<directory>%s</directory>\n" % filename411

        else:
            raise Error411, "I can only publish a regular file or a directory."

        if filter is not None:
            plaintext += "<filter>\n<![CDATA[\n"
            plaintext += base64.b64encode(filter)
            plaintext += "]]>\n</filter>\n"
        plaintext += "</service411>"

        # Call the cryptography engine.
        msg = self.encrypt(plaintext)

        # Write the file with the correct name
        dir411 = self.dir411
        if self.group:
            dir411 = os.path.join(self.dir411, self.group)
            mkdir(dir411)

        file = open(os.path.join(dir411, filename411), 'w')
        file.write(msg)
        file.close()

        print "411 Wrote: %s/%s" % (dir411, filename411)
        print "Size: %s/%s bytes (encrypted/plain)" % \
         (len(msg), len(plaintext))
        if self.doAlert:
            self.sendAlert(filename411)
Ejemplo n.º 4
0
	def run(self, filename):
		"""Prepare a 411 file for publishing. Uses a hybrid
		RSA-Blowfish encryption method."""

		if not filename:
			raise Error411, "Please specify a file."
		if self.chroot:
			filename = os.path.join(self.chroot, filename)

		if self.doName:
			# Used by 411 makefile.
			fullpath = os.path.abspath(filename)
			if self.chroot:
				fullpath = fullpath.replace(self.chroot, '')
			print os.path.join(self.dir411, self.group,
				self.path411(fullpath))
			return

		if not os.access(filename, os.R_OK):
			raise Error411, "I cannot find or see '%s'" % filename
		if not os.access(self.dir411, os.W_OK):
			raise Error411, "I do not have permission to write to '%s'" % self.dir411

		# What kind of file is this?
		s = os.stat(filename)
		# File type and permissions in a decimal int.
		mode = s[stat.ST_MODE]
		# File type and permission bits in an octal string.
		mode_oct = oct(mode)
		# Owner and group id as numeric ints (dont want to rely on
		# etc/group file).
		owner = "%d.%d" % (s[stat.ST_UID], s[stat.ST_GID])

		# Get the normalized, absolute path name of the file.
		fullpath = os.path.abspath(filename)
		if self.chroot:
			fullpath = fullpath.replace(self.chroot, '')
		filename411 = self.path411(fullpath)


		#####MODIFICATIONS TO SUPPORT FILTERING######
		# Look for filters in /opt/rocks/var/plugins/411
		plugin411_path = '/opt/rocks/var/plugins/411/'
		sys.path.append(plugin411_path)
		mod_file = None
		# Iterate through all the plugins to find the one
		# that will filter the file.
		for plugin_file in os.listdir(plugin411_path):
			if not plugin_file.endswith('.py'):
				continue
			mod_name = plugin_file.split('.py')[0]
			# Import the plugin
			mod = __import__(mod_name)
			# Get the filename that the plugin will
			# process
			plugin = mod.Plugin()
			if plugin.get_filename() == fullpath:
				mod_file = plugin_file
				break
			else:
				plugin = None
		if mod_file == None:
			filter = None
		else:
			f = open(os.path.join(plugin411_path,mod_file), 'r')
			filter = f.read()
			f.close()
		#####END MODIFICATIONS TO SUPPORT FILTERING######


		# 411 file meta data encoded in file. HTTP header style.
		header = "<?xml version='1.0' standalone='yes'?>\n"
		header = "<service411>\n"
		header += "<name>%s</name>\n" 	% fullpath
		header += "<mode>%s</mode>\n" 	% mode_oct
		header += "<owner>%s</owner>\n"	% owner

		plaintext = header

		if stat.S_ISREG(mode):
			file = open(filename, 'r')
			plaintext += "<content>\n<![CDATA[\n"

			content = file.read()
			# Support for pre-send function
			try:
				# if the pre_send function exists in the plugin
				# filter the content through it, before sending
				# it over
				f = getattr(plugin, 'pre_send')
				content = f(content)
			except AttributeError:
				pass
			plaintext += base64.b64encode(content)
			plaintext += "]]>\n</content>\n"
			file.close()

		elif stat.S_ISDIR(mode):
			plaintext += "<directory>%s</directory>\n" % filename411

		else:
			raise Error411, "I can only publish a regular file or a directory."

		if filter is not None:
			plaintext += "<filter>\n<![CDATA[\n"
			plaintext += base64.b64encode(filter)
			plaintext += "]]>\n</filter>\n"
		plaintext += "</service411>"

		# Call the cryptography engine.
		msg = self.encrypt(plaintext)

		# Write the file with the correct name
		dir411 = self.dir411
		if self.group:
			dir411 = os.path.join(self.dir411, self.group)
			mkdir(dir411)

		file = open(os.path.join(dir411, filename411), 'w')
		file.write(msg)
		file.close()

		print "411 Wrote: %s/%s" % (dir411, filename411)
		print "Size: %s/%s bytes (encrypted/plain)" % \
			(len(msg), len(plaintext))
		if self.doAlert:
			self.sendAlert(filename411)