示例#1
0
func bundlerNew(bm *BundleManager) *bundler {
	self := &bundler{bm: bm, wake: sync.NewCond(&sync.Mutex{})}
	err := bm.BundleStateChangeWatchSet(BundleState_ToBundle, func(user string, bundle_id int, state BundleState) { self.Wakeup() })
	if err != nil {
		log.Printf("Failed to register wakeup callback with the bundle manager.")
		return nil
	}
	go func() {
		for {
			found := false
			ids, err := self.bm.bundleIdsForStateGet(BundleState_ToBundle)
			if err == nil && ids != nil {
				for user, list := range ids {
					for _, id := range list {
						log.Printf("Got bundles %v %v\n", user, id)
						bundle_filename := filepath.Join(common.BaseDir, "bundles", "inprogress", strconv.Itoa(id)+".tar")
						file, err := os.Create(bundle_filename)
						if err != nil {
							log.Printf("Could not open temp file. %v", err)
							continue
						}
						file.Close()
						if common.System {
							if platform.PlatformGet() == platform.Linux {
								cmd := exec.Command("setfacl", "-m", "user:"******":rw", bundle_filename)
								err = cmd.Run()
								if err != nil {
									log.Printf("Failed to set acl on bundle. %v\n", err)
									continue
								}
							} else if platform.PlatformGet() == platform.Windows {
								//TODO - remove or fix so that permissions are set for system user only.
								/*userstr := common.UserdDefaultUsername() + ":C"
								err = common.Cacls(bundle_filename, "/p", "NT AUTHORITY\\SYSTEM:f", userstr, "BUILTIN\\Administrators:F")
								if err != nil {
									log.Printf("Failed to run cacls %v\n", err)
									continue
								}*/
							}
						}
						err = bundleFiles(bundle_filename, user, id)
						if err != nil {
							continue
						}
						var newstate BundleState
						metadata_filename := filepath.Join(common.BaseDir, "bundles", "inprogress", strconv.Itoa(id)+".txt")
						file, err = os.Create(metadata_filename)
						if err != nil {
							log.Printf("Could not open temp file. %v", err)
							continue
						} else {
							err = writeMetadata(self.bm, user, id, file)
							file.Close()
							if err != nil {
								continue
							}
							_, _, fe, _, err := common.UserBundleFile(user, bundle_filename, metadata_filename, "metadata.txt")
							if err != nil || rpc.BundleFileErrorIsError(fe) {
								log.Printf("Failed to add metadata to bundle. %v", err)
								continue
							}
							newstate = BundleState_ToUpload
						}
						if common.System {
							if platform.PlatformGet() == platform.Linux {
								cmd := exec.Command("setfacl", "-x", "user:"******"Failed to remove acl on bundle. %v\n", err)
									continue
								}
							} else if platform.PlatformGet() == platform.Windows {
								err = common.Cacls(bundle_filename, "/p", "NT AUTHORITY\\SYSTEM:f", "BUILTIN\\Administrators:F")
								if err != nil {
									log.Printf("Failed to run cacls %v\n", err)
									continue
								}
							}
						}
						bundle_new_filename := filepath.Join(common.BaseDir, "bundles", strconv.Itoa(id)+".tar")
						os.Remove(bundle_new_filename)
						err = os.Rename(bundle_filename, bundle_new_filename)
						if err != nil {
							log.Printf("Failed to rename bundle. %v\n", err)
							continue
						}
						err = bm.bundleStringSet(user, id, "bundle_location", bundle_new_filename)
						if err != nil {
							log.Printf("Failed to set bundle location. %v\n", err)
							continue
						}
						err = bm.BundleStateSet(user, id, newstate)
						if err != nil {
							log.Printf("Failed to transition bundle from bundling to upload. %v\n", err)
							continue
						}
						found = true
					}
				}
			}
			if found == false {
				self.wake.L.Lock()
				count, err := self.bm.bundleIdsForStateCount(BundleState_ToBundle)
				if err == nil && count < 1 {
					self.wake.Wait()
				}
				self.wake.L.Unlock()
			}
		}
	}()
	return self
}
示例#2
0
func bundleFiles(bundle_filename string, user string, bundle_id int) error {
	b, err := bm.BundleGet(user, bundle_id)
	if err != nil {
		return err
	}
	bfs, err := b.FilesGet()
	if err != nil {
		log.Printf("Failed to get files from bundle.\n")
		return nil
	}
	for _, bf := range bfs {
		pacifica_filename, err := bf.PacificaFilenameGet()
		if err != nil {
			return err
		}
		if strings.HasPrefix(pacifica_filename, "/") {
			pacifica_filename = pacifica_filename[1:]
		}
		local_filename, err := bf.LocalFilenameGet()
		if err != nil {
			return err
		}
		disable_on_error, err := bf.DisableOnErrorGet()
		if err != nil {
			return err
		}
		log.Printf("To bundle: %v %v %v\n", bundle_filename, local_filename, pacifica_filename)
		result, sha1, fe, mtime, err := common.UserBundleFile(user, bundle_filename, local_filename, pacifica_filename)
		if err != nil {
			log.Printf("Failed to talk to the userd bundler. %v\n", err)
			return err
		}
		if rpc.BundleFileErrorIsError(fe) {
			if rpc.BundleFileErrorIsTransient(fe, !disable_on_error) {
				log.Printf("Transient issue. %v\n", fe)
				return errors.New("Transient error.")
			} else if disable_on_error && rpc.BundleFileErrorIsTransientError(fe) {
				//FIXME better error message.
				log.Printf("Transient error. %v\n", fe)
				err = bm.BundleFileStringSet(user, bf.bundle.id, bf.id, "disable_on_error_msg", strconv.Itoa(int(fe)))
				if err != nil {
					return err
				}
				continue
			} else if rpc.BundleFileErrorIsPermanentError(fe, !disable_on_error) {
				log.Printf("Permanent error. %v\n", fe)
				err = bm.BundleStateSet(user, bundle_id, BundleState_Error)
				if err != nil {
					log.Printf("Failed to transition bundle to error state. %v\n", err)
					return err
				}
				err = bm.bundleStringSet(user, bundle_id, "error", "Bundle Error")
				if err != nil {
					log.Printf("Failed to set error message. %v\n", err)
					return err
				}
				//FIXME better error message.
				err = bm.BundleFileStringSet(user, bf.bundle.id, bf.id, "disable_on_error_msg", strconv.Itoa(int(fe)))
				if err != nil {
					return err
				}
				return errors.New("PermanentError")
			}
		}
		err = bf.Sha1Set(sha1)
		if err != nil {
			log.Printf("Failed to set sha1. %v\n", err)
			return errors.New("Sha1")
		}
		err = bf.MtimeSet(mtime)
		if err != nil {
			log.Printf("Failed to set  mtime. %v\n", err)
			return errors.New("Mtime")
		}
		log.Printf("Got from bundler: %v %v %v\n", fe, result, sha1)
	}
	return nil
}