Example #1
0
func NewTestCase(t *testing.T) *testCase {
	me := new(testCase)
	me.tester = t
	me.secret = RandomBytes(20)
	me.tmp, _ = ioutil.TempDir("", "")

	workerTmp := me.tmp + "/worker-tmp"
	os.Mkdir(workerTmp, 0700)
	opts := WorkerOptions{
		Secret:   me.secret,
		TempDir:  workerTmp,
		CacheDir: me.tmp + "/worker-cache",
		Jobs:     1,
	}

	me.worker = NewWorkerDaemon(&opts)

	// TODO - pick unused port
	me.coordinatorPort = int(rand.Int31n(60000) + 1024)
	me.workerPort = int(rand.Int31n(60000) + 1024)

	coordinatorAddr := fmt.Sprintf(":%d", me.coordinatorPort)
	var wg sync.WaitGroup

	wg.Add(2)
	go func() {
		me.coordinator = NewCoordinator(me.secret)
		go me.coordinator.PeriodicCheck()
		go me.coordinator.ServeHTTP(me.coordinatorPort)
		wg.Done()
	}()
	go func() {
		masterCache := NewContentCache(me.tmp + "/master-cache")
		me.master = NewMaster(
			masterCache, coordinatorAddr,
			[]string{},
			me.secret, []string{}, 1)
		me.master.SetKeepAlive(1.0)
		me.socket = me.tmp + "/master-socket"
		go me.master.Start(me.socket)
		wg.Done()
	}()
	go me.worker.RunWorkerServer(me.workerPort, coordinatorAddr)

	wg.Wait()
	for me.coordinator.WorkerCount() == 0 {
		log.Println("reporting...")
		me.worker.report(coordinatorAddr, me.workerPort)
	}

	wd := me.tmp + "/wd"
	os.MkdirAll(wd, 0755)
	return me
}
Example #2
0
File: misc.go Project: lht/termite
func RandomBytes(n int) []byte {
	c := make([]byte, 0)
	for i := 0; i < n; i++ {
		c = append(c, byte(rand.Int31n(256)))
	}
	return c
}
Example #3
0
func AuthenticateClient(conn net.Conn, secret []byte) os.Error {
	challenge := make([]byte, 0)
	for i := 0; i < challengeLength; i++ {
		challenge = append(challenge, byte(rand.Int31n(256)))
	}

	_, err := conn.Write(challenge)
	if err != nil {
		return err
	}

	h := hmac.NewSHA1(secret)
	_, err = h.Write(challenge)
	expected := h.Sum()
	response := make([]byte, len(expected))
	n, err := conn.Read(response)
	if err != nil {
		return err
	}
	response = response[:n]

	if bytes.Compare(response, expected) != 0 {
		log.Println("Authentication failure from", conn.RemoteAddr())
		return os.NewError("Mismatch in response")
	}
	conn.Write([]byte("OK"))
	return nil
}
Example #4
0
func randomPlusMinus() (val int) {
	switch rand.Int31n(3) {
	case 0:
		val = 0
	case 1:
		val = 1
	case 2:
		val = -1
	}
	return
}
Example #5
0
/*
	return a random particle from p that is not in e
*/
func randParticle(p Particles, e Particles) (r *Particle) {
	contains := true
	for contains {
		r = p[rand.Int31n(int32(len(p)))]
		contains = false
		for i := 0; e != nil && i < len(e); i++ {
			if e[i] == r {
				contains = true
			}
		}
	}
	return
}
Example #6
0
func TestAuthenticate(t *testing.T) {
	secret := RandomBytes(20)
	port := int(rand.Int31n(60000) + 1024)

	out := make(chan net.Conn)
	go SetupServer(int(port), secret, out)
	time.Sleep(1e9)
	hostname, _ := os.Hostname()
	addr := fmt.Sprintf("%s:%d", hostname, port)
	_, err := DialTypedConnection(addr, RPC_CHANNEL, secret)
	if err != nil {
		t.Fatal("unexpected failure", err)
	}
	<-out

	c, err := DialTypedConnection(addr, RPC_CHANNEL, []byte("foobar"))
	if c != nil {
		t.Error("expect failure")
	}
}
Example #7
0
/*
 Query1, DML(INSERT) on role_info.
*/
func queryRoleInfo1(cli *mysql.Client) {
	query_name := "[RoleInfo.query1]"
	loop_count := 1000
	type row struct {
		// PK: (userid, roleid, server, date)
		userid int32
		roleid int32
		server int8
		date   string
		name   string
	}
	fmt.Printf("%s started...\n", query_name)
	starttime := time.Nanoseconds()
	for i := 0; i < loop_count; i++ {
		newrow := &row{}
		newrow.userid = rand.Int31n(100000)
		newrow.roleid = rand.Int31n(10000)
		newrow.server = int8(rand.Intn(100))
		newrow.date = "2011-06-26 10:20:24"
		newrow.name = "testname"
		if VERBOSE_MODE {
			fmt.Printf("%s INSERT IGNORE INTO role_info "+
				"VALUES(%d, %d, %d, %q, %q)\n",
				query_name, newrow.userid, newrow.roleid, newrow.server,
				newrow.date, newrow.name)
		} else {
			if i == 0 {
				fmt.Printf("%s INSERT IGNORE INTO role_info "+
					"VALUES(%d, %d, %d, %q, %q)\n",
					query_name, newrow.userid, newrow.roleid, newrow.server,
					newrow.date, newrow.name)
				fmt.Printf("%s ...(%d lines)\n", query_name, loop_count-1)
			}
		}
		if !DRYRUN_MODE {
			cli.Start()
			stmt, err := cli.Prepare("INSERT IGNORE INTO role_info " +
				"VALUES(?,?,?,?,?)")
			if err != nil {
				fmt.Fprintln(os.Stderr, query_name+err.String())
				os.Exit(1)
			}
			err = stmt.BindParams(newrow.userid, newrow.roleid, newrow.server,
				newrow.date, newrow.name)
			if err != nil {
				fmt.Fprintln(os.Stderr, query_name+err.String())
				os.Exit(1)
			}
			err = stmt.Execute()
			if err != nil {
				fmt.Fprintln(os.Stderr, query_name+err.String())
				os.Exit(1)
			}
			cli.Rollback()
		}
	}
	endtime := time.Nanoseconds()
	if !DRYRUN_MODE {
		fmt.Printf("%s ended. Averange query time: %d nanosecs.\n", query_name,
			(endtime-starttime)/((int64)(loop_count)))
	} else {
		fmt.Printf("%s ended.\n", query_name)
	}
}
Example #8
0
func nagomi_bot(nagomin vector.StringVector) vector.StringVector {
	rand.Seed(time.Nanoseconds())

	// client情報を作成
	c := twitter.ClientBuilder(BOT_NAME, PASSWORD)

	// Timelineを取得
	tweets := c.HomeTimeline(0, 0, 0, 0)
	//fmt.Printf("tweet = %v\n",tweets)

	// botへのリプライを取得
	direct_messages := c.DirectMessages(0, 0, 0, 0)

	//DMの内容をデータベースに登録する処理の開始
	for i := 0; i < len(direct_messages); i++ {
		f, _ := file.Open("nagomin.txt", os.O_WRONLY|os.O_APPEND, 644)
		bufy := bufio.NewWriter(f)

		//DMの内容がすでにデータベースに登録されてないかチェック
		flag := 0
		for j := 0; j < len(nagomin); j++ {
			//登録されている場合の処理
			if string(nagomin[j]) == direct_messages[i].Text {
				flag = 1
				break
			}
		}
		//データベースに新規登録処理
		if flag == 0 {
			fmt.Printf("%v\n", direct_messages[i].Text)
			bufy.WriteString(direct_messages[i].Text + "\n")
			//データベースに追加
			bufy.Flush()
			//すでに読み込んでいるデータベースを更新
			nagomin.Push(direct_messages[i].Text)
		}
		f.Close()
	}

	//現在の時刻の取得
	current_t := time.LocalTime()

	tweet := tweets[0]
	//最新ツイートの時刻を取得
	last_tweet_t, _ := time.Parse(time.RubyDate, tweet.Created_at)
	timespan := (current_t.Seconds() - last_tweet_t.Seconds())

	//ツイートする最低間隔を設定(botが連続でツイートすることを防止する)
	if timespan < MIN_TIME {
		return nagomin
	}

	//APIからのつぶやきを最新ツイートに含めないようにする処理
	i := 0
	for i = 0; tweets[i].Source == "<a href=\"http://apiwiki.twitter.com/\" rel=\"nofollow\">API</a>"; i++ {
	}

	tweet = tweets[i]
	last_tweet_t, _ = time.Parse(time.RubyDate, tweet.Created_at)

	//現在時刻と最新ツイートの時刻との差分を取得
	timespan = (current_t.Seconds() - last_tweet_t.Seconds())

	//ツイート処理
	if timespan > TWEET_TIME {
		str := nagomin[rand.Int31n(int32(len(nagomin)))]
		fmt.Printf("%d秒経過したので,ツイートします.\n", timespan)
		c.Post(fmt.Sprintf("【nagomin】%s", str))
	}
	return nagomin
}
Example #9
0
func NewCellState() *CellState {
	return &CellState{rand.Int31n(6)}
}